wdf) dZddlmZddlZddlmZddlmZddlm Z ddlm Z m Z d Z d Z Gd d eZGd deZGddeZ ddZde dddddddf dZdS)z8 Tools for generating forms based on SQLAlchemy models. )unicode_literalsN)fields) validators)Form)QuerySelectFieldQuerySelectMultipleField) model_fields model_formcfd}|S)Nc0t|_|SN) frozenset_converter_for)funcargss _/var/www/book.euthymeo.com/html/venv/lib/python3.11/site-packages/wtforms/ext/sqlalchemy/orm.py_innerzconverts.._inners'oo )rrs` rconvertsrs# MrceZdZdZdS)ModelConversionErrorc<t||dSr) Exception__init__)selfmessages rrzModelConversionError.__init__s4)))))rN)__name__ __module__ __qualname__rrrrrrs#*****rrceZdZddZddZdS)ModelConverterBaseTc||_|si}t|D]1}t||}t|dr|jD]}|||<2||_dS)Nr)use_mrodirgetattrhasattrr converters)rr)r%nameobj classnames rrzModelConverterBase.__init__sx  JII 0 0D$%%Cs,-- 0!$!300I,/Jy))$rNc@t|dst|dsdSt|ds't|jdkrtdggdd}d}d}d} t|ds|jd}t |dd} | /t | dd} | t | r | dn| } | |d<|jr-|d tj n,|d tj |j r'tj t|j} nt|jg} | D]J} | jd | j} | d r | d d} | |jvr|j| }nIK| D]$} | j|jvr|j| j}n"%t'd |jd| ddnsst'd|jz|jjd}|jD]}|djsd}||fdd|j|jj}|r||||||||S)Ncolumns directionrz?Do not know how to convert multiple-column properties currently)rfiltersdefaultrr1argr. sqlalchemy z#Could not find field converter for z (z).z)Cannot convert field %s, need DB session.TFcRSr)queryall) db_session foreign_modelsrz,ModelConverterBase.convert..ys!)9)9-)H)H)L)L)N)Nr) allow_blank query_factory)modelmapperpropcolumn field_args)r(lenr. TypeErrorr'callablenullableappendrOptionalRequiredr%inspectgetmrotyper r startswithr)rkeyr?class_local_remote_pairsupdater/r*)rr>r?r@rBr9kwargs converterrAtypesr1callable_defaultcol_type type_stringrFpairr:s ` @rconvertzModelConverterBase.convert-sTtY'' k0J0J  F{++ DL0A0AQ0F0FQ      t[))= =\!_F fi66G"$+7E4#@#@ #/8@AQ8R8Rh..t444XhG 'F9  C|$++J,?,A,ABBBB|$++J,?,A,ABBB| ,tFK'8'899fk**+! u u)1)<)<)O>OP )),773"-bcc"2K$/11 $ z,ModelConverter.conv_Enum..s$G$G$GaV$G$G$Grr)rLenumsrk SelectField)rrArBres r conv_EnumzModelConverter.conv_Enums@ J & &$G$GV[5F$G$G$GJy !}**z***rInteger SmallIntegerc t|jdd}|r.|dtjdt jdi|S)NunsignedFrr)minr)r'rLrGr NumberRangerk IntegerField)rrArBrers rhandle_integer_typesz#ModelConverter.handle_integer_typess[6; E::  K | $ + +J,Bq,I,I,I J J J~++ +++rNumericFloatc ^t|jdd}|||d<tjdi|S)Nscaleplacesr)r'rLrk DecimalField)rrArBrers rhandle_decimal_typesz#ModelConverter.handle_decimal_typess;gq11  #)Jx ~++ +++rzdatabases.mysql.MSYearzdialects.mysql.base.YEARc |dtjddtjdi|S)Nrimik)rrar)rGrrrkrlrms r conv_MSYearzModelConverter.conv_MSYearsA< '' (>4T(R(R(RSSS{((Z(((rzdatabases.postgres.PGInetzdialects.postgresql.base.INETc |dd|dtjt jdi|S)Nlabelz IP Addressrr) setdefaultrGr IPAddressrkrlrms r conv_PGInetzModelConverter.conv_PGInetsPg|444< '' (<(>(>???{((Z(((rz dialects.postgresql.base.MACADDRc |dd|dtjt jdi|S)Nrz MAC Addressrr)rrGr MacAddressrkrlrms rconv_PGMacaddrzModelConverter.conv_PGMacaddrsPg}555< '' (=(?(?@@@{((Z(((rzdialects.postgresql.base.UUIDc |dd|dtjt jdi|S)NrUUIDrr)rrGrrrkrlrms r conv_PGUuidzModelConverter.conv_PGUuidsOgv...< '' (9(9:::{((Z(((r MANYTOONEc tdi|Sru)rrms rconv_ManyToOnezModelConverter.conv_ManyToOnes--*---r MANYTOMANY ONETOMANYc tdi|Sru)r rms rconv_ManyToManyzModelConverter.conv_ManyToManys'55*555r)NT)rr r!r classmethodrfrrnrrrwr{rrrrrrrrrr __classcell__)r_s@rr[r[sPPPPPPWW[WXh ""))#")XlM+>Pcdd--ed-Xi,,,Xf)))Xj---Xf+++ Xi((,,)(, Xi!!,,"!, X&(BCC))DC)X)+JKK))LK) X011))21) X-..))/.) Xk...XlK((66)(66666rr[Fc |jj}|p t}|pi}g} |jD]Y} t | ddr*|r| jdjr(|r| jdjr=| | j | fZrfd| D} nrfd| D} i} | D]8\} } | ||| | | |} | | | | <9| S)z Generate a dictionary of fields for a given SQLAlchemy model. See `model_form` docstring for description of parameters. r.Nrc30K|]}|dv |VdSrNr)rxonlys r zmodel_fields..s+<.s1CCAqt7/B/Ba/B/B/B/BCCr) _sa_class_managerr?r[iterate_propertiesr'r. foreign_keys primary_keyrGrNrYget)r>r9rrrBrS exclude_pk exclude_fkr? propertiesr@ field_dictr*fields `` rr r sP $ +F-^--I!rJJ),, 4D ) )  dl1o:   Q ; 48T*++++ D<<<<<<< DCCCCCCC J %% d!! 64 NN4 *    $Jt  rTc t|dstd| pt|jdz} t ||||||||} 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. rz'model must be a sqlalchemy mapped modelr)rr)r(rDstrrr rL) r>r9 base_classrrrBrSrr type_namers rr r s}F 5- . .CABBB9S&!899I z4*i*J  J>: 6 66r)NNNNNFF)__doc__ __future__rrJwtformsrrkr wtforms.formrrr __all__rrrobjectr#r[r r rrrrsw('''''>>>>>>>>  *****9*** i i i i i i i i XP6P6P6P6P6'P6P6P6f=A=B!%%%%P"&$T$+7+7+7+7+7+7r