h-i!jdZddlmZddlZddlZddlZddlmZddl m Z ddl m Z ddlm Z mZmZmZddlZddlmZmZdd lmZmZd d lmZmZd d lmZd d lmZmZddlmZddl m!Z!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9ddl:m;Z;ejxrddl=m>Z>d dl?m@ZAd dl?mBZBmCZCd dlDmEZEn eZFeGZAeGjZIGddeJZKedeAf Gd!d"eZLd0d#ZMd1d$ZN d2d%ZOd3d&ZP d4d'ZQddd( d5d)ZR d6d*ZSGd+d,ZTd7d-ZUd7d.ZVd8d/ZWy)9z"Private logic for creating models.) annotationsN)ABCMeta)partial) FunctionType)AnyCallableGenericMapping)PydanticUndefinedSchemaSerializer)dataclass_transform deprecated)PydanticUndefinedAnnotationPydanticUserError)create_schema_validator)GenericBeforeBaseModelWarningPydanticDeprecatedSince20) ConfigWrapper)DecoratorInfosPydanticDescriptorProxyget_attribute_from_bases)collect_model_fieldsis_valid_field_nameis_valid_privateattr_name)GenerateSchemagenerate_pydantic_signature)PydanticGenericMetadataget_model_typevars_map) MockValSerset_model_mocks)CallbackGetCoreSchemaHandler)get_cls_types_namespace is_annotated is_classvarparent_frame_namespace)ClassAttribute)ValidateCallWrapper) Signature)Field) FieldInfoModelPrivateAttr BaseModelc$eZdZdZdfd ZxZS)_ModelNamespaceDictz{A dictionary subclass that intercepts attribute setting on model classes and warns about overriding of decorators. c|j|d}|rD||ur@t|tr0tjd|d|j j dt|!||S)N`z"` overrides an existing Pydantic `z ` decorator) get isinstancerwarningswarndecorator_infodecorator_reprsuper __setitem__)selfkvexisting __class__s eD:\DhanshreeandTeamAI\Github\ai_code\venv\Lib\site-packages\pydantic/_internal/_model_construction.pyr;z_ModelNamespaceDict.__setitem__2scD) )jCZ.[ MMAaS B8CZCZCiCiBjjuv ww"1a(()r=strr>objectreturnNone)__name__ __module__ __qualname____doc__r; __classcell__r@s@rAr1r1-s))rBr1T)kw_only_defaultfield_specifiersceZdZ d d fd Zej sd dZed dZd fd Z e ddZ e e deddZxZS)ModelMetaclassc t|r|j|\}} } tj|||} | j|d<t || j | |} | r#t || dfd } | |d<n t|d<| |d<i| | |d<| jr t||t|,||||fi|}ddl m }|j}t|vrG|j!t|j!|kr t#j$t'd d t)|j*d d  |_|j.|j.urdnd|_t3j4||_|r||_nt)|d ij;ddt)|ddxsrrt=fdDst?fdDz}djA|Dcgc] }tC|c}}d|d}d|d}t|vrCdjA|Dcgc]}|jDc}|gz}|d|jDd|dz }tG|ddd|_d |_$| jKD]\}}|jM|||rtOtQ|_)t)|dd}tU|tVr tY|}t[||}t]||| |t_||| d ||t||di||St|,||||fi|Scc}wcc}w)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_configNc0t||||y)zWe need to both initialize private attributes and call the user-defined model_post_init method. N)init_private_attributes)r<_ModelMetaclass__contextoriginal_model_post_inits rAwrapped_model_post_initz7ModelMetaclass.__new__..wrapped_model_post_initds0i@0yArBmodel_post_init__class_vars____private_attributes__rr.zClasses should inherit from `BaseModel` before generic classes (e.g. `typing.Generic[T]`) for pydantic generics to work properly.) stacklevel__pydantic_base_init__F__pydantic_generic_metadata__ parameters__parameters__c3&K|]}|v ywNr_).0xr^s rA z)ModelMetaclass.__new__..s?kYjTUZYjsc3,K|] }|vs| ywrbr_)rcrdparent_parameterss rArez)ModelMetaclass.__new__..sCxz!]^fw]wAzs z, ztyping.Generic[]zJAll parameters must be present on typing.Generic; you should inherit from .z- Note: `typing.Generic` must go last: `class (z): ...`))originargsr^__pydantic_parent_namespace__) raise_errorstypes_namespacecreate_model_module)r<r/rUrrErF)1_collect_bases_datar for_model config_dictinspect_namespace ignored_typesget_model_post_initrTfrozenset_default_hash_funcr:__new__mainr/__mro__r indexr6r7rgetattr__init____pydantic_custom_init__rX__pydantic_post_init__rbuild__pydantic_decorators__r]r4alltuplejoinrCrG TypeError__pydantic_complete__items __set_name__build_lenient_weakvaluedictr'rmr5dictunpack_lenient_weakvaluedictr$set_model_fieldscomplete_model_class__pydantic_init_subclass__)mcscls_namebases namespacer]#__pydantic_reset_parent_namespace___create_model_modulekwargsbase_field_names class_varsbase_private_attributesconfig_wrapperprivate_attributesrWclsr/mrocombined_parametersrdparameters_strgeneric_type_label error_message bases_strnameobjparent_namespacerorVr^rgr@s @@@rAryzModelMetaclass.__new__<s6 DGD[D[\aDb A j*A*44UIvNN(6(B(BIn %!2>77EU" "+>y%+P(+7B4KI/03JI/0*4I& '2c5L2cPb2cI. /$$%i7#(7?3%#]V\#]C (++C#~#))G"4syy7K"K 1B ! 07s||E]_d/e+eC (141D1D HaHa1agxC &*8*>*>s*CC '-4Q1$+C1PRT$U$Y$YZfhj$k!$S*:DAVEV "3C?kYj?kx*x'%)YY@S/T@S1A@S/T%UN+:>:J!)L&44F3GqJ"e+ %)II5.I5aqzz5.IM_L`.`$a %KCLL>YZ[dZeemn $M22#",51 ).C %0557 c  d+834OPfPh4i1&s,KTR *D1#?@P#Q 5c;KLO S% I " /$8   #s 6 @ @J7?3%MfM Mk0U/Js (L0$L5c|jjd}|r ||vr||S|dk(r;t|dd}t|tr|j }| t|dSt |)zNThis is necessary to keep attribute access working for class attribute access.rZ__pydantic_core_schema____pydantic_validator__N)__dict__r4r}r5r!rebuildAttributeError)r<itemrmaybe_mock_validatorrebuilt_validators rA __getattr__zModelMetaclass.__getattr__s!%!2!23K!L !d.@&@)$//11'.t5Mt'T$2J?(<(D(D(F%(4&t-GHH & &rBctSrb)r1)rrlrs rA __prepare__zModelMetaclass.__prepare__s "$$rBc>t|dxrt| |S)zsAvoid calling ABC _abc_subclasscheck unless we're pretty sure. See #3829 and python/cpython#92810 r)hasattrr:__instancecheck__)r<instancer@s rArz ModelMetaclass.__instancecheck__s# x!9:buw?XYa?bbrBc6ddlm}t}t}i}|D]t}t||s||us|j t |dij |j |j|j |jv|||fS)Nrr. model_fields) rzr/set issubclassupdater}keysrYrZ)rr/ field_namesrrbases rArqz"ModelMetaclass._collect_bases_datas$ # "u :<D$ *t9/D""74#D#I#I#KL!!$"5"56"))$*E*EF  J(:::rBEThe `__fields__` attribute is deprecated, use `model_fields` instead.)categorycNtjdt|jS)Nr)r6r7DeprecationWarningrr<s rA __fields__zModelMetaclass.__fields__s  ]_qr   rB)NTN)rrCrtuple[type[Any], ...]rdict[str, Any]r]zPydanticGenericMetadata | Nonerboolr str | NonerrrEtype)rrCrEr)rlrrrrEzMapping[str, object])rrrEr)rrrEz6tuple[set[str], set[str], dict[str, ModelPrivateAttr]])rEdict[str, FieldInfo])rGrHrIrytyping TYPE_CHECKINGr classmethodrr staticmethodrqpropertyrrrrKrLs@rArPrP:sIM48+/ININ%IN" IN (F IN .2 IN)ININ INV    '%%c ; ;OZs!!rBrPct|ddPi}|jjD]#\}}|j}|tus|||<%t |d|yy)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)r}rZr get_defaultr object_setattr)r< __contextpydantic_privater private_attrdefaults rArTrTspt+T2:"&"="="C"C"E D,"..0G//)0 &#F t35EF ;rBcZd|vr|dSddlm}t|d}||jur|Sy)zaGet the `model_post_init` method from the namespace or the class bases, or `None` if not defined.rXrr.N)rzr/rrX)rrr/rXs rArvrvs?I%*++ .u6GHOi7778rBc ^ddlm}mm}|t z}i}|j di}d|vsd|vr t dt} t|jD]\} } | dk(r t| tr1| j|dk(r| jj|drNt| |s| jjd k(r| j!| t| rI| jd rt#d | d t%| rt#d d| zd| d | || <|| =t| |r2t%| s'| j'dxsd} t#d| d| d | jd r+t)| r%| |vst+|| rJ|| || <|| =[| |vra| |vsg| |vrt-d| ddt| |rt-d| ddt-d| d| d| dd|jD]\} }t)| s| |vs| | vst+|r(||vs-t/|ddd k7s>t1|r4t3j4|^}}t7fd|Dd}|||| <}||| <|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__'rRrHrI functools__zXPrivate attributes must not use dunder names; use a single underscore prefix instead of rizJPrivate 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']`.Nc3<K|]}t|s|ywrb)r5)rcr>r-s rArez$inspect_namespace..s$\1JqJZ<[Qs)fieldsr,r-rdefault_ignored_typesr4rrlistrr5rrHrI startswithr@add NameErrorrlstriprr&rr}r%typing_extensionsget_argsnext)rrubase_class_varsbase_class_fieldsr,rall_ignored_typesrraw_annotations ignored_namesvar_namevaluesuggested_nameann_nameann_typermetadatarr-s @rArtrtsG4BA%(=(??68mm$5r:O_$ i(?opp!eM  12% ~ %  ud #  Il$;;""--i.GH  0 1U__5O5OS^5^   h '  / 0""4(BBJQP%X.//2X~.@ XLXY[,1 x ((# y )2Eh2O%__S1?ZN,/|H.hash_funcs-'$uT]]5I5I5K/L*MM MrB)r<rrEint)rrDr)rrbase_hash_funcrs rArxrxsAY-eZ@N$00 N!* *1rBc$t|}t|||||\}}||_|jj ||D]K}|j j |d}|"|jtus5t|||jMy)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) r rrrYrrZpoprr setattr) rrrrorrrr=rs rArrs*#.L-c5./htuFJCj) **..q$7  6G!G CEMM *rB)rnrpc t|}t|||}tt|jd|d}|j r t ||y |j||} |j|} |j| } | |_ t| ||xs |j|j |rdnd| |j"|_t'| | |_d |_t-d t/|j0|j2||_y #t$r(} |rt ||d| jdYd} ~ yd} ~ wwxYw#|j$rt ||YywxYw) 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_moder3N create_modelr/T __signature__)r rr#rgenerate_schema defer_buildr"__get_pydantic_core_schema__rr core_config clean_schemaCollectedInvalidrrrHrIplugin_settingsrr __pydantic_serializer__rr(generate_model_signaturer~rr) rrrrnrorpr gen_schemahandlerschemaer s rArrs|:*#.LJ + **NG !!X&11#w?!,,S1K((0 $*C !8 -s~~ -;&&"C#36;"GC $C'1#,,@P@PR`aC C '  X166(!}5   & &X&s*D:E  E EE E)(E)ct|||S)zGenerate signature for model based on its fields. Args: init: The class init. fields: The model fields. config_wrapper: The config wrapper instance. Returns: The model signature. )r)initrrs rArrs 'tV^ DDrBc(eZdZdZddZddZddZy) _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`). cL|d|_ytj||_yrb)_wrweakrefref)r<rs rAr~z_PydanticWeakRef.__init__1s ;DH{{3'DHrBc<|jy|jSrb)rrs rA__call__z_PydanticWeakRef.__call__9s 88 88: rBct|ffSrb)rrs rA __reduce__z_PydanticWeakRef.__reduce__?s$&**rBN)rr)rEr)rEz4tuple[Callable, tuple[weakref.ReferenceType | None]])rGrHrIrJr~rr!r_rBrArr s ( +rBrc|yi}|jD]\}} t|}|||<|S#t$r|}YwxYw)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)dresultr=r>proxys rArrCs\ y F 1 $Q'Eq  M E s 1 ??c|yi}|jD]*\}}t|tr|}| |||<&|||<,|S)zAInverts the transform performed by `build_lenient_weakvaluedict`.N)rr5r)r#r$r=r>s rArrWsSy F 1 a) *A}q F1I  MrBcPddlm}tttt t |tfS)NrComputedFieldInfo)rr)rrrrrr)r(s rArrgs&*  rB)r<r/rrrErF)rrrrrEzCallable[..., Any] | None) rrrurrset[str]rr*rEzdict[str, ModelPrivateAttr])rrrrrErF) rtype[BaseModel]rrrrrorrErF)rr+rrCrrrnrrodict[str, Any] | NonerprrEr)rzCallable[..., None]rrrrrEr*)r#r,rEr,)rEr)XrJ __future__r _annotationsrr6rabcrrrtypesrrrr r r pydantic_corer r r rerrorsrrplugin._schema_validatorrrr_configr _decoratorsrrr_fieldsrrr_generate_schemarr _genericsrr _mock_val_serr!r"_schema_generation_sharedr# _typing_extrar$r%r&r'_utilsr(_validate_callr)rinspectr*rr+PydanticModelFieldr,r-rzr/rrD __setattr__rrr1rPrTrvrtrxrrrrrrrr_rBrArAs(2 22==C>O"ZZYYIF6Cee"/ !44 3## )$ )T=O