U em @sdZddlmZddlZddlZddlZddlZddlm 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 dlDmEZEn eZFeGZAeGjHZIGdddeJZKedeAfd Gd!d"d"e ZLd#d$d%d&d'd(ZMd)d*d+d,d-d.ZNd)d*d/d/d0d1d2d3ZOd4d*d%d5d6d7ZPd4d$d8d9d:ZQd4d*d;d)d%d<d=d>ZRddd?d4d@d;dAdBdCdAdDdEdFZSGdGdHdHZTdBdBdIdJdKZUdBdBdIdLdMZVd*dNdOdPZWdS)Qz"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*eZdZdZddddfdd ZZS)_ModelNamespaceDictz{A dictionary subclass that intercepts attribute setting on model classes and warns about overriding of decorators. strobjectNone)kvreturncsL||d}|r>||k r>t|tr>td|d|jjdt||S)N`z"` overrides an existing Pydantic `z ` decorator) get isinstancerwarningswarnZdecorator_infoZdecorator_reprsuper __setitem__)selfr2r3existing __class__J/tmp/pip-unpacked-wheel-6q_yhcy8/pydantic/_internal/_model_construction.pyr;2s z_ModelNamespaceDict.__setitem__)__name__ __module__ __qualname____doc__r; __classcell__r@r@r>rAr.-sr.T)Zkw_only_defaultZfield_specifiersc seZdZd%ddddddd d d fd d Zejs@dd dddZed d ddddZd ddfdd Z e dddddZ e e dddddd d!Zd"dfd#d$ ZZS)&ModelMetaclassNTr/tuple[type[Any], ...]dict[str, Any]zPydanticGenericMetadata | Nonebool str | Nonertype)cls_namebases namespace__pydantic_generic_metadata__#__pydantic_reset_parent_namespace___create_model_modulekwargsr4c s|r||\}} } t|||} | j|d<t|| j| |} | r|t||dk rtddddfdd } | |d <nt|d <| |d <| | |d <tj ||||f|}d d l m }|j }t |kr|t ||krtjtdd dt|jdd |_|j|jkr dnd |_t||_|r*||_nt|diddt|ddpNrrtfddDstfddD}ddd|D}d|d}d|d}t |krdd d|D|g}|d!|jd"|d#7}t |ddd$|_d|_!| "D]\}}|#||q|rDt$t%|_&t|d%d}t'|t(rdt)|}t*||}t+||| || j,rd&|krt-||t.||| d||d'd(d)|jj/"D|_0t||j1f||Stj ||||f|SdS)*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-rr1)r<_ModelMetaclass__contextr4cst||||dS)zWe need to both initialize private attributes and call the user-defined model_post_init method. N)init_private_attributes)r<rU)original_model_post_initr@rAwrapped_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.) stacklevelZ__pydantic_base_init__FrP parametersr@__parameters__c3s|]}|kVqdSNr@.0x)r]r@rA sz)ModelMetaclass.__new__..c3s|]}|kr|VqdSr_r@r`)parent_parametersr@rArcsz, cSsg|] }t|qSr@)r/r`r@r@rA sz*ModelMetaclass.__new__..ztyping.Generic[]zJAll parameters must be present on typing.Generic; you should inherit from .cSsg|] }|jqSr@)rBr`r@r@rAresz- Note: `typing.Generic` must go last: `class (z): ...`))originargsr]__pydantic_parent_namespace____hash__) raise_errorstypes_namespacecreate_model_modulecSsi|]\}}||jqSr@)info)rar2r3r@r@rA sz*ModelMetaclass.__new__..)2_collect_bases_datarZ for_modelZ config_dictinspect_namespace ignored_typesget_model_post_initrVr:__new__mainr-__mro__rindexr8r9rgetattr__init__Z__pydantic_custom_init__rYZ__pydantic_post_init__rbuildZ__pydantic_decorators__rPr6alltuplejoinrB TypeError__pydantic_complete__items __set_name__build_lenient_weakvaluedictr%rkr7dictunpack_lenient_weakvaluedictr"set_model_fieldsfrozenset_default_hash_funccomplete_model_classZcomputed_fieldsZmodel_computed_fieldsZ__pydantic_init_subclass__)ZmcsrMrNrOrPrQrRrSZbase_field_names class_varsZbase_private_attributesconfig_wrapperprivate_attributesrXclsr-mroZcombined_parametersZparameters_strZgeneric_type_label error_messageZ bases_strnameobjZparent_namespacernr>)rWr]rdrArv<s      $         zModelMetaclass.__new__)itemr4cCsd|jd}|r ||kr ||S|dkrXt|dd}t|trX|}|dk rXt|dSt|dS)zNThis is necessary to keep attribute access working for class attribute access.r[__pydantic_core_schema____pydantic_validator__N)__dict__r6rzr7rZrebuildAttributeError)r<rrZmaybe_mock_validatorZrebuilt_validatorr@r@rA __getattr__s     zModelMetaclass.__getattr__zdict[str, object])rjrSr4cOstSr_)r.)rrjrSr@r@rA __prepare__szModelMetaclass.__prepare__)instancer4cst|dot|S)zsAvoid calling ABC _abc_subclasscheck unless we're pretty sure. See #3829 and python/cpython#92810 r)hasattrr:__instancecheck__)r<rr>r@rArsz ModelMetaclass.__instancecheck__z6tuple[set[str], set[str], dict[str, ModelPrivateAttr]])rNr4cCspddlm}t}t}i}|D]D}t||r ||k r |t|di||j||jq |||fS)Nr r, model_fields) rwr-set issubclassupdaterzkeysrZr[)rNr- field_namesrrbaser@r@rArrs  z"ModelMetaclass._collect_bases_dataEThe `__fields__` attribute is deprecated, use `model_fields` instead.)categoryzdict[str, FieldInfo]r4cCstdt|jS)Nr)r8r9rrr<r@r@rA __fields__s zModelMetaclass.__fields__z list[str]cs$tt}d|kr |d|S)Nr)listr:__dir__remove)r< attributesr>r@rArs zModelMetaclass.__dir__)NTN)rBrCrDrvtyping TYPE_CHECKINGr classmethodrr staticmethodrrpropertyr rrrFr@r@r>rArG:s " rGr-rr1)r< __contextr4cCsPt|dddkrLi}|jD] \}}|}|tk r|||<qt|d|dS)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. Z__pydantic_private__N)rzr[r get_defaultr object_setattr)r<rZpydantic_privater private_attrdefaultr@r@rArVs  rVrIrHzCallable[..., Any] | None)rOrNr4cCs8d|kr|dSddlm}t|d}||jk r4|SdS)zaGet the `model_post_init` method from the namespace or the class bases, or `None` if not defined.rYr r,N)rwr-rrY)rOrNr-rYr@r@rArus    ruzset[str]zdict[str, ModelPrivateAttr])rOrtbase_class_varsbase_class_fieldsr4c sddlm}mm}|t}i}|di}d|ks>d|krFtdt} t| D]\} } | dkrnqXqXt | t r| j |dkr| j |drqXqXt | |s| jj d kr| | qXqXt | r| d rtd | d n$t| r td d| d| d | || <|| =qXt | |rXt| sX| dp>d} td| d| d qX| d rhqXqXt| r| |kst|| s|| d|| <|| =qX| |krqXqX| |krX| |krtd| dddqXt | |rtd| dddqXtd| d| d| dddqX| D]\} }t| r"| |kr"| | kr"t|s"||kr"t|ddd kr"t|rt|^}}tfdd|Dd}|dk r||| <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__Z__root__zUTo define root models, use `pydantic.RootModel` rather than a field called '__root__'rTrCrD functools__zXPrivate attributes must not use dunder names; use a single underscore prefix instead of rgzJPrivate attributes must not use valid field names; use sunder names, e.g. _z instead of Zmy_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|VqdSr_)r7)rar3r+r@rArcs z$inspect_namespace..)fieldsr*r+rdefault_ignored_typesr6rrrrr7rLrCrD startswithr?add NameErrorrlstriprr$rrzr#typing_extensionsget_argsnext)rOrtrrr*rZall_ignored_typesrZraw_annotations ignored_namesvar_namevalueZsuggested_nameZann_nameZann_typermetadatarr@rrArs&s                 rsztype[BaseModel])rrNr4cCs<t|d}t|}|dtjhks2t|dd|jkr8||_dS)Nrl__code__)rmake_hash_funcr0rlrzr)rrNZbase_hash_funcZ new_hash_funcr@r@rArs  r)rr4cs6|jrtj|jndddddfdd }|S)NcSsdS)Nrr@)rr@r@rAz make_hash_func..rint)r<r4cs<zt|jWStk r6tt|jYSXdSr_)hashrKeyErrorr'rgetterr@rA hash_funcsz!make_hash_func..hash_func)roperator itemgetterr)rrr@rrArs rr)rrNrrnr4c Cslt|}t|||||d\}}||_|j||D]2}|j|d}|dk r4|jtk r4t |||jq4dS)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) rrrrZrr[poprr setattr) rrNrrnrrrr2rr@r@rArs  r)rmror/rJzdict[str, Any] | NonerK)rrMrrmrnror4c Cs0t|}t|||}tt|jdd|dd}|jr@t||dSz|||} WnDtk r} z&|rht||d| j dWYdSd} ~ XYnX| |} z| | } Wn"|j k rt||YdSX| |_ t| ||p|j|j|rdnd| |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)Zfrom_dunder_get_core_schemaunpack)Zref_moder5NZ create_modelr-T __signature__)initrr)rrr rZgenerate_schemaZ defer_buildrZ__get_pydantic_core_schema__rr core_configZ clean_schemaZCollectedInvalidrrrCrDZplugin_settingsrr Z__pydantic_serializer__rr&r!r{rr) rrMrrmrnrorZ gen_schemahandlerZschemaerr@r@rArsX      rc@s:eZdZdZddddZddddZd dd 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`). r)rcCs |dkrd|_n t||_dSr_)_wrweakrefref)r<rr@r@rAr{<sz_PydanticWeakRef.__init__rcCs|jdkrdS|SdSr_)rrr@r@rA__call__Ds z_PydanticWeakRef.__call__z4tuple[Callable, tuple[weakref.ReferenceType | None]]cCs t|ffSr_)rrr@r@rA __reduce__Jsz_PydanticWeakRef.__reduce__N)rBrCrDrEr{rrr@r@r@rAr+sr)dr4c CsT|dkr dSi}|D]6\}}z t|}Wntk rD|}YnX|||<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)rresultr2r3proxyr@r@rArNs   rcCsP|dkr dSi}|D]2\}}t|trB|}|dk rJ|||<q|||<q|S)zAInverts the transform performed by `build_lenient_weakvaluedict`.N)rr7r)rrr2r3r@r@rArbs   rrcCsddlm}ttttt|tfS)Nr ComputedFieldInfo)rrrrrrrr(rr@r@rArrs r)XrE __future__rZ _annotationsrrr8rabcrrrtypesrrrrrZ pydantic_corer r r r errorsrrZplugin._schema_validatorrrr_configrZ _decoratorsrrr_fieldsrrrZ_generate_schemarZ _genericsrrZ _mock_val_serrrZ_schema_generation_sharedr _signaturer!Z _typing_extrar"r#r$r%_utilsr&r'Z_validate_callr(rrr)ZPydanticModelFieldr*r+rwr-DeprecationWarningr0 __setattr__rrr.rGrVrursrrrrrrrrr@r@r@rAsd            N u #U#