o ~j6hQ@sdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZddlmZdd lmZmZmZmZdd lmZdd lmZmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2erddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;me dXd'd(Z?dYdZd1d2Z@dd3d[d._PydanticGeneralMetadataz,Pydantic general metadata like `max_digits`.r.r cSs ||_dS)N)__dict__)selfr.r'r'r-__init__As z@_general_metadata_cls.._PydanticGeneralMetadata.__init__N)r.r )r(r)r*r+r5r'r'r'r-_PydanticGeneralMetadata>sr6)annotated_typesr r&)r r6r'r'r-r09s r0Fcls type[Any]fieldsdict[str, FieldInfo] use_inspectboolNonecCs>t||d}|D]\}}|jdur||vr|||_q dS)Nr<)ritems description)r8r:r< fields_docsann_name field_infor'r'r-_update_fields_from_docstringsGs  rE) typevars_maptype[BaseModel]config_wrapperr ns_resolverNsResolver | NonerFMapping[TypeVar, Any] | None%tuple[dict[str, FieldInfo], set[str]]c Cst}t}|j}i}t|D]}t|dd} r|| qtj||d} |j di} i} t } | D]\}\}}|dkrBq6|j D]}d}t |trV||du}n t |tr`||}|r|D]%}t||rt||rx|t|divstd|dt||d |d qdd }|j D]!}t |tr||s|d |jd f7}q||s||f7}qtd|d|jd|d|d tqEt|r| |q6t||t}t|sq6|jr|dkrtd|dt|di d}|D]>}ddt |rt!|nd D}t||r4||urq||vrq|| vr"qtd|d|j"d|j"dtq|tur]|| vsD||vrV|j#|t$j%d}|sUd|_&||_'nzt(||}nst)||t ||rvt*|j+rv|j+,d||_+|st ||r|-n|}|j.||t$j%d}|sd|_&||_'||_/n#d|j0vr|1stjd|d |d!|d"t2d#d$| |q6zt3||Wn t4yYnw|jd%}||j5vrt6d&|d'|jd(|| |<q6|r| 7D] }|j&r|8|q|j9r t:|| | | fS))a*Collect 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 two-tuple containing model fields 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. __pydantic_fields__N)rI__annotations__ model_configFzField "z" conflicts with member z of protected namespace "z".r'z re.compile()z" in z( has conflict with protected namespace "z_". You may be able to resolve this warning by setting `model_config['protected_namespaces'] = z`.rootzUnexpected field with name z4; only 'root' is allowed as a field of a `RootModel`__pydantic_generic_metadata__origincSsh|]}|jqSr'name).0fieldr'r'r- sz'collect_model_fields..z Field name "z" in "z"" shadows an attribute in parent ""_sourcefinalz 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[] = .`)category stacklevel__pydantic_decorators__zField z of class zf overrides symbol of same name in a parent class. This override with a computed_field is incompatible.);rr __bases__reversedgetattrupdaterget_model_type_hintsr3getsetr@protected_namespaces isinstancer matchstr startswithhasattr issubclass NameErrorpatternwarningswarnr( UserWarningis_classvar_annotationaddris_valid_field_name__pydantic_root_model__ dataclasses is_dataclassr:r*from_annotationrCLASS _complete_original_annotationr#_warn_on_nested_alias_in_annotationrdefault__get___copyfrom_annotated_attribute_original_assignment _qualifiers is_requiredrdelattrAttributeErrorcomputed_fields TypeErrorvaluesapply_typevars_mapuse_attribute_docstringsrE)r8rHrIrFr# FieldInfo_basesparent_fields_lookupbase model_fields type_hintsrr: class_varsrCann_type evaluatedprotected_namespace ns_violationbvalid_namespacespnassigned_valuegeneric_origindataclass_fieldsrDoriginal_assignment decoratorsrWr'r'r-collect_model_fieldsNs                             rrrCrkcCsvt}t|dd}|r7|D]+}tt|r6t|D]}t||r5|jdur5t d|dt dSqq dSdS)N__args__z `alias` specification on field "z5" must be set on outermost annotation to take effect.) rrcr is_annotatedrrget_argsrialiasrqrrrs)rrCr"argsanno_arg anno_type_argr'r'r-r~s  r~rMapping[TypeVar, Any]c Cst}i}||W|jD]H\}}|jr|||<q|j}tj|jg|j R}t ||}|j } t ur@|j|tjd} n |j|| tjd} | jdurQ| jn|| _| ||<qWd|S1sdwY|S)aRebuild the (already present) model fields by trying to reevaluate annotations. This function should be called whenever a model with incomplete fields is encountered. 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. rZN)rpushrMr@r|rAr eval_typer}types_namespacer replace_typesrrrzrr{r) r8rIrFrrebuilt_fieldsf_namerD existing_descannassign new_fieldr'r'r-rebuild_model_fields,s.     r)rIrFrHtype[StandardDataclass]dict[Any, Any] | NoneConfigWrapper | Nonec Cst}i}|p t}|j}t|jD]}t|sq|||D]{\}} ||j divr2q$|j \} } t | j| | \} } t | rGq$| jsW| jtjurW| jtjurWq$t| j|r}| jjrr| jjdurqtd|dddq$|j| | jtjd}n |j| | tjd}|||<|jturtt||||rt|||jq$Wdn1swYq|r|D]}||q|dur|jrt ||t!|d  d |S) a|Collect the fields of a dataclass. Args: cls: dataclass. 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. config_wrapper: The config wrapper instance. Returns: The dataclass fields. rNFzDataclass field zD has init=False and init_var=True, but these are mutually exclusive.zclashing-init-and-init-var)coderZN__is_pydantic_dataclass__r?)"rr__dataclass_fields__rb__mro__rxryrr@r3rfrr try_eval_typetypertinitrMISSINGdefault_factoryriinit_varrrr DATACLASSrrcsetattrrrrrErm)r8rIrFrHrr:rrrCdataclass_fieldglobalnslocalnsr_rDrWr'r'r-collect_dataclass_fieldsWsj             0   rrUcCs |d S)NrrlrTr'r'r-rvs rvcCs|do |d S)Nr__rrTr'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-|djt j uS)zMWhether the provided default factory callable has a validated data parameter.Frr) r ValueErrorrlist parametersrlenrrrempty)rsigrr'r'r-takes_validated_data_arguments (r)r.r r/r )r/r2)F)r8r9r:r;r<r=r/r>) r8rGrHrrIrJrFrKr/rL)rr9rCrkr/r>)r8rGrIrrFrr/r;) r8rrIrJrFrrHrr/r;)rUrkr/r=)rrr/r)Hr+ __future__r _annotationsrxrqcollections.abcrr functoolsrinspectrrrrer typingr r r r pydantic_corertyping_extensionsrrtyping_inspectionrtyping_inspection.introspectionrpydanticrpydantic.errorsrrr_configr_docs_extractionr _import_utilsrr_namespace_utilsr_reprr_utilsrr7r r:r"mainr# _dataclassesr$ _decoratorsr%r&r1r0rErr~rrrvrrr'r'r'r-s\                         O . c