U PN^) @sdZddlmZddlZddlmZddlmZddlm Z ddlm Z m Z d Z d d Z Gd d d eZGdddeZGdddeZdddZde dddddddf ddZdS)z8 Tools for generating forms based on SQLAlchemy models. )unicode_literalsN)fields) validators)Form)QuerySelectFieldQuerySelectMultipleField) model_fields model_formcsfdd}|S)Ncst|_|SN) frozenset_converter_for)funcargs?/tmp/pip-install-o1yuzyg2/WTForms/wtforms/ext/sqlalchemy/orm.py_inners zconverts.._innerr)rrrrrconvertss rc@seZdZddZdS)ModelConversionErrorcCst||dSr ) Exception__init__)selfmessagerrrrszModelConversionError.__init__N)__name__ __module__ __qualname__rrrrrrsrc@s eZdZdddZdddZdS) ModelConverterBaseTcCsN||_|si}t|D],}t||}t|dr|jD] }|||<q4q||_dS)Nr )use_mrodirgetattrhasattrr converters)rr"rnameobj classnamerrrrs     zModelConverterBase.__init__Ncst|dst|dsdSt|ds8t|jdkr8tdggdd}d}d}d} t|ds|jd}t|dd} | dk rt| dd} | dk rt| r| dn| } | |d<|jr|d t n|d t |j rt t|j} n t|jg} | D]J} d | j| jf} | d r*| d d} | |jkr|j| }qq| D]&} | j|jkrN|j| j}qqNtd |j| dfnfstd|j|jjd}|jD]}|djsd}q||fddd|j|jj}|r||||||||dS)Ncolumns directionrz?Do not know how to convert multiple-column properties currently)rfiltersdefaultrr)argrz%s.%sZ sqlalchemy z+Could not find field converter for %s (%r).z)Cannot convert field %s, need DB session.TFcsSr )queryallr db_sessionZ foreign_modelrryz,ModelConverterBase.convert..)Z allow_blankZ query_factory)modelmapperpropcolumn field_args)r!lenr& TypeErrorr callablenullableappendrOptionalZRequiredrinspectgetmrotyperr startswithr"rkeyr3class_Zlocal_remote_pairsupdater'r#)rr2r3r4r6r/kwargs converterr5typesr)Zcallable_defaultZcol_typeZ type_stringr:pairrr.rconvert-sx                zModelConverterBase.convert)T)N)rrrrrHrrrrrs rcseZdZd;fdd ZeddZeddd d Zed d d ddddZedddZ edddZ edddZ edddZ eddd d!Z ed"d#d$d%Zed&d'd(d)Zed*d+d,d-Zed.d/d0Zed1d2d3Zed4d5d6Zed7d8d9d:ZZS)<ModelConverterNTcstt|j||ddS)N)r)superrIr)rZextra_convertersr __class__rrrszModelConverter.__init__cKs&|jjr"|dtj|jjddS)Nr)max)r?lengthr;rZLength)clsr5r6extrarrr_string_commonszModelConverter._string_commonStringZUnicodecKs |jfd|i|tjf|SNr6)rQf TextFieldrr6rPrrr conv_StringszModelConverter.conv_Stringz types.TextZ UnicodeTextztypes.LargeBinaryz types.Binaryzsql.sqltypes.TextcKs |jfd|i|tjf|SrS)rQrTZ TextAreaFieldrVrrr conv_TextszModelConverter.conv_TextBooleancKs tjf|Sr )rTZ BooleanFieldrVrrr conv_BooleanszModelConverter.conv_BooleanDatecKs tjf|Sr )rTZ DateFieldrVrrr conv_DateszModelConverter.conv_DateDateTimecKs tjf|Sr )rTZ DateTimeFieldrVrrr conv_DateTimeszModelConverter.conv_DateTimeEnumcKs*d|krdd|jjD|d<tjf|S)NchoicescSsg|] }||fqSrr).0errr sz,ModelConverter.conv_Enum..)r?enumsrTZ SelectField)rr5r6rPrrr conv_EnumszModelConverter.conv_EnumZIntegerZ SmallIntegercKs4t|jdd}|r(|dtjddtjf|S)NunsignedFrr)min)r r?r;r NumberRangerTZ IntegerField)rr5r6rPrfrrrhandle_integer_typessz#ModelConverter.handle_integer_typesZNumericZFloatcKs*t|jdd}|dk r||d<tjf|S)NZscaleplaces)r r?rTZ DecimalField)rr5r6rPrkrrrhandle_decimal_typessz#ModelConverter.handle_decimal_typeszdatabases.mysql.MSYearzdialects.mysql.base.YEARcKs$|dtjdddtjf|S)Nrimik)rgrM)r;rrhrTrUrVrrr conv_MSYearszModelConverter.conv_MSYearzdatabases.postgres.PGInetzdialects.postgresql.base.INETcKs*|dd|dttjf|S)Nlabelz IP Addressr) setdefaultr;rZ IPAddressrTrUrVrrr conv_PGInets zModelConverter.conv_PGInetz dialects.postgresql.base.MACADDRcKs*|dd|dttjf|S)Nrnz MAC Addressr)ror;rZ MacAddressrTrUrVrrrconv_PGMacaddrs zModelConverter.conv_PGMacaddrzdialects.postgresql.base.UUIDcKs*|dd|dttjf|S)NrnUUIDr)ror;rrrrTrUrVrrr conv_PGUuids zModelConverter.conv_PGUuidZ MANYTOONEcKs tf|Sr )rrVrrrconv_ManyToOneszModelConverter.conv_ManyToOneZ MANYTOMANYZ ONETOMANYcKs tf|Sr )rrVrrrconv_ManyToManyszModelConverter.conv_ManyToMany)NT)rrrr classmethodrQrrWrXrZr\r^rerirlrmrprqrsrtru __classcell__rrrKrrIs>              rIFc s|jj}|pt}|pi}g} |jD]F} t| ddrZ|rH| jdjrHq$n|rZ| jdjrZq$| | j | fq$rfdd| D} nrfdd| D} i} | D]0\} } | ||| | | |} | dk r| | | <q| S)z Generate a dictionary of fields for a given SQLAlchemy model. See `model_form` docstring for description of parameters. r&Nrc3s|]}|dkr|VqdSrNrrax)onlyrr s zmodel_fields..c3s|]}|dkr|VqdSrxrry)excluderrr|s ) _sa_class_managerr3rIZiterate_propertiesr r&Z foreign_keysZ primary_keyr;rArHget)r2r/r{r}r6rE exclude_pk exclude_fkr3Z propertiesr4 field_dictr#fieldr)r}r{rr s6     r Tc CsJt|dstd| p"t|jd} t||||||||d} t| |f| S)a Create a wtforms Form for a given SQLAlchemy model class:: from wtforms.ext.sqlalchemy.orm import model_form from myapp.models import User UserForm = model_form(User) :param model: A SQLAlchemy mapped model class. :param db_session: An optional SQLAlchemy Session. :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. :param exclude_pk: An optional boolean to force primary key exclusion. :param exclude_fk: An optional boolean to force foreign keys exclusion. :param type_name: An optional string to set returned type name. r~z'model must be a sqlalchemy mapped modelr)rr)r!r8strrr r?) r2r/Z base_classr{r}r6rErr type_namerrrrr s# r )NNNNNFF)__doc__ __future__rr=ZwtformsrrTrZ wtforms.formrrr__all__rrrobjectrrIr r rrrrs2    lS (