U fg@stddlZddlmZddlmZddlmZdZGdddeZGdd d eZ Gd d d e Z Gd d d e Z dS)N)widgets)Field)ValidationError) SelectFieldSelectMultipleField RadioFieldcsVeZdZeZdfdd ZddZddZdd Z d d Z Gd d d e Z Z S)SelectFieldBaseNc s$tj||f||dk r ||_dSN)super__init__ option_widget)selflabel validatorsr kwargs __class__:/tmp/pip-unpacked-wheel-p06mjha7/wtforms/fields/choices.pyr szSelectFieldBase.__init__cCs tdS)z Provides data for choice widget rendering. Must return a sequence or iterable of (value, label, selected, render_kw) tuples. NNotImplementedErrorr rrr iter_choicesszSelectFieldBase.iter_choicescCsdS)NFrrrrr has_groups%szSelectFieldBase.has_groupscCs tdSr rrrrr iter_groups(szSelectFieldBase.iter_groupsc cst|j|j|j|jd|jd}t|D]j\}}t|dkrL|\}}}}n|\}}}i}|j f|d|j |fd||}| d|||_ |Vq*dS)N)widgetrname render_kwZ_formZ_metaz%s-%d)rid) dictr rrrmeta enumeraterlen_Optionrprocesschecked) r optsichoicevaluerr&roptrrr__iter__+s.    zSelectFieldBase.__iter__c@seZdZdZddZdS)zSelectFieldBase._OptionFcCs t|jSr )strdatarrrr_valueEszSelectFieldBase._Option._valueN)__name__ __module__ __qualname__r&r/rrrrr$Bsr$)NNN)r0r1r2rOptionr r rrrr,rr$ __classcell__rrrrrs rcsheZdZeZddeddffdd ZddZddZ d d Z d d Z d dZ ddZ ddZZS)rNTc sXtj||f|||_t|r&|}|dk rHt|tr<|nt||_nd|_||_dSr ) r r coercecallable isinstancer listchoicesvalidate_choice)r rrr5r9r:rrrrr Ls zSelectField.__init__cCs@|js g}n*t|jtr0ttj|j}n|j}||Sr ) r9r7r r8 itertoolschain from_iterablevalues_choices_generator)r r9rrrr_s  zSelectField.iter_choicescCs t|jtSr )r7r9r rrrrriszSelectField.has_groupsccs4t|jtr0|jD]\}}|||fVqdSr )r7r9r itemsr?)r rr9rrrrls zSelectField.iter_groupsccsr|s g}n"t|dttfr"|}n t||}|D]<^}}}|||jk}t|rZ|dni}||||fVq0dSNr)r7r8tuplezipr5r.r#r r9Z_choicesr*rZ other_argsselectedrrrrr?qs zSelectField._choices_generatorc Cs@z|dk r||nd|_Wnttfk r:d|_YnXdSr )r5r. ValueError TypeErrorr r*rrr process_dataszSelectField.process_datac CsT|sdSz||d|_Wn2tk rN}zt|d|W5d}~XYnXdS)Nrz!Invalid Choice: could not coerce.)r5r.rFgettextr Z valuelistexcrrrprocess_formdatas zSelectField.process_formdatacCsR|js dS|jdkr"t|d|D]^}}}}|r*qNq*t|ddS)NChoices cannot be None.zNot a valid choice.)r:r9rGrJrr)r form_matchrrr pre_validates zSelectField.pre_validate)r0r1r2rSelectrr-r rrrr?rIrMrRr4rrrrrIs  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||s g}n"t|dttfr"|}n t||}|D]F^}}}|jdk oR|||jk}t|rd|dni}||||fVq0dSrA)r7r8rBrCr.r5r#rDrrrr?s z&SelectMultipleField._choices_generatorc s@ztfdd|D_Wnttfk r:d_YnXdS)Nc3s|]}|VqdSr r5).0vrrr sz3SelectMultipleField.process_data..)r8r.rFrGrHrrrrIsz SelectMultipleField.process_datac sTztfdd|D_Wn2tk rN}ztd|W5d}~XYnXdS)Nc3s|]}|VqdSr rU)rVxrrrrXsz7SelectMultipleField.process_formdata..z@Invalid choice(s): one or more data inputs could not be coerced.)r8r.rFrJrKrrrrMsz$SelectMultipleField.process_formdatacsjr jsdSjdkr(tdfddDtfddjDrfddtjD}t ddt |t d |d dS) NrNcsg|]}|dqS)rrU)rVr)rrr sz4SelectMultipleField.pre_validate..c3s|]}|kVqdSr rrVr. acceptablerrrXsz3SelectMultipleField.pre_validate..csg|]}|krt|qSr)r-r[r\rrrZsz1'%(value)s' is not a valid choice for this field.z1'%(value)s' are not valid choices for this field.z', ')r*) r:r.r9rGrJranysetrngettextr#r join)r rOZ unacceptabler)r]r rrRs$   z SelectMultipleField.pre_validateN) r0r1r2__doc__rrSrr?rIrMrRrrrrrs   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)Z prefix_labelN) r0r1r2rbrZ ListWidgetrZ RadioInputr rrrrrs r) r;ZwtformsrZwtforms.fields.corerZwtforms.validatorsr__all__rrrrrrrrs   ;U>