o pfm@sTdZddlmZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZddlZddlmZmZdd lmZmZd d lmZmZd d lmZd d lmZmZddlmZddl m!Z!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5m6Z6m7Z7m8Z8ddl9m:Z:m;Z;ddlrd dl?m@ZAd dl?mBZBmCZCd dlDmEZEneZFeGZAeGjHZIGdddeJZKedeAfd Gd!d"d"e ZLdWd)d*ZMdXd0d1ZNdYd7d8ZOdZd;d<ZPd[d=d>ZQd\dBdCZRdddDd]dLdMZSGdNdOdOZTd^dQdRZUd^dSdTZVd_dUdVZWdS)`z"Private logic for creating models.) annotationsN)ABCMeta)partial) FunctionType)AnyCallableGeneric)PydanticUndefinedSchemaSerializer)dataclass_transform deprecated)PydanticUndefinedAnnotationPydanticUserError)create_schema_validator)GenericBeforeBaseModelWarningPydanticDeprecatedSince20) ConfigWrapper)DecoratorInfosPydanticDescriptorProxyget_attribute_from_bases)collect_model_fieldsis_valid_field_nameis_valid_privateattr_name)GenerateSchema)PydanticGenericMetadataget_model_typevars_map) MockValSerset_model_mocks)CallbackGetCoreSchemaHandler)generate_pydantic_signature)get_cls_types_namespace is_annotated is_classvarparent_frame_namespace)ClassAttributeSafeGetItemProxy)ValidateCallWrapper)Field) FieldInfoModelPrivateAttr BaseModelcs"eZdZdZd fdd ZZS) _ModelNamespaceDictz{A dictionary subclass that intercepts attribute setting on model classes and warns about overriding of decorators. kstrvobjectreturnNonecsL||d}|r||urt|trtd|d|jjdt||S)N`z"` overrides an existing Pydantic `z ` decorator) get isinstancerwarningswarndecorator_infodecorator_reprsuper __setitem__)selfr/r1existing __class___/home/ertert/spirit/venv/lib/python3.10/site-packages/pydantic/_internal/_model_construction.pyr=2s z_ModelNamespaceDict.__setitem__)r/r0r1r2r3r4)__name__ __module__ __qualname____doc__r= __classcell__rBrBr@rCr.-sr.T)kw_only_defaultfield_specifierscseZdZ   d*d+fdd Zejsd,ddZed-ddZd.fdd Z e d/d d!Z e e d"dd#d0d%d&Zd1fd(d) ZZS)2ModelMetaclassNTcls_namer0basestuple[type[Any], ...] namespacedict[str, Any]__pydantic_generic_metadata__PydanticGenericMetadata | None#__pydantic_reset_parent_namespace__bool_create_model_module str | Nonekwargsrr3typec s|rh||\}} } t|||} | j|d<t|| j| |} | r:t||dur6d,fd d } | |d <nt|d <| |d <i| | |d <tj ||||fi|}ddl m }|j }t |vrs|t ||krstjtdddt|jdd |_|j|jurdnd |_t||_|r||_nlt|diddt|ddprrtfddDstfddD}ddd|D}d|d}d |d!}t |vrdd"d|D|g}|d#|jd$|d%7}t |ddd&|_d|_!| "D] \}}|#||q|rt$t%|_&t|d'd}t'|t(r+t)|}t*||}t+||| || j,rEd(|vrEt-||t.||| d||d)d*d+|jj/"D|_0t||j1di||Stj ||||fi|S)-aMetaclass for creating Pydantic models. Args: cls_name: The name of the class to be created. bases: The base classes of the class to be created. namespace: The attribute dictionary of the class to be created. __pydantic_generic_metadata__: Metadata for generic models. __pydantic_reset_parent_namespace__: Reset parent namespace. _create_model_module: The module of the class to be created, if created by `create_model`. **kwargs: Catch-all for any other keyword arguments. Returns: The new class created by the metaclass. model_configNr>r-_ModelMetaclass__contextrr3r4cst||||dS)zWe need to both initialize private attributes and call the user-defined model_post_init method. N)init_private_attributes)r>rZ)original_model_post_initrBrCwrapped_model_post_initds z7ModelMetaclass.__new__..wrapped_model_post_initmodel_post_init__class_vars____private_attributes__r r,zClasses should inherit from `BaseModel` before generic classes (e.g. `typing.Generic[T]`) for pydantic generics to work properly.) stacklevel__pydantic_base_init__FrQ parametersrB__parameters__c3s|]}|vVqdSNrB.0x)rcrBrC sz)ModelMetaclass.__new__..c3s|] }|vr|VqdSrerBrf)parent_parametersrBrCrisz, cSsg|]}t|qSrB)r0rfrBrBrC sz*ModelMetaclass.__new__..ztyping.Generic[]zJAll parameters must be present on typing.Generic; you should inherit from .cSsg|]}|jqSrB)rDrfrBrBrCrksz- Note: `typing.Generic` must go last: `class (z): ...`))originargsrc__pydantic_parent_namespace____hash__) raise_errorstypes_namespacecreate_model_modulecSsi|]\}}||jqSrB)info)rgr/r1rBrBrC sz*ModelMetaclass.__new__..)r>r-rZrr3r4)2_collect_bases_datar for_model config_dictinspect_namespace ignored_typesget_model_post_initr[r<__new__mainr-__mro__rindexr8r9rgetattr__init____pydantic_custom_init__r^__pydantic_post_init__rbuild__pydantic_decorators__rQr6alltuplejoinrD TypeError__pydantic_complete__items __set_name__build_lenient_weakvaluedictr%rqr7dictunpack_lenient_weakvaluedictr"set_model_fieldsfrozenset_default_hash_funccomplete_model_classcomputed_fieldsmodel_computed_fields__pydantic_init_subclass__)mcsrLrMrOrQrSrUrWbase_field_names class_varsbase_private_attributesconfig_wrapperprivate_attributesr]clsr-mrocombined_parametersparameters_strgeneric_type_label error_message bases_strnameobjparent_namespacertr@)r\rcrjrCr~<s             zModelMetaclass.__new__itemcCs`|jd}|r||vr||S|dkr,t|dd}t|tr,|}|dur,t|dSt|)zNThis is necessary to keep attribute access working for class attribute access.r`__pydantic_core_schema____pydantic_validator__N)__dict__r6rr7rrebuildAttributeError)r>rrmaybe_mock_validatorrebuilt_validatorrBrBrC __getattr__s     zModelMetaclass.__getattr__rpdict[str, object]cOstSre)r.)rrprWrBrBrC __prepare__szModelMetaclass.__prepare__instancecst|do t|S)zsAvoid calling ABC _abc_subclasscheck unless we're pretty sure. See #3829 and python/cpython#92810 r)hasattrr<__instancecheck__)r>rr@rBrCrsz ModelMetaclass.__instancecheck__6tuple[set[str], set[str], dict[str, ModelPrivateAttr]]cCspddlm}t}t}i}|D]"}t||r2||ur2|t|di||j||jq|||fS)Nr r, model_fields) rr-set issubclassupdaterkeysr_r`)rMr- field_namesrrbaserBrBrCrxs    z"ModelMetaclass._collect_bases_dataEThe `__fields__` attribute is deprecated, use `model_fields` instead.)categorydict[str, FieldInfo]cCstdt|jS)Nr)r8r9rrr>rBrBrC __fields__szModelMetaclass.__fields__ list[str]cs$tt}d|vr|d|S)Nr)listr<__dir__remove)r> attributesr@rBrCrs zModelMetaclass.__dir__)NTN)rLr0rMrNrOrPrQrRrSrTrUrVrWrr3rX)rr0r3r)rprrWrr3r)rrr3rT)rMrNr3r)r3r)r3r)rDrErFr~typing TYPE_CHECKINGr classmethodrr staticmethodrxpropertyr rrrHrBrBr@rCrK:s"    rKr>r- __contextrr3r4cCsTt|dddur(i}|jD]\}}|}|tur|||<qt|d|dSdS)a This function is meant to behave like a BaseModel method to initialise private attributes. It takes context as an argument since that's what pydantic-core passes when calling it. Args: self: The BaseModel instance. __context: The context. __pydantic_private__N)rr`r get_defaultr object_setattr)r>rpydantic_privater private_attrdefaultrBrBrCr[s r[rOrPrMrNCallable[..., Any] | NonecCs8d|vr|dSddlm}t|d}||jur|SdS)zaGet the `model_post_init` method from the namespace or the class bases, or `None` if not defined.r^r r,N)rr-rr^)rOrMr-r^rBrBrCr}s   r}r|base_class_varsset[str]base_class_fieldsdict[str, ModelPrivateAttr]c sddlm}mm}|t}i}|di}d|vsd|vr#tdt} t| D]\} } | dkr5q,t | t rJ| j |dkrJ| j |drJq,t | |sU| jj d kr[| | q,t | r| d rmtd | d t| r~td d| d| d | || <|| =q,t | |rt| s| dpd} td| d| d | d rq,t| r| |vst|| s|| d|| <|| =q,| |vrq,| |vr| |vrtd| dddt | |rtd| dddtd| d| d| dddq,| D]M\} }t| rH| |vrH| | vrHt|sH||vrHt|ddd krHt|rCt|^}}tfdd|Dd}|durC||| <q||| <q|S) aIterate over the namespace and: * gather private attributes * check for items which look like fields but are not (e.g. have no annotation) and warn. Args: namespace: The attribute dictionary of the class to be created. ignored_types: A tuple of ignore types. base_class_vars: A set of base class class variables. base_class_fields: A set of base class fields. Returns: A dict contains private attributes info. Raises: TypeError: If there is a `__root__` field in model. NameError: If private attribute name is invalid. PydanticUserError: - If a field does not have a type annotation. - If a field on base class was overridden by a non-annotated attribute. r )r*r+ PrivateAttr__annotations____root__zUTo define root models, use `pydantic.RootModel` rather than a field called '__root__'rYrErF functools__zXPrivate attributes must not use dunder names; use a single underscore prefix instead of rmzJPrivate attributes must not use valid field names; use sunder names, e.g. _z instead of my_fieldz>Fields must not use names with leading underscores; e.g., use )rzField z defined on a base class was overridden by a non-annotated attribute. All field definitions, including overrides, require a type annotation.zmodel-field-overridden)codez requires a type annotationzmodel-field-missing-annotationz)A non-annotated attribute was detected: `z = z3`. All model fields require a type annotation; if `z` is not meant to be a field, you may be able to resolve this error by annotating it as a `ClassVar` or updating `model_config['ignored_types']`.Nc3s|] }t|r|VqdSre)r7)rgr1r+rBrCrisz$inspect_namespace..)fieldsr*r+rdefault_ignored_typesr6rrrrr7rXrErF startswithrAadd NameErrorrlstriprr$rrr#typing_extensionsget_argsnext)rOr|rrr*rall_ignored_typesrraw_annotations ignored_namesvar_namevaluesuggested_nameann_nameann_typermetadatarrBrrCr{&s               r{rtype[BaseModel]cCs@t|d}t|}|dtjhvst|dd|jkr||_dSdS)Nrr__code__)rmake_hash_funcr2rrrr)rrMbase_hash_func new_hash_funcrBrBrCrs   rcs0|jr tj|jnddd fdd }|S) NcSsdS)NrrB)rrBrBrCsz make_hash_func..r>rr3intcs6zt|jWStytt|jYSwre)hashrKeyErrorr'rgetterrBrC hash_funcs  z!make_hash_func..hash_func)r>rr3r)roperator itemgetterr)rrrBrrCrs rrrrtc Cslt|}t|||||d\}}||_|j||D]}|j|d}|dur3|jtur3t |||jqdS)a.Collect and set `cls.model_fields` and `cls.__class_vars__`. Args: cls: BaseModel or dataclass. bases: Parents of the class, generally `cls.__bases__`. config_wrapper: The config wrapper instance. types_namespace: Optional extra namespace to look for types in. ) typevars_mapN) rrrr_rr`poprr setattr) rrMrrtrrrr/rrBrBrCrs  r)rsrurLr0rsrTdict[str, Any] | NonerurVc Cs,t|}t|||}tt|jdd|dd}|jr t||dSz|||} Wn!tyI} z|r3t||d| j dWYd} ~ dSd} ~ ww| |} z| | } Wn|j yft||YdSw| |_ t| ||pq|j|j|rwdnd| |j|_t| | |_d |_td t|j|j|d |_d S) aFinish building a model class. This logic must be called after class has been created since validation functions must be bound and `get_type_hints` requires a class object. Args: cls: BaseModel or dataclass. cls_name: The model or dataclass name. config_wrapper: The config wrapper instance. raise_errors: Whether to raise errors. types_namespace: Optional extra namespace to look for types in. create_model_module: The module of the class to be created, if created by `create_model`. Returns: `True` if the model is successfully completed, else `False`. Raises: PydanticUndefinedAnnotation: If `PydanticUndefinedAnnotation` occurs in`__get_pydantic_core_schema__` and `raise_errors=True`. F)from_dunder_get_core_schemaunpack)ref_moder5N create_modelr-T __signature__)initrr)rrr rgenerate_schema defer_buildr__get_pydantic_core_schema__rr core_config clean_schemaCollectedInvalidrrrErFplugin_settingsrr __pydantic_serializer__rr&r!rrr ) rrLrrsrtrur gen_schemahandlerschemaerrBrBrCrs^      rc@s.eZdZdZd ddZdddZdd d Zd S)_PydanticWeakRefaWrapper for `weakref.ref` that enables `pickle` serialization. Cloudpickle fails to serialize `weakref.ref` objects due to an arcane error related to abstract base classes (`abc.ABC`). This class works around the issue by wrapping `weakref.ref` instead of subclassing it. See https://github.com/pydantic/pydantic/issues/6763 for context. Semantics: - If not pickled, behaves the same as a `weakref.ref`. - If pickled along with the referenced object, the same `weakref.ref` behavior will be maintained between them after unpickling. - If pickled without the referenced object, after unpickling the underlying reference will be cleared (`__call__` will always return `None`). rrcCs"|dur d|_dSt||_dSre)_wrweakrefref)r>rrBrBrCr<s z_PydanticWeakRef.__init__r3cCs|jdurdS|Sre)rrrBrBrC__call__Ds z_PydanticWeakRef.__call__4tuple[Callable, tuple[weakref.ReferenceType | None]]cCs t|ffSre)rrrBrBrC __reduce__Js z_PydanticWeakRef.__reduce__N)rr)r3r)r3r)rDrErFrGrrrrBrBrBrCr+s   rdc CsR|durdSi}|D]\}}zt|}Wn ty!|}Ynw|||<q |S)aXTakes an input dictionary, and produces a new value that (invertibly) replaces the values with weakrefs. We can't just use a WeakValueDictionary because many types (including int, str, etc.) can't be stored as values in a WeakValueDictionary. The `unpack_lenient_weakvaluedict` function can be used to reverse this operation. N)rrr)r resultr/r1proxyrBrBrCrNs   rcCsP|durdSi}|D]\}}t|tr!|}|dur |||<q |||<q |S)zAInverts the transform performed by `build_lenient_weakvaluedict`.N)rr7r)r r!r/r1rBrBrCrbs  rcCsddlm}ttttt|tfS)Nr ComputedFieldInfo)rr$rrrrrr(r#rBrBrCrrs r)r>r-rrr3r4)rOrPrMrNr3r) rOrPr|rNrrrrr3r)rrrMrNr3r4)rrr3r) rrrMrNrrrtrPr3r4)rrrLr0rrrsrTrtrrurVr3rT)r rr3r)r3rN)XrG __future__r _annotationsrrr8rabcrrrtypesrrrrr pydantic_corer r r r errorsrrplugin._schema_validatorrrr_configr _decoratorsrrr_fieldsrrr_generate_schemar _genericsrr _mock_val_serrr_schema_generation_sharedr _signaturer! _typing_extrar"r#r$r%_utilsr&r'_validate_callr(rrr)PydanticModelFieldr*r+rr-DeprecationWarningr2 __setattr__rrr.rKr[r}r{rrrrrrrrrBrBrBrCsf             N  u #U #