o 6WjE{@sfdZddlmZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2m3Z3erddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mZ>Gddde0Z?ej@dzie,Gd d!d!ZAd{d&d'ZBe d|d)d*ZCd}d3d4ZDd~dd._PydanticGeneralMetadataz,Pydantic general metadata like `max_digits`.r4r cSs ||_dSN)__dict__)selfr4r(r(r.__init__Hs z@_general_metadata_cls.._PydanticGeneralMetadata.__init__N)r4r )r)r*r+r,r<r(r(r(r._PydanticGeneralMetadataEsr=)annotated_typesr!r')r!r=r(r(r.r6@s r6protected_namespacestuple[str | Pattern[str], ...]ann_namestrbasestuple[type[Any], ...]cls_nameNonec Cs<t}|D]}d}t|tr||du}n t|tr ||}|r|D]%}t||rIt||r8|t|divsIt d|dt||d|dq$g}|D]%} t| tre| |sd| d| j dqN|| ss| d | d qNd d |t |d krd nd} tjd|d|d|d| d tddqdS)NF__pydantic_fields__Field z conflicts with member z of protected namespace .z re.compile()'(z, rz,)z in z$ conflicts with protected namespace zX. You may be able to solve this by setting the 'protected_namespaces' configuration to  stacklevel)r isinstancermatchrB startswithhasattr issubclassgetattr ValueErrorappendpatternjoinlenwarningswarn UserWarning) r?rArCrEr#protected_namespace ns_violationbvalid_namespacespnvalid_namespaces_strr(r(r._check_protected_namespacesNsH       $rdFcls type[Any]fieldsdict[str, FieldInfo] use_inspectr1cCs>t||d}|D]\}}|jdur||vr|||_q dS)Nri)ritems description)rergri fields_docsrA field_infor(r(r._update_fields_from_docstringsys  rotitle_generatorCallable[[str, FieldInfo], str] field_namernr"cCsB|jdur|||}t|tstd|d|j||_dSdS)Nzfield_title_generator  must return str, not )titlerPrB TypeError __class__)rprrrnrtr(r(r.*_apply_field_title_generator_to_field_infos    rwalias_generator%Callable[[str], str] | AliasGeneratorcCs|jdus|jdks|jdus|jdus|jdurd\}}}t|tr,||\}}}nt|rD||}t|tsDt d|d|j |jdusN|jdkrQd|_|jdkret |||_t |||_||_|jdurm||_|jdurxt |||_|jdurt |||_dSdSdS)avApply an alias generator to aliases on a `FieldInfo` instance if appropriate. Args: alias_generator: A callable that takes a string and returns a string, or an `AliasGenerator` instance. field_name: The name of the field from which to generate the alias. field_info: The `FieldInfo` instance to which the alias generator is (maybe) applied. Nr)NNNzalias_generator rs) alias_priorityaliasvalidation_aliasserialization_aliasrPrgenerate_aliasescallablerBrurvr)rxrrrnr{r|r}r(r(r.$_apply_alias_generator_to_field_infos4               rconfig_wrapperrcCs@|jp|j}|durt||||jdurt|j||dSdS)apUpdate the `FieldInfo` instance from the configuration set on the model it belongs to. This will apply the title and alias generators from the configuration. Args: config_wrapper: The configuration from the model. field_name: The field name the `FieldInfo` instance is attached to. field_info: The `FieldInfo` instance to update. N)field_title_generatorrwrxr)rrrrnrr(r(r.update_field_from_configs   r>copydictjson_iter_calculate_keys_copy_and_set_values> schemafrom_ormvalidate construct parse_obj parse_raw _get_value parse_file schema_jsonupdate_forward_refs) typevars_maptype[BaseModel] ns_resolverrrMapping[TypeVar, Any] | None?tuple[dict[str, FieldInfo], PydanticExtraInfo | None, set[str]]c srt}t|j}i}t|D]}t|dd}r||qtj||d} t|} i} t } | D]b\} \}}| dkr@q4t |j | ||j dt|rU| | q4t|| ttur~tfddtDs|tdr~tfd dtDr~tt| sq4|jr| d krtd | d |D]D}t|| r| | vrqt|d id}||urqddt|rt|ndD}| |vrqtjd| d|jd|jdtddqtur | | vs| |vr|j |t!j"d}||_#|sd|_$n{|| %}|r t&|||dd}ni|}nft'|r't(j)r'j)*d|}|_)|j+d<|j,|t!j"d}|_-||_#|s=d|_$n#d|j.vr`|/s`tjd| d| d | d!t0dd"| | q4zt1|| Wn t2yqYnw|j3d#}| |j4vrt5d$| d%|j d&|| | <|j$rt6|| |q4|j7rt8|| d}d'| vr| d'\}}t9||d(}| || fS))aCollect the fields and class variables names of a nascent Pydantic model. The fields collection process is *lenient*, meaning it won't error if string annotations fail to evaluate. If this happens, the original annotation (and assigned value, if any) is stored on the created `FieldInfo` instance. The `rebuild_model_fields()` should be called at a later point (e.g. when rebuilding the model), and will make use of these stored attributes. Args: cls: BaseModel or dataclass. config_wrapper: The config wrapper instance. ns_resolver: Namespace resolver to use when getting model annotations. typevars_map: A dictionary mapping type variables to their concrete types. Returns: A three-tuple containing the model fields, the `PydanticExtraInfo` instance if the `__pydantic_extra__` annotation is set, and class variables names. Raises: NameError: - If there is a conflict between a field name and protected namespaces. - If there is a field other than `root` in `RootModel`. - If a field shadows an attribute in the parent model. rGN)r model_config)r?rArCrEc3s |] }t|duVqdSr9)rU.0 depr_name BaseModel_assigned_valuer(r. &sz'collect_model_fields..__func__c3s*|]}tt|dddjuVqdS)Nr)rUrrrr(r.r*s  rootzUnexpected field with name z4; only 'root' is allowed as a field of a `RootModel`__pydantic_generic_metadata__origincSsh|]}|jqSr(name)rfieldr(r(r. Gsz'collect_model_fields..r(z Field name "z" in "z"" shadows an attribute in parent ""rN_sourceFTrrlenientdefaultfinalz Annotation z= is marked as final and has a default value. Pydantic treats zz as a class variable, but it will be considered as a normal field in V3 to be aligned with dataclasses. If you still want zV to be considered as a class variable, annotate it as: `ClassVar[] = .`)categoryrO__pydantic_decorators__rHz of class zf overrides symbol of same name in a parent class. This override with a computed_field is incompatible.__pydantic_extra__r0r2):rr __bases__reversedrUupdaterget_model_type_hintssafe_get_annotationssetrkrdr?r)is_classvar_annotationaddr any_deprecated_method_namesrS_deprecated_classmethod_namesis_valid_field_name__pydantic_root_model__ NameErrorget dataclasses is_dataclassrgr[r\r+r]from_annotationrCLASS_original_annotation _complete_copy_recreate_field_inforPrr__get___attributes_setfrom_annotated_attribute_original_assignment _qualifiers is_requiredrdelattrAttributeErrorr:computed_fieldsruruse_attribute_docstringsror/)rerrr FieldInfo_rCparent_fields_lookupbase model_fields type_hintscls_annotationsrg class_varsrAann_type evaluatedgeneric_origindataclass_fieldsrnparent_field_infor decoratorspydantic_extra_infoannr2r(rr.collect_model_fieldss                     rMapping[TypeVar, Any]5tuple[dict[str, FieldInfo], PydanticExtraInfo | None]c Csi}||V|jD]\}}|jr|||<q t|||dd}t||||||<q |jdurF|jjsFtt j |jj g|j Rdd}n |j}Wd||fSWd||fS1s`wY||fS)a^Rebuild the (already present) model fields by trying to reevaluate annotations. This function should be called whenever a model with incomplete fields is encountered. Returns: A two-tuple, the first element being the rebuilt fields, the second element being the rebuild `PydanticExtraInfo` instance, if available. Raises: NameError: If one of the annotations failed to evaluate. Note: This function *doesn't* mutate the model fields in place, as it can be called during schema generation, where you don't want to mutate other model's fields. FrNTr) pushrGrkrrr__pydantic_extra_info__r2r/r eval_typer0types_namespace) rerrrrebuilt_fieldsf_namern new_fieldrebuilt_extra_infor(r(r.rebuild_model_fieldss6      rrc Cst}|j}|rt|j|}tj|g|jR\}}ntj|jg|jR}t||}tj|g|jR}d}|j }t urI|j |t j d} n |j||t j d} || _ || _| jdur`| jn|| _|shd| _| S)NTrF)rrlr replace_typesrr try_eval_typerrrr rrrrr) rnrrrr existing_descrrassignrr(r(r.rs< r)rrtype[StandardDataclass]NsResolver | Nonedict[Any, Any] | Nonec Cst}i}|p t}|j}t|jD]}t|sq|||D]\}} t |} || vr2q$|j \} } t | j | | \} }t | rGq$| jsW| jtjurW| jtjurWq$t| j|r| jjrr| jjdurqtd|dddq$|j| | jtjd}| j|_n |j| | tjd}| |_|sd|_| |_|||<t||||jturtt||||rt|||jq$Wdn1swYq|r| D]}|!|q|j"rt#||t$|d d |S) a|Collect the fields of a dataclass. Args: cls: dataclass. config_wrapper: The config wrapper instance. ns_resolver: Namespace resolver to use when getting dataclass annotations. Defaults to an empty instance. typevars_map: A dictionary mapping type variables to their concrete types. Returns: The dataclass fields. FzDataclass field zD has init=False and init_var=True, but these are mutually exclusive.zclashing-init-and-init-var)coderN__is_pydantic_dataclass__rj)%rr__dataclass_fields__r__mro__rrrrkrrrrtyperinitrMISSINGdefault_factoryrPinit_varrrr DATACLASSrrrrr rUsetattrvaluesapply_typevars_maprrorS)rerrrrrgrrrAdataclass_field base_annsglobalnslocalnsrrrnrr(r(r.collect_dataclass_fieldssx                9   r type[PydanticDataclass]c Cst}i}||N|jD]?\}}|jr|||<q|j}tj|jg|j R} t | |} |j | |j tjd} | jdurB| jn|| _t||| | ||<qWd|S1s[wY|S)aRebuild the (already present) dataclass fields by trying to reevaluate annotations. This function should be called whenever a dataclass with incomplete fields is encountered. Raises: NameError: If one of the annotations failed to evaluate. Note: This function *doesn't* mutate the dataclass fields in place, as it can be called during schema generation, where you don't want to mutate other dataclass's fields. rN)rrrGrkrrlrrrrrrrrrrr) rerrrrrrrnrrrr(r(r.rebuild_dataclass_fields{s4      r rcCs |d S)N_rRrr(r(r.rs rcCs|do |d S)Nr __r rr(r(r.is_valid_privateattr_namesrr3Callable[[], Any] | Callable[[dict[str, Any]], Any]'TypeIs[Callable[[dict[str, Any]], Any]]c Cs^zt|}Wn ttfyYdSwt|j}t|dko.t|do.|dj t j uS)zMWhether the provided default factory callable has a validated data parameter.Frr) rsignature_no_evalrVrulist parametersrrZrrrempty)rsigrr(r(r.takes_validated_data_arguments(r)validated_datacall_default_factoryr:Callable[[], Any] | Callable[[dict[str, Any]], Any] | Nonerdict[str, Any] | NonercCs`ddlm}|dur||S|r.t|dr&td|}|dur"td||Std|}|StS)zMResolve the default value using either a static default or a default_factory.r)smart_deepcopyN)rzCallable[[dict[str, Any]], Any]zpThe default factory requires the 'validated_data' argument, which was not provided when calling 'get_default()'.zCallable[[], Any])_utilsrrr rVr )rrrrrfacr(r(r.resolve_default_values    rr()r4r r5r!)r5r8) r?r@rArBrCrDrErBr5rF)F)rerfrgrhrir1r5rF)rprqrrrBrnr")rxryrrrBrnr")rrrrrBrnr"r5rF) rerrrrrrrr5r) rerrrrrrrr5r) rnr"rrrrrr1r5r") rerrrrrrrr5rh) rer rrrrrrr5rh)rrBr5r1)rrr5r) rr rrrrrr1r5r )Tr, __future__r _annotationsrr[collections.abcr functoolsrinspectrrrertypingrr r r r pydantic_corer typing_extensionsrtyping_inspection.introspectionrpydanticrpydantic.errorsraliasesrrr_configr_docs_extractionr _import_utilsrr_internal_dataclassr_namespace_utilsr_reprrrrrr>r!rgr"mainr# _dataclassesr$r% _decoratorsr&r' dataclassr/r7r6rdrorwrrrrrrrr r rrrrr(r(r(r.sr                      +  2 U /0 l .