o 2c@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 L/var/www/html/gps/gps/lib/python3.10/site-packages/wtforms/ext/django/orm.py__init__s zModelConverterBase.__init__cCs|j|jgg|jd}|r|||jr|dt|jdur4|jdkr4|dtj |jdt |j }|j rI|j |d<t jdi|S||jvrW|j||||St|d|d}|duri||||SdS)N)label descriptionrfiltersdefaultrr)maxchoiceszconv_%sr ) verbose_name help_textrupdateblankappendrOptional max_lengthLengthtype__name__rf SelectFieldr getattr)r modelfield field_argskwargsftype converterr r rconverts,     zModelConverterBase.convertN)r __module__ __qualname__rr)r r r rr s r cseZdZejgdejddgejgdejdgejdgej dgej gdej d d giZ dfd d Z ddZddZddZddZddZddZddZZS)ModelConverter) AutoField IntegerFieldSmallIntegerFieldPositiveIntegerFieldPositiveSmallIntegerField DecimalField FloatField) FileField FilePathField ImageField DateTimeField DateField BooleanField) CharFieldPhoneNumberField SlugField TextFieldXMLFieldNcsbi}|dur |j}t|D]\}}||}|D]}|||<qq |r'||tt||dSr )DEFAULT_SIMPLE_CONVERSIONSrmake_simple_converterrsuperr,r)r extra_converterssimple_conversionsr field_type django_fieldsr(name __class__r rr<s   zModelConverter.__init__csfdd}|S)Ncsdi|S)Nr r )r#r$r&rDr r _converterJsz8ModelConverter.make_simple_converter.._converterr )r rDrJr rIrr@Is z$ModelConverter.make_simple_convertercCstdd|jji|S)Nr#r )rreltor r#r$r&r r rconv_ForeignKeyNszModelConverter.conv_ForeignKeycCs*dd}|d|tjdddi|S)NcSs"z|WSty|YSwr )timeAttributeError)objr r r time_onlyRs   z0ModelConverter.conv_TimeField..time_onlyrformatz%H:%M:%Sr )rr r7)r r#r$r&rRr r rconv_TimeFieldQszModelConverter.conv_TimeFieldcC"|dttjdi|SNrr )rremailr r=rMr r rconv_EmailFieldZzModelConverter.conv_EmailFieldcCrUrV)rr ip_addressr r=rMr r rconv_IPAddressField^rYz"ModelConverter.conv_IPAddressFieldcCrUrV)rrurlr r=rMr r r conv_URLFieldbrYzModelConverter.conv_URLFieldcs2ddlmfdd}d}tjd||d|S)Nr NOT_PROVIDEDcs8ddddd}t|rdS||vr||Stt|S)NTF)NoneNTrueFalse) isinstanceboolint)valuedr^r rcoerce_nullboolis   z=ModelConverter.conv_NullBooleanField..coerce_nullbool))NUnknown)TYes)FNo)rcoercer )django.db.models.fieldsr_r r!)r r#r$r&rhrr r^rconv_NullBooleanFieldfs   z$ModelConverter.conv_NullBooleanField)NN)rr*r+r r.r2r4r7r8r9r= TextAreaFieldr?rr@rNrTrXr[r]rn __classcell__r r rGrr,0s$        r,Nc s|pt}|pi}dd|jjD}rfdd|D}n r)fdd|D}i}|D]\}}|||||} | durC| ||<q-|S)z Generate a dictionary of fields for a given Django model. See `model_form` docstring for description of parameters. css|]}|j|fVqdSr )attname).0r r r r szmodel_fields..c3s |] }|dvr|VqdSrNr rrx)onlyr rrsc3s |] }|dvr|VqdSrtr ru)excluder rrsrxN)r,_metarr)get) r#rwryr%r(r field_dictrF model_fieldr$r )ryrwrrvs  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)rrrz object_name)r# base_classrwryr%r(r|r r rrsr)NNNN)__doc__wtformsrr rrwtforms.compatrwtforms.ext.django.fieldsr__all__objectr r,rrr r r rs      F