ManipulatorScript: make_manipulator.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.db import models
from django.template import Context, Template

p = OptionParser()

p.add_option("-p", "--project", dest="project", help="The project which contains the model")
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__("%s.%s.models" % (options.project, options.app), '', '', [options.model])

a = getattr(m, options.model)

def argValue(s, args):
    if s == 'related_query_name':
        return "???"
    else:
        return args[s]
        
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'
    if field.__class__.__name__ == 'TextField':
        fieldType = 'LargeTextField'
    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, argValue(s, args)) for s in args if args[s] not in [None, '', False, []]]
    #print field.name, extra_args
    if extra_args:        
        return '''\
            forms.%s(field_name='%s', %s),''' % (fieldType, field.name, ', '.join(extra_args))
    else:
        return '''\
            forms.%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(forms.Manipulator):
    def __init__(self, pk=None):
        if pk:
            # change
            self.original_object = {{model}}.objects.get(id=pk)
            self.pk = pk
        else:
            # add
            self.original_object = None
            self.pk = None
        
        self.fields = (
{{ fields }}
        )

    def save(self, new_data):
        if self.original_object:
            # update
            temp = dict(
{{ values }}            
            )
            for k,v in temp.iteritems():
                self.original_object.__setattr__(k, v)
            self.original_object.save()            
            return {{model}}.objects.get(id=pk)
            
        else:
            # insert
            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)