ManipulatorScript: manipulator.5.py

#!/usr/bin/env python

from optparse import OptionParser
import sys
import os

try:
    import settings
except ImportError:
    print "Settings file not found.  Place this file in the same place as manage.py"
    sys.exit()

project_directory = os.path.dirname(settings.__file__)
project_name = os.path.basename(project_directory)
sys.path.append(os.path.join(project_directory, '..'))
project_module = __import__(project_name, '', '', [''])
sys.path.pop()
os.environ['DJANGO_SETTINGS_MODULE'] = '%s.settings' % project_name

from django.core import meta
from django.core.template import Context, Template

p = OptionParser()

p.add_option("-a", "--app", dest="app", help="The app which contains the model")
p.add_option("-m", "--model", dest="model", help="The model to produce the form for")
p.add_option("-n", "--name", dest="name", help="The name of the custom manipulator")

options, args = p.parse_args()

if not (options.model and options.app):
    p.print_help()
    sys.exit()

if not options.name:
    options.name = "%s%s" % (options.app, options.model)

m = __import__("django.models.%s" % (options.app,), '', '', [options.model])

a = getattr(m, options.model)

def field_text(field):
    d = field.__dict__
    args = dict([(f, isinstance(d[f], str) and "'%s'" % d[f] or d[f]) for f in d if f not in ['db_index', 'primary_key', 'rel', 'core', 'creation_counter',
    'editable', 'name', 'default', 'column', 'attname', 'auto_now_add', 'help_text', 'auto_now', 'unique', 'verbose_name']])
    if args['blank'] is True or args['null'] is True:
        args['is_required'] = False
    else:
        args['is_required'] = True
    try:
        del(args['blank'])
        del(args['null'])
    except KeyError:
        pass
    try:
        args['validator_list'] = ['%s' % v.__name__ for v in args['validator_list']]
    except KeyError:
        pass
    if field.__class__.__name__ == 'CharField':
        fieldType = 'TextField'
    elif field.__class__.__name__ == 'ForeignKey':
        fieldType = 'SelectField'
        args['choices'] = "[('','-------')] + [(o.id, o) for o in %s.get_list()]" % (field.name.lower() + "s",)
    elif field.__class__.__name__ == 'BooleanField':
        fieldType = 'CheckboxField'
    elif field.__class__.__name__ == 'DateTimeField':
        fieldType = 'DatetimeField'
    elif field.__class__.__name__ == 'ManyToManyField':
        fieldType = 'SelectMultipleField'
        args['choices'] = "[(o.id, o) for o in %s.get_list()]" % (field.name.lower() + "s",)
    else:
        fieldType = field.__class__.__name__
    extra_args = ["%s=%s" % (s, args[s]) for s in args if args[s] not in [None, '', False, []]]
    if extra_args:
        return '''\
            formfields.%s(field_name='%s', %s),''' % (fieldType, field.name, ', '.join(extra_args))
    else:
        return '''\
            formfields.%s(field_name='%s'),''' % (fieldType, field.name)

def value_text(field):
    return """\
            %s=new_data['%s']""" % (field.name, field.name)

def m2m_text(field):
    return """        temp.set_%s(newdata['%s'])""" % (field.name, field.name)

template = Template(
"""class {{ name }}Manipulator(formfields.Manipulator):
    def __init__(self, pk=None):
        self.fields = (
{{ fields }}
        )

    def save(self, new_data):
        temp = {{ model }}(
{{ values }}
        )
{{ m2m }}
        temp.save()
        return temp
        """)

context = Context({'fields': '\n'.join([field_text(f) for f in a._meta.fields + a._meta.many_to_many if f.name !='id']), 'name': options.name, 
                'model': options.model, 'values': ',\n'.join([value_text(f) for f in a._meta.fields if f.name != 'id']),
                'm2m': '\n'.join([m2m_text(f) for f in a._meta.many_to_many])})
print template.render(context)