恕f4dZddlZddlmZmZddlmZ ddlmZddl m Z dd l mZmZmZdd lmZdd lmZmZmZdd lmZmZdd lmZ ddlmZdZGddeZGddeZ Gdde Z!GddeZ"Gdde#Z$GddeZ%GddeZ&Gdd eZ'd!Z(d"Z)d#Z*y#e $r ddl mZYwxYw#e $rdZY|wxYw)$z5 Useful form fields for use with SQLAlchemy ORM. N)SelectFieldBase StringField)ValidationError) _unset_value) unset_value)get_primary_key) text_type string_types iteritems)CheckboxListInput)FormOptsBaseForm Select2Widget)InlineFieldListInlineModelFormField) lazy_gettext) identity_keyTFcneZdZdZeZ d fd ZdZdZe eeZ dZ dZ dZ dZxZS) QuerySelectFielda Will display a select drop-down field to choose between ORM results in a sqlalchemy `Query`. The `data` property actually will store/keep an ORM model instance, not the ID. Submitting a choice which is not in the query will result in a validation error. This field only works for queries on models whose primary key column(s) have a consistent string representation. This means it mostly only works for those composed of string, unicode, and integer types. For the most part, the primary keys will be auto-detected from the model, alternately pass a one-argument callable to `get_pk` which can return a unique comparable key. The `query` property on the field can be set from within a view to assign a query per-instance to the field. If the property is not set, the `query_factory` callable passed to the field constructor will be called to obtain a query. Specify `get_label` to customize the label associated with each option. If a string, this is the name of an attribute on the model object to use as the label text. If a one-argument callable, this callable will be passed model instance and expected to return the label text. Otherwise, the model object's `__str__` or `__unicode__` will be used. If `allow_blank` is set to `True`, then a blank choice will be added to the top of the list. Selecting this choice will result in the `data` property being `None`. The label for this blank choice can be set by specifying the `blank_text` parameter. c <tt| ||fi|||_|ts t dt |_n||_| d|_n2t|trtj||_n||_||_ ||_d|_d|_y)Nz;The sqlalchemy identity_key function could not be imported.c|SN)xs ^/var/www/feuerwehr-webapp/venv/lib/python3.12/site-packages/flask_admin/contrib/sqla/fields.pyz+QuerySelectField.__init__..Jsq)superr__init__ query_factoryhas_identity_key Exceptionget_pk_from_identityget_pk get_label isinstancer operator attrgetter allow_blank blank_textquery _object_list) selflabel validatorsr!r%r&r*r+kwargs __class__s rr zQuerySelectField.__init__<s .ujKFK* ># ^__.DK DK  (DN  < 0%00;DN&DN&$  rc|jE|jD]2\}}||jk(s|j||jS|jSr) _formdata_get_object_list _set_data_datar.pkobjs r _get_datazQuerySelectField._get_dataUsY >> %002 C'NN3'zz  zzrc ||_d|_yrr7r4r.datas rr6zQuerySelectField._set_data] rc|jP|jxs|j}|j}|Dcgc]}t |||fc}|_|jScc}wr)r-r,r!r%r )r.r,r%r:s rr5z!QuerySelectField._get_object_listcsa    $JJ6$"4"4"6E[[FJO P3)F3K"8#!> PD    !QsA)c#K|jrd|j|jduf|jD](\}}||j |||jk(f*yw)N__None)r*r+r?r5r&r8s r iter_choiceszQuerySelectField.iter_choicesjsd   dootyyD/@A A,,. >GBt~~c*C499,<= = >sA&A(cd|r.|jr|ddk(rd|_yd|_|d|_yy)NrrC)r*r?r7r4r. valuelists rprocess_formdataz!QuerySelectField.process_formdataqs7 IaLI$=  ! !*1 rc|jr |jC|jD]\}}|j|k(syt|j dy)NNot a valid choice)r*r?r5rgettext)r.formr9r:s r pre_validatezQuerySelectField.pre_validateys]499#8002 KC99# K&dll3H&IJJ $9r)NNNNNF)__name__ __module__ __qualname____doc__rwidgetr r;r6propertyr?r5rDrHrM __classcell__r2s@rrrsL:_FBF:?!2 Iy )D!>.KrrcfeZdZdZedZd fd ZdZdZe eeZ dZ dZ d Z xZS) QuerySelectMultipleFieldal Very similar to QuerySelectField with the difference that this will display a multiple select. The data property will hold a list with ORM model instances and will be an empty list when no value is selected. If any of the items in the data list or submitted form data cannot be found in the query, this will result in a validation error. T)multiplec J|g}tt| ||fd|i|d|_y)NdefaultF)rrXr _invalid_formdata)r.r/r0r[r1r2s rr z!QuerySelectMultipleField.__init__s3 ?G &6ujdRYd]cd!&rc|j}|_g}|jD]0\}}|sn)||vs|j||j|2|rd|_|j ||j S)NT)r4r5removeappendr\r6r7)r.formdatar?r9r:s rr;z"QuerySelectMultipleField._get_datasz>>  D002 %C8^OOB'KK$  % )-& NN4 zzrc ||_d|_yrr=r>s rr6z"QuerySelectMultipleField._set_datar@rc#K|jD]'\}}||j|||jvf)ywr)r5r&r?r8s rrDz%QuerySelectMultipleField.iter_choicessA,,. >GBt~~c*C499,<= = >s<>c$t||_yr)setr4rFs rrHz)QuerySelectMultipleField.process_formdatas Yrc|jrt|jd|jrPt d|j D}|jD] }||vst|jdyy)NrJc3&K|] }|d yw)rNr.0rs r z8QuerySelectMultipleField.pre_validate..sBQAaDBs)r\rrKr?listr5)r.rLobj_listvs rrMz%QuerySelectMultipleField.pre_validatesw  ! !!$,,/D"EF F YYB$*?*?*ABBHYY OH$)$,,7L*MNN Or)NNN)rOrPrQrRrrSr r;r6rTr?rDrHrMrUrVs@rrXrXsAD )F'  Iy )D>(OrrXceZdZdZeZy)CheckboxListFielda Alternative field for many-to-many relationships. Can be used instead of `QuerySelectMultipleField`. Appears as the list of checkboxes. Example:: class MyView(ModelView): form_columns = ( 'languages', ) form_args = { 'languages': { 'query_factory': Language.query, }, } form_overrides = { 'languages': CheckboxListField, } N)rOrPrQrRr rSrrrrnrns( FrrncHeZdZdZeedZeedZy) HstoreFormzB Form used in InlineFormField/InlineHstoreList for HSTORE columns KeyValueN)rOrPrQrRrrkeyvaluerrrrprps#L l5) *C  W- .ErrpceZdZdZddZy)KeyValuezk Used by InlineHstoreList to simulate a key and a value field instead of the single HSTORE column. Nc ||_||_yr)rsrt)r.rsrts rr zKeyValue.__init__s r)NN)rOrPrQrRr rrrrvrvs &rrvc,eZdZdZeffd ZdZxZS)InlineHstoreListzA Version of InlineFieldList for use with Postgres HSTORE columns ct|tr't|Dcgc]\}}t||}}}tt |||ycc}}w)z SQLAlchemy returns a dict for HSTORE columns, but WTForms cannot process a dict. This overrides `process` to convert the dict returned by SQLAlchemy to a list of classes before processing. N)r'dictr rvrryprocess)r.r`r?krlr2s rr|zInlineHstoreList.processsH dD !/8?tq!HQN?D? -h=@sAcHttdtfi}i}|jD]i}|j |r|}t |_|j|d|j j||j j<kt|||y)zA Combines each FormField key/value into a dictionary for storage _faker?N) typestrobjectentries should_deletervr? populate_objrtrssetattr)r.r:nameroutput form_fieldfake_objs rrzInlineHstoreList.populate_objsS\F:r2,, @J%%j1 7 (  ''&9,4MM,?,?x}}(()  @ T6"r)rOrPrQrRrr|rrUrVs@rryrysK%0> #rryc4eZdZdZeZ fdZdZdZxZ S)InlineModelFormListz6 Customized inline model form list field. c ||_||_||_||_||_t ||_tt|dd|j}|j||j |}tt|6|fi|y)aa Default constructor. :param form: Form for the related model :param session: SQLAlchemy session :param model: Related model :param prop: Related property name :param inline_view: Inline view form_widget_argsN widget_args form_rules) form_opts)rLsessionmodelprop inline_viewr _pkrgetattr _form_rulesform_field_typerrr ) r.rLrrrrr1rrr2s rr zInlineModelFormList.__init__s    &"5)>PRV)W(3(?(?A ))$I)N  !41*GGrc&|jduSr)r%)r.fields rdisplay_row_controlsz(InlineModelFormList.display_row_controls"s||~T))rct||d}|ytfd|D}jD]}t|}||v}|s2||}j |r=j j |Fj}|j||j|djj|||y)Nc3NK|]}t|j|fywr) get_obj_pkr)rhrlr.s rriz3InlineModelFormList.populate_obj..,s!Cqz!TXX.2Cs"%) rr{r get_field_idrrdeleterr_rr_on_model_change) r.r:rvaluespk_maprfield_id is_createdrs ` rrz InlineModelFormList.populate_obj%sdD) > CFCC\\ HE#E*H!/Jx(%%e,LL''.  e$   ud +    - -eUJ G! Hr) rOrPrQrRrrr rrrUrVs@rrrs%+OH>*Hrrc4eZdZfdZedZdZxZS)InlineModelOneToOneFieldc ||_||_||_||_||_t ||_tt|dd|j}t|,||j fd|i|y)Nrrr) rLrrrrr rrrrrr ) r.rLrrrrr1rr2s rr z!InlineModelOneToOneField.__init__Cst    &"5) -?F"..  txxG9GGrc0|yt|tr|syy)z: Check while installed fields is not null TF)r'r)rs r _looks_emptyz%InlineModelOneToOneField._looks_emptySs = eS !%rct||d}d}d}|sd}|j}t|jjD]G\}}||j k7r|j |||s*|j|jrFd}I|ryt||||jj|j||y)NFT) rrr rL_fieldsrrrr?rron_model_change)r.r field_name inline_modelr form_is_emptyrrs rrz%InlineModelOneToOneField.populate_obj`suj$7   J::.s/aYq\/)rjoin)r:rss rr$r$|s&  $Q 'C 99/3/ //rcvt|trtfd|DStt|S)z: get and format pk from obj :rtype: text_type c3HK|]}tt|ywr)r r)rhr}r:s rrizget_obj_pk..s.s4aYq\4r)r%r'rr )rrs rrrs5 ||~H(E"48444 X r)+rRr(wtforms.fieldsrrwtforms.validatorsrrr ImportError wtforms.utilstoolsr flask_admin._compatr r r flask_admin.contrib.sqla.widgetsr flask_admin.formrrrflask_admin.model.fieldsrrflask_admin.babelrsqlalchemy.orm.utilrr"rrXrnrprrvryrrr$rrrrrrs7.*:#BB>>>J*0 cKcKL4O/4On!0!0// v##4FH/FHR7G37Gt0 ' G *)*s#B5 C5 CCCC