恕f4TRddlZddlmZmZmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZddlZddlZddlmZdd lmZdd lmZd d lmZmZd d lmZm Z d dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(m)Z)d dl*m+Z+ejXdZ-e.ej^ej`ejbejdejfejhejjejlejnejpejrejtf Z;Gdde Zz-ModelView._get_model_fields..-sad>S>Sr1)key)r)sortedr _fieldsr(r)s r0_get_model_fieldszModelView._get_model_fields#s) =JJEi .4STTr1cy)Nidr(s r0r&zModelView.scaffold_pk/sr1c|jS)z Return the primary key value from the model instance :param model: Model instance pkrAs r0 get_pk_valuezModelView.get_pk_value3s xxr1cJg}|jD]\}}t|}|tjk(r%t |j tj rI|tj k(r]|js|tjk7s}|j||S)z+ Scaffold list columns ) rBtype mongoengine ListField isinstancefieldEmbeddedDocumentFieldcolumn_display_pk ObjectIdFieldappend)r(columnsr<f field_classs r0scaffold_list_columnszModelView.scaffold_list_columns<s**, "DAqq'K{444qww (I(IJk???%% 8Q8Q)Qq! "r1ci}|jD]E\}}t|tvs|jst|tj k7sA|||<G|S)zK Return a dictionary of sortable columns (name, field) )rBrLSORTABLE_FIELDSrRrMrS)r(rUr<rVs r0scaffold_sortable_columnsz#ModelView.scaffold_sortable_columnsRs[**, #DAqAw/)))T!W 8Q8Q-Q!"GAJ # r1ct|jr|jD]}t|tr%|jjj |}| t dt|}||jvrt dd|zz|jj|t|jS)z! Init search zInvalid search fieldz!Can only search on text columns. zFailed to setup search for "%s") column_searchable_listrOrr)r@get ExceptionrLallowed_search_typesr#rTbool)r(p field_types r0 init_searchzModelView.init_search_s  & &00 .a. **..q1A9#$:;;!!W d&?&??#$G$E$I%JKK##**1- . D''((r1czt|tr&|jjj |}n|}|t d|zd}t|ts|j |j}|s|j |}t|j}|jj|||}|S)z Return filter object(s) for the field :param name: Either field name or field instance Nz#Failed to find field for filter: %s) rOrr)r@r^r_get_column_namer*rL__name__filter_converterconvert)r(r*attr visible_name type_nameflts r0scaffold_filterszModelView.scaffold_filtersvs dL )::%%))$/DD <ADHI I $ -// :L//5LJ'' ##++I,0,8: r1c"t|tS)z Validate if the provided filter is a valid MongoEngine filter :param filter: Filter object )rOr)r(filters r0is_valid_filterzModelView.is_valid_filters&"788r1c t|j|j||j|j|j |j |j}|S)z1 Create form from the model. ) base_classonlyexclude field_args extra_fields)rr)model_form_converterform_base_class form_columnsform_excluded_columns form_argsform_extra_fields)r( form_classs r0 scaffold_formzModelView.scaffold_formsVdjj"77=)-)=)=#'#4#4&*&@&@)-+/+A+A C r1ct|j|j||j|j|}t |j||S)ah Create form for the `index_view` using only the columns from `self.column_editable_list`. :param widget: WTForms widget class. Defaults to `XEditableWidget`. :param validators: `form_args` dict with only validators {'name': {'validators': [required()]}} )rsrtrv)rr)rxrycolumn_editable_listr )r(widget validatorsr~s r0scaffold_list_formzModelView.scaffold_list_formsSdjj"77=)-)=)=#'#<#<)3 5 ))=)=z)/1 1r1c2t|j|||SN)rr))r(r*optss r0_create_ajax_loaderzModelView._create_ajax_loaders!$**dD$??r1c.|jjS)z| Returns the QuerySet for this view. By default, it returns all the objects for the current model. )r)objectsrFs r0 get_queryzModelView.get_querys zz!!!r1c8t|\}}d}|jD]k}t|tjk(rddl}|j d|z}n|}|jd||i} tjdi| } || }g|| z}m|j|S)Nrz.*%s.*__rE) rr#rLrMReferenceFieldrecompiler*Qrp) r(query search_termoptermcriteriarPrregexrmqs r0_searchzModelView._searchs #;/D(( EE{k888 8d?3$zz2.6C $$AA  ||H%%r1c|j}|jr;|D]6\} } } |j| } | j|| j| }8|jr|r|j ||}|j s|jnd} |r|j|rdnd|}n=|j}|r+|Dcgc]\}}|rdnd|}}}|j|}| |j}|r|j|}|r|r|j||z}|r|j}| |fScc}}w)a Get list of objects from MongoEngine :param page: Page number :param sort_column: Sort column :param sort_desc: Sort descending :param search: Search criteria :param filters: List of applied filters :param execute: Run query immediately or not :param page_size: Number of results. Defaults to ModelView's page_size. Can be overriden to change the page_size limit. Removing the page_size limit requires setting page_size to 0 or False. N-)r_filtersapplyclean_search_supportedrsimple_list_pagercountorder_by_get_default_order page_sizelimitskipall)r(page sort_column sort_descsearchfiltersexecuterrrmflt_namevaluerVrordercoldesckeyss r0get_listzModelView.get_listsM,  ==(/ 7$XuMM#&qwwu~6 7  ! !fLL/E&*%;%;  NNIS2-E{#STE++-E+02'S$*.#2"5s;22&-  I KK *E IJJti/0E IIKEe|#2s D>c  |jj|jS#tj$r*}t t dt|dYd}~yd}~wwxYw)zo Return a single model instance by its ID :param id: Model ID rHzFailed to get model. %(error)serrorrN)rrpfirstrMValidationErrorrrr)r(rDexs r0get_onezModelView.get_one$sc >>#**b*1779 9**  ': ,R 02   s-0A- A((A-c n |j}|j||j||d|j|j ||d|S#t $rP}|j |s5ttdt|dtjdYd}~yd}~wwxYw)za Create model helper :param form: Form instance Tz"Failed to create record. %(error)srrzFailed to create record.NF) r) populate_obj_on_model_changesaveafter_model_changer_handle_view_exceptionrrrlog exceptionr(formr)rs r0 create_modelzModelView.create_model3s 7JJLE   e $  ! !$t 4 JJL  # #D% 6  --b1gB$0$46 89 sAA B4$AB//B4c L |j||j||d|j|j||dy#t$rP}|j |s5t tdt|dtjdYd}~yd}~wwxYw)z Update model helper :param form: Form instance :param model: Model instance to update Fz"Failed to update record. %(error)srrzFailed to update record.NT) rrrrr_rrrrrrrs r0 update_modelzModelView.update_modelLs 8   e $  ! !$u 5 JJL  # #D% 7 --b1gB$0$46 89 s4A B#ABB#c " |j||j|j|y#t$rP}|j |s5t t dt|dtjdYd}~yd}~wwxYw)zc Delete model helper :param model: Model instance z"Failed to delete record. %(error)srrzFailed to delete record.NFT) on_model_deletedeleteafter_model_deleter_rrrrrr)r(r)rs r0 delete_modelzModelView.delete_modelfs +   ' LLN  # #E * --b1gB$0$46 89 s!5 BAB  Bz /api/file/ctjjd}tjjd}tjjdd}|r|r|s tdt j t ||}|j|j|}|s tdt|j|jd|jiS)NrDcolldbdefaultizContent-Length) content_typeheaders) rargsr^rgridfsGridFSrobject_id_converterrreadrlength)r(rIrrfsdatas r0 api_file_viewzModelView.api_file_view~s \\  d #||' \\  dI .R #J ]]6":t ,vvd..r23 #J %)%6%6!14;; ?A Ar1cN|dk(r |jsytt||S)NrF) can_deleter$ris_action_allowed)r(r*r/s r0rzModelView.is_action_alloweds& 8 DOOY7==r1rDeletez1Are you sure you want to delete selected records?c  d}|Dcgc]}|j|}}|jj|jD]}||j |z }t t dd||dycc}w#t$r@}|j|s%t tdt|dYd}~yYd}~yd}~wwxYw) Nrz Record was successfully deleted.z,%(count)s records were successfully deleted.)rsuccessz#Failed to delete records. %(error)srr) rrin_bulkvaluesrrr r_rrstr)r(idsrrIall_idsobjrs r0 action_deletezModelView.action_deletes E>ABt//3BGB~~'//8??A 0**3// 0 (=I !&(*3 4 C --b1gC3r7S2 s(BBA!BB C1C  C)NNNNNNNNr)NN)TN)0rg __module__ __qualname____doc__column_filtersrrxrrrrhrcolumn_type_formattersrM StringFieldURLField EmailFieldrr`r3r%r5r7rBr&rJrXr[rdrnrqrrrrrrrrrrrrrrr r __classcell__)r/s@r0rr-sZN*X0"# '( 0(33'00'22'668L\$(GKLP$/L 0 9 U, ).@9 1*@"&4*.=~ 240 LAA&> H  " L MOOr1r)=loggingflaskrrrr flask_adminrflask_admin.babelrr r flask_admin.modelr flask_admin.model.formr flask_admin._compatr rrMrmongoengine.connectionr bson.objectidrflask_admin.actionsrrrrrrrtypefmtrtoolsrhelpersrajaxrrsubdocr getLoggerrsetrIntField FloatField BooleanField DateTimeFieldComplexDateTimeFieldrS DecimalFieldrr UUIDFieldrrZrrEr1r0r s11==+<7 )"&;0'"!=(g+,$$   ~  ~ r1