| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- # -*- coding: utf-8 -*-
- from django.apps import apps
- from django.core.management.base import CommandError, LabelCommand
- from django.utils.encoding import force_str
- from django_extensions.management.utils import signalcommand
- class Command(LabelCommand):
- help = "Outputs the specified model as a form definition to the shell."
- def add_arguments(self, parser):
- parser.add_argument('label', type=str,
- help='application name and model name')
- parser.add_argument(
- "--fields", "-f", action="append", dest="fields", default=[],
- help="Describe form with these fields only"
- )
- @signalcommand
- def handle(self, *args, **options):
- label = options['label']
- fields = options['fields']
- return describe_form(label, fields)
- def describe_form(label, fields):
- """ Return a string describing a form based on the model """
- try:
- app_name, model_name = label.split('.')[-2:]
- except (IndexError, ValueError):
- raise CommandError("Need application and model name in the form: appname.model")
- model = apps.get_model(app_name, model_name)
- opts = model._meta
- field_list = []
- for f in opts.fields + opts.many_to_many:
- if not f.editable:
- continue
- if fields and f.name not in fields:
- continue
- formfield = f.formfield()
- if '__dict__' not in dir(formfield):
- continue
- attrs = {}
- valid_fields = ['required', 'initial', 'max_length', 'min_length', 'max_value', 'min_value', 'max_digits', 'decimal_places', 'choices', 'help_text', 'label']
- for k, v in formfield.__dict__.items():
- if k in valid_fields and v is not None:
- # ignore defaults, to minimize verbosity
- if k == 'required' and v:
- continue
- if k == 'help_text' and not v:
- continue
- if k == 'widget':
- attrs[k] = v.__class__
- elif k in ['help_text', 'label']:
- attrs[k] = str(force_str(v).strip())
- else:
- attrs[k] = v
- params = ', '.join(['%s=%r' % (k, v) for k, v in sorted(attrs.items())])
- field_list.append(' %(field_name)s = forms.%(field_type)s(%(params)s)' % {
- 'field_name': f.name,
- 'field_type': formfield.__class__.__name__,
- 'params': params
- })
- return '''
- from django import forms
- from %(app_name)s.models import %(object_name)s
- class %(object_name)sForm(forms.Form):
- %(field_list)s
- ''' % {'app_name': app_name, 'object_name': opts.object_name, 'field_list': '\n'.join(field_list)}
|
