o ds@szUddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZdd l m!Z!d d l"m#Z#m$Z$m%Z%m&Z&d d l'm(Z(m)Z)m*Z*m+Z+d d l,m-Z-m.Z.d dl/m0Z0m1Z1m2Z2m3Z3d dl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?d dl@mAZAmBZBmCZCd dlDmEZEmFZFd dl mGZGmHZHd dlmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPd dlQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_erddl`maZad dl"mbZbd dl mcZcd dlmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZledddZmdZnedZodeepe8fd dfd!d"Zqd#erd eeegesffd$d%ZteuevewexfZyeed&fezd'<e feyZ{eed&fezd(<d)Z|e!d*e6e7fd+Gd,d-d-eZ}e~jZGd.ddeVe}d/Zd*Z|eddeddd0d1epd2eee(d3dd4epd5eepd6fd7eepefd8ed edfd9d:Zededdd;d1epd2eee(d3eedeedd&ffd4epd5eepd6fd7eepefd8ed edfdeeepd&fd8ed edfd?d:Ze~Z dHd@eedAdBdCdDd edBdEee.ffdFdGZdS)IN)ABCMeta)deepcopy)Enum)partial)Path) FunctionType prepare_class resolve_bases) TYPE_CHECKING AbstractSetAnyCallableClassVarDictListMappingOptionalTupleTypeTypeVarUnioncast no_type_checkoverload)dataclass_transform)ValidatorGroupextract_root_validatorsextract_validatorsinherit_validators) BaseConfigExtrainherit_configprepare_config) ErrorWrapperValidationError) ConfigError DictError ExtraError MissingError)MAPPING_LIKE_SHAPESField FieldInfo ModelFieldModelPrivateAttr PrivateAttr Undefinedis_finalvar_with_default_val)custom_pydantic_encoderpydantic_encoder)Protocol load_fileload_str_bytes)default_ref_template model_schema)PyObjectStrBytes) AnyCallableget_args get_origin is_classvar is_namedtupleis_unionresolve_annotationsupdate_model_forward_refs)DUNDER_ATTRIBUTESROOT_KEYClassAttribute GetterDictRepresentation ValueItemsgenerate_model_signatureis_valid_fieldis_valid_private_namelenient_issubclass sequence_likesmart_deepcopy unique_listvalidate_field_name) Signature)ValidatorListDict) ModelOrDc) AbstractSetIntStrAnyClassMethodCallableGeneratorDictAny DictStrAnyMappingIntStrAnyReprArgsSetStrTupleGeneratorModel BaseModel)bound)r^ create_modelvalidate_model_TfieldsreturncCst|dkr ttddS)Nrz" cannot be mixed with other fields)len ValueErrorrD)rcrg8/usr/local/lib/python3.10/dist-packages/pydantic/main.pyvalidate_custom_root_typebs rifrozencCsdtdtfdd}|r |SdS)Nself_rdcSst|jtt|jSN)hash __class__tuple__dict__values)rkrgrgrh hash_functionhsz-generate_hash_function..hash_function)r int)rjrrrgrgrhgenerate_hash_functiongs rt.ANNOTATED_FIELD_UNTOUCHED_TYPESUNTOUCHED_TYPESFT)kw_only_defaultfield_specifierscs6eZdZefddZdedeffdd ZZS)ModelMetaclassc( sji}t}i}gg}} i} i} |dd} t| tr| hnt| } t} d}t|D]:}trdt|trd|tkrd| t |j t |j |}t|j|}||j7}| |j7} | |j| |j|j}q*dd}ddt|D}fdd|@D}|d }|r|rtd t ||fi|}tt||}t|}|D]}||||j}|r|j || qt!||t"d t#d t$ffd d }|d|dfdkrt%|di|dd}|&D]o\}}t'|r| (|qt)|||t*r| (|qt+|rQt,||||t*}t-t.|r)t/|n|f}||rA|t0krAt1dd|DsAqt2j3||||||d||<q||vr_|j4r_t5| |<qt6|j7|&D]\}}|| vov|| }t|t8rt9|st:d|d|d|d|| |<qi|j4rt9|r|rt5|d| |<qit+|r||vr|rt,||t2j3||||t*|||d}||vrt;|j<||j<r||j<|_||?|j@rtAtB|j@}!ntC}!tD|\}"}#|dur!tE|jF}|| BdhB||dd|&Dp7ddd|&DpBd|jGtH||"d d!d"tH| |#d#d!d"itI|!| i| | | | B|| d$fd%d|&D}$tJjK||||$fi}%tLd&tM|%jN|||%_O|r|%P|&D]\}}&||$vrtQ|&d'd}'tR|'r|'|%|q|%S)(N __slots__rg__resolve_forward_refs__TcSs$h|]}|dr|ds|qS)__) startswithendswith.0keyrgrgrh sz)ModelMetaclass.__new__..csi|]}||qSrg)popr)kwargsrgrh sz*ModelMetaclass.__new__..ConfigzYSpecifying config in two places is ambiguous, use either Config attribute or class kwargsvrdcst|p |jjdkS)Ncython_function_or_method) isinstancern__name__r)untouched_typesrgrh is_untouchedsz,ModelMetaclass.__new__..is_untouched __module__ __qualname__)z pydantic.mainr^__annotations__css|] }tt|tVqdSrl)rLr=r)r allowed_typergrgrh s z)ModelMetaclass.__new__..)namevalue annotationclass_validatorsconfigzPrivate attributes "zF" must not be a valid field name; Use sunder or dunder names, e. g. "_z" or "__z__")defaultz The type of .zo differs from the new default value; if you wish to change the type of this field, please use a type annotationcS&i|]\}}|jjdur||jjqSrl) field_infoexcluderrfieldrgrgrhr cSrrl)rincluderrgrgrhrrcSs|jSrlrrrgrgrh sz(ModelMetaclass.__new__..) name_factorycSs |djS)Nrr)skip_on_failure_and_vrgrgrhrs ) __config__ __fields____exclude_fields____include_fields____validators____pre_root_validators____post_root_validators____schema_cache____json_encoder____custom_root_type____private_attributes__rz__hash____class_vars__csi|] \}}|vr||qSrgrg)rnr)exclude_from_namespacergrhr __signature__ __set_name__)Sr getrstrsetreversed_is_base_model_class_defined issubclassr^updaterNrr"rrrrrrrrrdirkeys TypeErrorrrrq set_configget_validatorsrrpopulate_validatorsr#rur boolrAitemsr>addr1r0rJrPr@r=r<r9anyr-inferunderscore_attrs_are_privater/rvkeep_untouchedr.rK NameErrorrLtype_rDricheck_for_unused json_encodersrr2r3rrtrj validatorsrO staticmethodsuper__new__rErI__init__r__try_update_forward_refs__getattrcallable)(mcsrbases namespacerrcrrpre_root_validatorspost_root_validatorsprivate_attributesbase_private_attributesslots class_vars hash_funcbaseresolve_forward_refsallowed_config_kwargs config_kwargsconfig_from_namespacevgfextra_validatorsr annotationsann_nameann_typer allowed_typesvar_namecan_be_changedinferred_custom_root_type json_encoder pre_rv_new post_rv_new new_namespaceclsobjset_namern)rrrrhr{s6                                       zModelMetaclass.__new__instancerdcst|do t|S)z| Avoid calling ABC _abc_subclasscheck unless we're pretty sure. See #3829 and python/cpython#92810 r)hasattrr__instancecheck__)selfrrrgrhr+sz ModelMetaclass.__instancecheck__) rrrrrr rr __classcell__rgrgrrhryys 0ryc@s6eZdZUeriZeeeefe d<dZ ee e ee fe d<dZee e ee fe d<iZeeeefe d<eeee d<eeeeefe d<eZeeee d<d d Zeee ge fe d <iZed e d <dZeee d<ede d<eeeefe d<eee d<eZeee d<eZdZ dZ!de ddfddZ"e#ddZ$dddZ%dd d!Z&dd"d#Z'dddddddd$d%e e(d&d'e e(d&d(ed)e ed*ed+ed,edd-fd.d/Z)ddddddddd0d1 d%e e(d&d'e e(d&d(ed)e ed*ed+ed,ed2e ee ge fd3ed4e defd5d6Z*e+d7e de fd8d9Z,e+d:ed;d7e dd;fdddd?d:ed;d@e.dAedBedCe/dDedd;fdEdFZ0e+dd>ddd?d:ed;dGe(ee1fdAedBedCe/dDedd;fdHdIZ2e+d:ed;d7e dd;fdJdKZ3e+dd:ed;dLe dMdNe dd;fdOdPZ4dQd;dNd-dRdMdSedd;f dTdUZ5dddddVdQd;d%e e(d&d'e e(d&dWe d-dSedd;f dXdYZ6e+d0e7fd(edZedd-fd[d\Z8e+d0e7d]d(edZed4e defd^d_Z9e+ddadbZ:e+d:ed;dce dd;fdddeZ;e+d:ed;d7e dee+dle ddfdmdnZ?e+dle ddfdodpZ@ddrdsZA       ddied(ed%e e(d&d'e e(d&d*ed+ed,eddqfdtduZB dd%e dvd'e dvd*edWe d-de eCef dwdxZDdye defdzd{ZEdd}d~ZFdS)r^rNrrrrrrcCs|Srlrg)xrgrgrhrAszBaseModel.rrWrFrrQrrr__fields_set__)rprdatardc Ksft|j|\}}}|r |zt|d|Wnty&}ztd|d}~wwt|d||dS)z Create a new model by parsing and validating input data from keyword arguments. Raises ValidationError if the input data cannot be parsed to form a valid model. rpzYModel values must be a dict; you may not have returned a dictionary from a root validatorNr)rarnobject_setattrr_init_private_attributes)__pydantic_self__rrq fields_setvalidation_errorergrgrhrMs  zBaseModel.__init__c s*|jvs tvrt||S|jjtjur(|jvr(td|j j dd|jj r0|jj r:t d|j j d|jvrR|jjrRt d|j j dd|jjri|j|i}|jD]$}z||j |}Wqctt tfy}z tt|tdg|j d}~ww|jd}|r|jj st d|jdfdd |jD}|j|||j d \}}|rt|g|j ||<g} |jD]/\} }| r| rqz||j |}Wqtt tfy}z| t|tdWYd}~qd}~ww| rt| |j t|d |n||j<|jdS) N"z" object has no field "z3" is immutable and does not support item assignmentz " object "z2" field is final and does not support reassignmentlocz8" has allow_mutation set to False and cannot be assignedcsi|] \}}|kr||qSrgrgrkrrrgrhr~rz)BaseModel.__setattr__..r rrp)rrCrrextrar!allowrrfrnrallow_mutationrjrfinalvalidate_assignmentrprAssertionErrorr%r$rDrrrrvalidaterappendrr) rrr new_values validatorexc known_fielddict_without_original_valueerror_errorsskip_on_failurergr rh __setattr__`sV     zBaseModel.__setattr__cs.fddjD}jjdd|DdS)Nc3s |] }|t|tfVqdSrl)rr0)rr rrgrhrsz)BaseModel.__getstate__..cSsi|] \}}|tur||qSrg)r0r rgrgrhrrz*BaseModel.__getstate__..)rpr__private_attribute_values__)rrpr)r private_attrsrgr rh __getstate__s  zBaseModel.__getstate__statecCsJt|d|dt|d|d|diD] \}}t|||qdS)Nrprr!)rrr)rr$rrrgrgrh __setstate__s zBaseModel.__setstate__cCs4|jD]\}}|}|turt|||qdSrl)rr get_defaultr0r)rr private_attrrrgrgrhrs  z"BaseModel._init_private_attributes)rrby_alias skip_defaults exclude_unsetexclude_defaults exclude_noner)rTrYrr(r)r*r+r,rXc Cs>|durt|jjdt|}t|jd||||||dS)z Generate a dictionary representation of the model, optionally specifying which fields to include or exclude. NzF.dict(): "skip_defaults" is deprecated and replaced by "exclude_unset"Tto_dictr(rrr*r+r,)warningswarnrnrDeprecationWarningdict_iter)rrrr(r)r*r+r,rgrgrhr2s" zBaseModel.dictT) rrr(r)r*r+r,encodermodels_as_dictr4r5 dumps_kwargsc Ks~|durt|jjdt|}tttgtf|p|j}t |j | ||||||d} |j r3| t } |j j| fd|i| S)z Generate a JSON representation of the model, `include` and `exclude` arguments as per `dict()`. `encoder` is an optional function to supply as `default` to json.dumps(), other arguments as per `json.dumps()`. NzF.json(): "skip_defaults" is deprecated and replaced by "exclude_unset"r-r)r/r0rnrr1rr r rr2r3rrDr json_dumps) rrrr(r)r*r+r,r4r5r6rrgrgrhjsons*  zBaseModel.jsonrcCsT|jr(t|tr|thkst|tr$|jthkr$|jtjtvr(t|iS|Srl) rrr2rrDr^rshaper*rrrgrgrh_enforce_dict_if_roots zBaseModel._enforce_dict_if_rootrr]c Csz||}t|ts6zt|}Wn%ttfy5}zt|jd|jj}tt|t dg||d}~ww|di|S)Nz expected dict not r rg) r;rr2rrfrrnr%r$rD)rrrrrgrgrh parse_objs   zBaseModel.parse_objutf8) content_typeencodingproto allow_picklebr>r?r@rAc CsZzt||||||jjd}Wntttfy'}z tt|tdg|d}~ww| |S)Nr@r>r?rA json_loadsr ) r6rrDrfrUnicodeDecodeErrorr%r$rDr<)rrBr>r?r@rArrrgrgrh parse_raws   zBaseModel.parse_rawpathcCs"t||||||jjd}||S)NrC)r5rrDr<)rrGr>r?r@rArrgrgrh parse_file(s  zBaseModel.parse_filecCsn|jjstd|jrt|in||}||}t||\}}}|r%|t|d|t|d|| |S)Nz@You must have the config attribute orm_mode=True to use from_ormrpr) rorm_moder&rrD_decompose_classrrarr)rrmrqrrrgrgrhfrom_orm<s   zBaseModel.from_orm _fields_setr[rqcKs||}i}|jD](\}}|jr |j|vr ||j||<q ||vr+||||<q |js4|||<q ||t|d||durJt | }t|d|| |S)a Creates a new model setting __dict__ and __fields_set__ from trusted or pre-validated data. Default values are respected, but no other validation is performed. Behaves as if `Config.extra = 'allow'` was set since it adds all passed values rpNr) rrr alt_aliasaliasrequiredr&rrrrr)rrMrqrK fields_valuesrrrgrgrh constructJs"      zBaseModel.constructrrdeepcCsp|rt|}|j}||}t|d|t|d||jD]}t||t}|tur5|r/t|}t|||q|S)Nrpr)rrnrrrrr0)rrqrrSrrKrrrgrgrh_copy_and_set_valuesbs      zBaseModel._copy_and_set_values)rrrrSrcCsRt|jdd||ddfi|pi}|r|j|B}nt|j}|j|||dS)a Duplicate a model, optionally choose which fields to include, exclude and change. :param include: fields to include in new model :param exclude: fields to exclude from new model, as with values this takes precedence over include :param update: values to change/add in the new model. Note: the data is not validated before creating the new model: you should trust this data :param deep: set to `True` to make a deep copy of the model :return: new model instance F)r.r(rrr*rS)r2r3rrrrT)rrrrrSrqrrgrgrhcopyts zBaseModel.copy ref_templatecCs<|j||f}|dur|St|||d}||j||f<|S)Nr(rW)rrr8)rr(rWcachedsrgrgrhschemas zBaseModel.schemarXcKs.ddlm}|jj|j||dfd|i|S)Nr)r3rXr)r8r3rr7r[)rr(rWr6r3rgrgrh schema_jsons  zBaseModel.schema_jsonrVccs|jVdSrl)r)rrgrgrh__get_validators__s zBaseModel.__get_validators__rc Cst||r6|jj}d}|dvrtdt|rd}|dkr d}n|dkr&d}|dur,|S|j|j|j|dS| |}t|t rG|di|S|jj rP| |Szt |}Wnt tfyh}zt|d}~ww|di|S) N>rSnoneshallowzJ`copy_on_model_validation` should be a string: 'deep', 'shallow' or 'none'Fr_rSTrUrg)rrcopy_on_model_validationr/r0r1rTrprr;r2rIrLrrfr')rrr` deep_copy value_as_dictrrgrgrhrs8     zBaseModel.validatecCst|tr|S|j|Srl)rrFr getter_dictr:rgrgrhrJs  zBaseModel._decompose_classrr.c st|tr#r|j||d} t| vr| tS| S|j||dS|r*t||nd|r3t||ndt|trLfdd|DSt|rqfddt|D} t |j rl|j | S| | St|t rt j ddr|jS|S) N)r(r*r+rrr,)rrc s\i|]*\}}r |sr|r|j|o"|o(|dqS)r.r(r*r+rrr, is_excluded is_included _get_value for_element)rk_v_r(rr+r,r*r. value_exclude value_includergrhrs*      z(BaseModel._get_value..c 3s^|]*\}}r |sr|rj|o!|o'|dVqdS)rdNre)rirkrlrgrhr s,       z'BaseModel._get_value..use_enum_valuesF)rr^r2rDrVrHrrM enumerater?rnrrrr) rrr.r(rrr*r+r,v_dictseq_argsrgrlrhrhs8    zBaseModel._get_valuelocalnscKs t||j|jj|tfdS)z{ Same as update_forward_refs but will not raise exception when forward references are not defined. N)rBrrqrrrrrtrgrgrhr#s z%BaseModel.__try_update_forward_refs__cKst||j|jj|dS)z` Try to update ForwardRefs on fields based on this Model, globalns and localns. N)rBrrqrrrurgrgrhupdate_forward_refs+szBaseModel.update_forward_refsr\ccs|jEdHdS)z( so `dict(model)` works Nrprr rgrgrh__iter__2szBaseModel.__iter__c csz|dus |jdurt|j|}|dus|jdur#tj|j|dd}|j|||d}|durC|sC|sC|sC|sC|sC|jEdHdS|durLt||nd} |durWt||nd} |jD]\\} } |durj| |vsp|rq| durqq^|r|j| } t | ddst | dt | krq^|r| |jvr|j| j }n| }|s| s| r|j | ||| o| | | o| | |||d} || fVq^dS)NT) intersect)rrr*rPrr-)rrHmerger_calculate_keysrprrrr_missingrOrhri)rr.r(rrr*r+r, allowed_keysrmrn field_keyr model_fielddict_keyrgrgrhr38sH      zBaseModel._iterrYcCsx|dur|dur|durdS|r|j}n|j}|dur%||M}|r-||8}|r:|dd|D8}|S)NFcSsh|] \}}t|r|qSrg)rHis_truer rgrgrhrrz,BaseModel._calculate_keys..)rrVrprr)rrrr*rrrgrgrhr{qs    zBaseModel._calculate_keysothercCs&t|tr ||kS||kSrl)rr^r2)rrrgrgrh__eq__s  zBaseModel.__eq__rZcsfddjDS)Ncs8g|]\}}|tvr|jvsj|jjr||fqSrg)rCrrreprr r rgrh s z+BaseModel.__repr_args__..rwr rgr rh __repr_args__s zBaseModel.__repr_args__)rdrW)r$rWrdN)rdNrl)rdrV)rdr\)FFNNFFF)rdrZ)Grrrr rrrrr-rrrrr rrr;rrrr rrrr rrr.r[rrrrz__doc__rrrr#r%rrr2r8 classmethodr;r<r:r4rFrrHrLrRrTrVr7r[r\r]rrFrJrhrrvrxr3r r{rrrgrgrgrhr^7s     7       %       -      &      &    I    >  ) metaclass)r__base__rr__cls_kwargs__ __model_namerrrrrUrfield_definitionscKdSrlrgrrrrrrrrgrgrhr` r`)rrrrcKrrlrgrrgrgrhr`r)rrrrrrzrzc Ksb|dur tdt|dur|durtdt|ts|f}nttdtf}|p*i}i}i} | D]>\} } t | sEtd| dtt| trbz| \} } Wnt ya}ztd|d}~wwd| } } | rm| | | <| || <q3| |d}|r~| || ||rt |t|d <t|}t|||d \}}}||ur||d <| |||||fi|S) a Dynamically create a model. :param __model_name: name of the created model :param __config__: config class to use for the new model :param __base__: base class for the new model to inherit from :param __module__: module of the created model :param __validators__: a dict of method names and @validator class methods :param __cls_kwargs__: a dict for class creation :param __slots__: Deprecated, `__slots__` should not be passed to `create_model` :param field_definitions: fields of the model (or extra fields if a base is supplied) in the format `=(, )` or `=, e.g. `foobar=(str, ...)` or `foobar=123`, or, for complex use-cases, in the format `=` or `=(, )`, e.g. `foo=Field(datetime, default_factory=datetime.utcnow, alias='bar')` or `foo=(str, FieldInfo(title='Foo'))` Nz.__slots__ should not be passed to create_modelzBto avoid confusion __config__ and __base__ cannot be used togetherr]z3fields may not start with an underscore, ignoring "rzfield definitions should either be a tuple of (, ) or just a default value, unfortunately this means tuples as default values are not allowed)rrr)kwds__orig_bases__)r/r0RuntimeWarningr&rrorrr^rrJrfrr"r r r)rrrrrrrzrrcrf_namef_def f_annotationf_valuerrresolved_basesmetansrrgrgrhr`sT          model input_datarXrrSr[c Csi}g}t}t}|j}|jtju}|p|} |jD].} z| | |}WqtttfyH} zitt t | t dg| fWYd} ~ Sd} ~ ww|j D]{\} } || jt}d}|turn|jrn| jrn|| jt}d}|tur| jr|t t| jdqN| }|js| js||| <qNn|| |r||r| jn| j| j||| j| d\}}t|t r||qNt|tr||qN||| <qN|r t|tr| |}n|!|}|r ||O}|jtj"ur|D]}||||<qnt#|D] }|t t$|dq|j%D]4\}} |r|rq z| | |}Wq tttfy@} z|t | t dWYd} ~ q d} ~ ww|rL||t || fS||dfS)z( validate data against a model. r NFTr)&rrrr!ignorerrfrrr%r$rDrrrrOr|allow_population_by_field_namerNrrPrr)r& validate_allvalidate_alwaysrrrlistextendrF extra_keysrrsortedr(r)rrrrqr names_usedrr check_extracls_rrrrr using_namerkerrors_rrrrgrgrhra s|  ,              rarl)r/abcrrVrenumr functoolsrpathlibrtypesrrr typingr r r r rrrrrrrrrrrrtyping_extensionsrrrrrrrr r!r"r#error_wrappersr$r%rr&r'r(r)rcr*r+r,r-r.r/r0r1r8r2r3parser4r5r6r[r7r8r9r:r;r<r=r>r?r@rArButilsrCrDrErFrGrHrIrJrKrLrMrNrOrPinspectrQrRrSrTrUrVrWrXrYrZr[r\r]__all__rbrrirrsrtpropertytyperrrurrvrryobjectrrr^rr`r|rargrgrgrhs     H ( (@    ,  ;g              M