o Df@stddlZddlmZddlmZddlmZdZGdddeZGdd d eZ Gd d d e Z Gd d d e Z dS)N)widgets)Field)ValidationError) SelectFieldSelectMultipleField RadioFieldcsXeZdZeZ dfdd ZddZddZdd Z d d Z Gd d d e Z Z S)SelectFieldBaseNc s,tj||fi||dur||_dSdSN)super__init__ option_widget)selflabel validatorsr kwargs __class__O/home/ubuntu/webapp/venv/lib/python3.10/site-packages/wtforms/fields/choices.pyr s zSelectFieldBase.__init__cCt)z Provides data for choice widget rendering. Must return a sequence or iterable of (value, label, selected, render_kw) tuples. NotImplementedErrorr rrr iter_choicesszSelectFieldBase.iter_choicescCsdS)NFrrrrr has_groups%szSelectFieldBase.has_groupscCrr rrrrr iter_groups(szSelectFieldBase.iter_groupsc cst|j|j|j|jd|jd}t|D]6\}}t|dkr'|\}}}}n|\}}}i}|j d|d|j |fd||}| d|||_ |VqdS)N)widgetrname render_kw_form_metaz%s-%d)ridr) dictr rrrmeta enumeraterlen_Optionr"processchecked) r optsichoicevaluerr)roptrrr__iter__+s2   zSelectFieldBase.__iter__c@seZdZdZddZdS)zSelectFieldBase._OptionFcCs t|jSr )strdatarrrr_valueEs zSelectFieldBase._Option._valueN)__name__ __module__ __qualname__r)r2rrrrr'Bs r')NNN)r3r4r5rOptionr r rrrr/rr' __classcell__rrrrrsrcsheZdZeZddeddffdd ZddZddZ d d Z d d Z d dZ ddZ ddZZS)rNTc s\tj||fi|||_t|r|}|dur&t|tr |nt||_nd|_||_dSr ) r r coercecallable isinstancer#listchoicesvalidate_choice)r rrr8r<r=rrrrr Ls  zSelectField.__init__cCs@|jsg}nt|jtrttj|j}n|j}||Sr ) r<r:r#r; itertoolschain from_iterablevalues_choices_generator)r r<rrrr_s   zSelectField.iter_choicescCs t|jtSr )r:r<r#rrrrris zSelectField.has_groupsccs:t|jtr|jD]\}}|||fVq dSdSr )r:r<r#itemsrB)r rr<rrrrls  zSelectField.iter_groupsccst|sg}nt|dttfr|}nt||}|D]^}}}|||jk}t|r.|dni}||||fVqdSNr)r:r;tuplezipr8r1r&r r<_choicesr-r other_argsselectedrrrrrBqs zSelectField._choices_generatorc Cs>z|dur ||nd|_WdSttfyd|_YdSwr )r8r1 ValueError TypeErrorr r-rrr process_datas  zSelectField.process_datac CsH|sdSz ||d|_WdSty#}zt|d|d}~ww)Nrz!Invalid Choice: could not coerce.)r8r1rKgettextr valuelistexcrrrprocess_formdataszSelectField.process_formdatacCsP|jsdS|jdurt|d|D] ^}}}}|r dSqt|d)NChoices cannot be None.zNot a valid choice.)r=r<rLrOrr)r form_matchrrr pre_validates zSelectField.pre_validate)r3r4r5rSelectrr0r rrrrBrNrSrXr7rrrrrIs  rc@s<eZdZdZejddZddZddZdd Z d d Z d S) rz No different from a normal select field, except this one can take (and validate) multiple choices. You'll need to specify the HTML `size` attribute to the select field when rendering. T)multipleccs~|sg}nt|dttfr|}nt||}|D]#^}}}|jduo*|||jv}t|r3|dni}||||fVqdSrD)r:r;rErFr1r8r&rGrrrrBs z&SelectMultipleField._choices_generatorc s>ztfdd|D_WdSttfyd_YdSw)Nc3|]}|VqdSr r8).0vrrr z3SelectMultipleField.process_data..)r;r1rKrLrMrrrrNs  z SelectMultipleField.process_datac sHztfdd|D_WdSty#}ztd|d}~ww)Nc3r[r r\)r]xrrrr_r`z7SelectMultipleField.process_formdata..z@Invalid choice(s): one or more data inputs could not be coerced.)r;r1rKrOrPrrrrSsz$SelectMultipleField.process_formdatacsjrjsdSjdurtdfddDtfddjDrJfddtjD}t ddt |t d |d dS) NrTcsg|] }|dqS)rr\)r]r,rrr sz4SelectMultipleField.pre_validate..c3s|]}|vVqdSr rr]r1 acceptablerrr_sz3SelectMultipleField.pre_validate..csg|] }|vrt|qSr)r0rcrdrrrbsz1'%(value)s' is not a valid choice for this field.z1'%(value)s' are not valid choices for this field.z', ')r-) r=r1r<rLrOranysetrngettextr&r#join)r rU unacceptabler)rer rrXs&   z SelectMultipleField.pre_validateN) r3r4r5__doc__rrYrrBrNrSrXrrrrrs  rc@s$eZdZdZejddZeZdS)rz Like a SelectField, except displays a list of radio buttons. Iterating the field will produce subfields (each containing a label as well) in order to allow custom rendering of the individual radio fields. F) prefix_labelN) r3r4r5rkr ListWidgetr RadioInputr rrrrrs  r) r>wtformsrwtforms.fields.corerwtforms.validatorsr__all__rrrrrrrrs   ;U>