U PN^@sdZddlmZddlmZddlmZddlmZddlm Z dZ Gdd d e Z Gd d d e Z dd dZed d d d fddZd S)z4 Tools for generating forms based on Django models. )fields)Form) validators) iteritems)ModelSelectField) model_fields model_formc@seZdZddZddZdS)ModelConverterBasecCs ||_dSN) converters)selfr r ;/tmp/pip-install-o1yuzyg2/WTForms/wtforms/ext/django/orm.py__init__szModelConverterBase.__init__cCs|j|jgg|jd}|r$|||jr<|dt|jdk rh|jdkrh|dtj |jdt |j }|j r|j |d<t jf|S||jkr|j||||St|d|d}|dk r||||SdS)N)label descriptionrfiltersdefaultrr)maxchoiceszconv_%s)Z verbose_name help_textrupdateZblankappendrOptional max_lengthZLengthtype__name__rf SelectFieldr getattr)r modelfield field_argskwargsZftype converterr r rconverts*     zModelConverterBase.convertN)r __module__ __qualname__rr%r r r rr sr cseZdZejdddddgejddgejdd d gejd gejd gej d gej dddgej ddgiZ d$fdd Z ddZddZddZddZddZd d!Zd"d#ZZS)%ModelConverterZ AutoField IntegerFieldZSmallIntegerFieldZPositiveIntegerFieldZPositiveSmallIntegerField DecimalFieldZ FloatField FileFieldZ FilePathFieldZ ImageField DateTimeField DateField BooleanFieldZ CharFieldZPhoneNumberFieldZ SlugField TextFieldZXMLFieldNcsbi}|dkr|j}t|D]$\}}||}|D] }|||<q0q|rN||tt||dSr )DEFAULT_SIMPLE_CONVERSIONSrmake_simple_converterrsuperr(r)r Zextra_convertersZsimple_conversionsr field_typeZ django_fieldsr$name __class__r rr<s   zModelConverter.__init__csfdd}|S)Ncs f|Sr r )r r!r#r3r r _converterJsz8ModelConverter.make_simple_converter.._converterr )r r3r8r r7rr1Is z$ModelConverter.make_simple_convertercCstfd|jji|S)Nr )rreltor r r!r#r r rconv_ForeignKeyNszModelConverter.conv_ForeignKeycCs*dd}|d|tjfddi|S)NcSs(z |WStk r"|YSXdSr )timeAttributeError)objr r r time_onlyRs z0ModelConverter.conv_TimeField..time_onlyrformatz%H:%M:%S)rrr,)r r r!r#r@r r rconv_TimeFieldQszModelConverter.conv_TimeFieldcCs|dttjf|SNr)rremailrr/r;r r rconv_EmailFieldZszModelConverter.conv_EmailFieldcCs|dttjf|SrC)rr ip_addressrr/r;r r rconv_IPAddressField^sz"ModelConverter.conv_IPAddressFieldcCs|dttjf|SrC)rrurlrr/r;r r r conv_URLFieldbszModelConverter.conv_URLFieldcs2ddlmfdd}d}tjf||d|S)Nr NOT_PROVIDEDcs<ddddd}t|rdS||kr,||Stt|SdS)NTF)NoneNTrueFalse) isinstanceboolint)valuedrJr rcoerce_nullboolis  z=ModelConverter.conv_NullBooleanField..coerce_nullbool))NUnknown)TZYes)FZNo)rZcoerce)Zdjango.db.models.fieldsrKrr)r r r!r#rTrr rJrconv_NullBooleanFieldfs  z$ModelConverter.conv_NullBooleanField)NN)rr&r'rr)r*r+r,r-r.r/Z TextAreaFieldr0rr1r<rBrErGrIrV __classcell__r r r5rr(0s2    r(Nc s|pt}|pi}dd|jjD}r<fdd|D}nrRfdd|D}i}|D],\}}|||||} | dk rZ| ||<qZ|S)z Generate a dictionary of fields for a given Django model. See `model_form` docstring for description of parameters. css|]}|j|fVqdSr )Zattname).0rr r r szmodel_fields..c3s|]}|dkr|VqdSrNr rXx)onlyr rrYs c3s|]}|dkr|VqdSrZr r[)excluder rrYs N)r(_metarr%get) r r]r^r"r$r field_dictr4Z model_fieldr!r )r^r]rrvs   rcCs&t|||||}t|jjd|f|S)a Create a wtforms Form for a given Django model class:: from wtforms.ext.django.orm import model_form from myproject.myapp.models import User UserForm = model_form(User) :param model: A Django ORM model class :param base_class: Base form class to extend from. Must be a ``wtforms.Form`` subclass. :param only: An optional iterable with the property names that should be included in the form. Only these properties will have fields. :param exclude: An optional iterable with the property names that should be excluded from the form. All other properties will have fields. :param field_args: An optional dictionary of field names mapping to keyword arguments used to construct each field object. :param converter: A converter to generate the fields based on the model properties. If not set, ``ModelConverter`` is used. r)rrr_Z object_name)r Z base_classr]r^r"r$rar r rrsr)NNNN)__doc__ZwtformsrrrrZwtforms.compatrZwtforms.ext.django.fieldsr__all__objectr r(rrr r r rs      F