ManipulatorScript: manipulator.3.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']])
    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
    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)\
        ''' % (field.__class__.__name__, field.name, ', '.join(extra_args))
    else:
        return '''formfields.%s(field_name='%s')\
        ''' % (field.__class__.__name__, 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)