D6iNPddlZddlmZddlmZddlmZddlZddl m Z ddl m Z ddl mZmZmZmZmZddlmZmZdd lmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-d d l.m/Z/m0Z0m1Z1d d l2m3Z3m4Z4m5Z5d dl6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@GddZAy)N) OrderedDict)Decimal)Enum)ColumnProperty)types) BooleanFieldField FloatField PasswordField TextAreaField) CheckboxInputTextArea) ColorField DateFieldDateIntervalField DateTimeFieldDateTimeIntervalFieldDateTimeLocalField DecimalFieldDecimalIntervalField EmailField IntegerFieldIntIntervalField SelectField StringField TimeField) ColorInput DateInput DateTimeInputDateTimeLocalInput EmailInput NumberInput TextInput TimeInput)AttributeTypeExceptionInvalidAttributeExceptionUnknownTypeException) CountryFieldPhoneNumberField WeekDaysField) choice_type_coerce_factoryClassMapflattenis_date_column is_numberis_number_range is_scalarnull_or_unicode strip_stringtranslated_attributesc!LeZdZdZeej jefej je fej je fej jefej jefej jefej j"efej j&efej j*efej j.e fej j0efej j4efej j8efej j:efej j>efej@efejBefejDe#fejHe%fejLe'fejPe)fejTe+fejXe-fej\e/fej`e1fejde3fejhefejjefejlefejne8ffZ9e:ee;fe#efe?e@feeAfe+eBfeeAfe eAfeeCfeeDfeeEff ZFdZGdZHdZIdZJdZKdZLdZMd ZNd ZOd ZPd ZQd ZRdZSdZTdZUdZVdZWdZXdZYdZZdZ[dZ\dZ]dZ^dZ_dZ`y) FormGeneratorzb Base form generator, you can make your own form generators by inheriting this class. c||_|jjj|_|jj|_|j j |jjjy)z Initializes the form generator :param form_class: ModelForm class to be used as the base of generation process N) form_classMetamodel model_classmetaTYPE_MAPupdatetype_map)selfr9s \/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/wtforms_alchemy/generator.py__init__zFormGenerator.__init__}sV%??//55OO((  T__11::;ct}tj|jjj D]-\}}t |ts|j|r)|||</t|jD]}|j||j<|j||j|S)zL Creates the form. :param form: ModelForm instance )rsainspectr<attrsitems isinstancerskip_column_propertyr5propertykey create_fieldsfilter_attributes)rAformrHrM property_attrs rB create_formzFormGenerator.create_forms   jj)9)9:@@FFH #NCi8((3"E#J  #*$*:*:; ,D"mmE$((O ,!!$(>(>u(EFFrDcx|jjrLtt|j|jjDcgc] \}}|r||f c}}}|S|jj rP|j t|j|jj Dcgc] \}}|r||f c}}|jjr|jjD]} ||=|Scc}}wcc}}w#t$r$|jjr t|YCwxYw)z Filter set of model attributes based on only, exclude and include meta parameters. :param attrs: Set of attributes ) r=onlyrmapvalidate_attributeincluder?excludeKeyError attr_errorsr')rArHrMprops rBrOzFormGenerator.filter_attributess% 99>>&))@)@$))..%Q!T$KE0 !yy   *-T-D-DdiiFWFW)X%Cd yy  99,,ACA!#JA  /$A9900";C"@@1As D 1D 9D  *D98D9c t|j|} t|jts"|jj r t |y ||jfS#t$r[ |jjd}t||}n1#t$r%|jj r t |YYywxYwYwxYw#t$r t|wxYw)z Finds out whether or not given sqlalchemy model attribute name is valid. Returns attribute property if valid. :param attr_name: Attribute name class)NN) getattrr<AttributeError__translatable__r=r[r'rJrLrr&)rA attr_namerRtranslation_classs rBrWz FormGenerator.validate_attributes &4++Y7D 4dmm^<99((3I>>% =$--''# & &$($4$4$E$Eg$N!0)<! &99((3I>>%  & & 4(3 3 4s@A$;C $ C.%BC*C>CCCC C c|jD]A\}}|jd} |j||}t ||r5t|||Cy#t$r|jj sYgwxYw)z Creates fields for given form based on given model attributes. :param form: form to attach the generated fields into :param attributes: model attributes to generate the form fields from rN)rIcolumns create_fieldr(r=skip_unknown_typeshasattrsetattr)rArP propertiesrMr\columnfields rBrNzFormGenerator.create_fieldss$))+ *IC\\!_F ))$74%c5) *( yy33  sA A98A9cX|jry|j|jdS)z Whether or not to skip column property in the generation process. :param column_property: SQLAlchemy ColumnProperty object Tr)_is_polymorphic_discriminator skip_columnre)rAcolumn_propertys rBrKz"FormGenerator.skip_column_propertys,  8 8 7 7 :;;rDc|jjs |jry|jjs |jry|jj s;t |jtjjr |jryt |jtjry|jjr|j|syt |tjsyy)z Whether or not to skip column in the generation process. :param column_property: SQLAlchemy Column object TF)r=include_foreign_keys foreign_keysinclude_primary_keys primary_keyinclude_datetimes_with_defaultrJtyperFrDateTimedefault TSVectorTypeonly_indexed_fields has_indexColumnrArks rBrozFormGenerator.skip_columns yy--&2E2Eyy--&2D2D 886;;(9(9: fkk5#5#5 6 99 ( (1G&")),rDc|js |jry|j}|jD]5}t |j dk(s|j |j vs5yy)z} Whether or not given column has an index. :param column: Column object to inspect the indexes from Tr%F)rurstableindexeslenrename)rArkrindexs rBr|zFormGenerator.has_indexs\   !4!4 ]] E5==!Q&6;;%--+G rDcri}|j|}|j||d<|j|||d<|j||d<|j |j |j ||j |j||j |jjvr2|j |jj|j t|tr/t|jdr|jj|d<|di|}|S)z Create form field for given column. :param prop: SQLAlchemy ColumnProperty object. :param column: SQLAlchemy Column object. ry validatorsfiltersscaleplaces)get_field_classrycreate_validatorsrr?type_agnostic_parametersrMtype_specific_parametersr= field_args issubclassrrhrwr)rAr\rkkwargs field_classrls rBrfzFormGenerator.create_field's**62  LL0y#55dFC| LL0y d33DHHfEF d33F;< 88tyy++ + MM$))..txx8 9 k< 0v{{G,#);;#4#4x %f% rDc|jr5t|jjr|jjS|js|jjSy)zi Return field default for given column. :param column: SQLAlchemy Column object N)ryr2argnullabler=r~s rBryzFormGenerator.default>sH >>i(:(:;>>%% %??yy(((#rDc4|jjdd}|jjdg}t|jtj j r|jjr||dur|jt|S)zc Return filters for given column. :param column: SQLAlchemy Column object trimNrT) infogetrJrwrFrStringr=strip_string_fieldsappendr4)rArk should_trimrs rBrzFormGenerator.filtersJsn kkoofd3 ++//)R0 v{{BHHOO 4 --# D NN< (rDc\t|jtjjs$t|jtj r|j jSt|jtjjr|j jSy)zh Returns date format for given column. :param column: SQLAlchemy Column object N) rJrwrFrrx ArrowTyper=datetime_formatDate date_formatr~s rBrzFormGenerator.date_formatZsm fkk288#4#4 5 KK: 99,, , fkk288== 199(( ( 2rDci}t|jds?|jjds$t |jt j r |j|j||j|}|r||d<t|jdr|jj|d<|j||d<|S)zu Returns type specific parameters for given column. :param column: SQLAlchemy Column object enumschoicesformatregionwidget) rhrwrrrJr ChoiceTyper?select_field_kwargsrrr)rArkrrs rBrz&FormGenerator.type_specific_parametershs  FKK ){{y)&++u'7'78 MM$226: ;&&v. *F8  6;; )%{{11F8 ;;v.x rDc|jjdd}||Si}|jjdd}|||d<nt|jtj j rY|jjC|jjds(|j|jj|d<|r&|j|j|}|di|Sy)zk Returns WTForms widget for given column. :param column: SQLAlchemy Column object rNsteprr) rrrJrwrFrNumericr scale_to_step WIDGET_MAPr)rArkrrr widget_classs rBrzFormGenerator.widgets 40  M{{vt,  !F6N&++rxx'7'78;;$$09S%)%7%7 8I8I%JF6N ??4+?+?+GHL)&) ) rDc>tttd|S)z Returns HTML5 compatible step attribute for given decimal scale. :param scale: an integer that defines a Numeric column's scale z0.1)strpowr)rArs rBrzFormGenerator.scale_to_steps 3wu~u-..rDci}|jjdd|d<|jjd||d<|S)z Returns all type agnostic form field parameters for given column. :param column: SQLAlchemy Column object descriptionlabel)rr)rArMrkrs rBrz&FormGenerator.type_agnostic_parameterss?  &  r B} ++//'37w rDci}|j||d<t|jtjrm|jj }t Jt|tr:t|t r*|Dcgc]}|jt|fc}|d<|S||d<|Sd|jvr#|jdr|jd|d<|S|jjDcgc]}||fc}|d<|Scc}wcc}w)z Returns key value args for SelectField based on SQLAlchemy column definitions. :param column: SQLAlchemy Column object coercer) rrJrwrrrrrvaluerrr)rArkrrchoiceenums rBrz!FormGenerator.select_field_kwargss;;v.x fkk5#3#3 4kk))G w-w-OV$WVfllCK%@$Wy! %,y!  &++ %&++i*@ & I 6F9  ;A++:K:K L$$ LF9  %X !Ms ;D1 D cNd|jvr|jdSt|jtjrt |jS |jj }|jrt|trtS|S#t$r tcYSwxYw)zk Returns coerce callable for given column :param column: SQLAlchemy Column object r) rrJrwrrr, python_typeNotImplementedErrorr3rrr)rArkrs rBrzFormGenerator.coerces v{{ ";;x( ( fkk5#3#3 4-fkk: : # ++11K ??z+s;" " # #" " #sBB$#B$cH|j||j||j|j||j |g}t |j tjr |j|jdt |j tjr |j|jdt|Dcgc]}|| c}}|j|j|j||Scc}w)zf Returns validators for given column :param column: SQLAlchemy Column object emailurl)required_validatorlength_validatorunique_validatorrMrange_validatorrJrwr EmailTyper get_validatorURLTyper.extendadditional_validators)rAr\rkrvs rBrzFormGenerator.create_validatorss  # #F +  ! !& )  ! !$((F 3   (  fkk5?? 3   d009 : fkk5== 1   d007 8EAq}aEF $44TXXvFGFs D'Dc|jjs>|js2|js&|jj} ||j S|jdS#t $rt|j tjjr-|j j} ||cYS#t $rYnwxYw|jj|jjcYSYwxYw)z Returns required / optional validator for given column based on column nullability and form configuration. :param column: SQLAlchemy Column object optional)r=all_fields_optionalryrnot_null_validator_type_maprwrZrJrFr TypeDecoratorimplnot_null_validatorr)rArkr@type_s rBrz FormGenerator.required_validators --NNOOyy<|jj r'|jd|jj Syyy)zl Returns length validator for given column :param column: SQLAlchemy Column object length)rN)rJrwrFrrrhrrr~s rBrzFormGenerator.length_validatorDsd v{{BHHOO 4 X. ""%%hFKK4F4F%G G#/ 5rDc d|jvr|jdr|jdSd|jvr|jdrtS|j|jvrEt |jt j jr|jj}n |j} |j|}tj|rt|tr|S||S#t$r t|wxYw)z Returns WTForms field class. Class is based on a custom field class attribute or SQLAlchemy column type. :param column: SQLAlchemy Column object form_field_classr)rrrwr>rJrFrrrrGisclassrr rZr()rArk check_type column_types rBrzFormGenerator.get_field_classQs  ,=O1P;;12 2  # I(>  ;;dmm + KK//1  ))JJ /-- 3K{+ ;0N"""6** /&v. . /s:5C80C88D N)a__name__ __module__ __qualname____doc__r-rFr UnicodeTextr BigIntegerr SmallIntegerTextBooleanrrrrxrrrFloatr IntegerrrUnicoderrrrJSONrr ColorTyper CountryTyper) DateRangeTyperDateTimeRangeTyperrr IntRangeTyperNumericRangeTyper PasswordTyper PhoneNumberTyper*ScalarListTyperUUIDType WeekDaysTyper+r>rr rrrrr r"r!rr$r#rrCrSrOrWrNrKror|rfryrrrrrrrrrrrrrrrrrrDrBr7r7@s XX ! != 1 XX , / XX " "L 1 XX]]M * XX  | , XX]]I & XX   . XX]]K ( XX^^Z ( XX  | , XX  | , XX  { + XX__k * XX]]I & XX]]M * __m ,   { + __j )    -  "3 4  $ $&; < __j )   !1 2  # #%9 :    /  " "$4 5  ! !; / ]]K ( ^^[ )    /= !HF = )  $  " M * !3 4 ; '  $  % ; ' H %  " ) $ J" <G& D(8*( <@ . ) )0*0/ 2&*.6* L$ H/rDr7)BrG collectionsrdecimalrrr sqlalchemyrFsqlalchemy.orm.propertiesrsqlalchemy_utilsrwtformsrr r r r wtforms.widgetsr rwtforms_componentsrrrrrrrrrrrrrrwtforms_components.widgetsrrrr r!r"r#r$excr&r'r(fieldsr)r*r+utilsr,r-r.r/r0r1r2r3r4r5r7rrDrBrsz#4"4    BA   k/k/rD