o hE @sXUddlZddlZddlZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZmZddlmZddlmZddlmZmZdd lm Z dd lm!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(ej)d krzdd lm*Z*ej)dkrddlmZedddZ+eZ,e eeee edffZ-e e,eefZ.ej)dkree-eefZ/eeee.fZ0neZ/eZ0e/Z1e0Z2GdddeZ3dede eefdefddZ4dee3de edfddfddZ5i6j7Z8eee9d<d ede e,fd!d"Z:de e e;efd(e ee>fddf d)d*Z?dS)+N) TYPE_CHECKINGAnyClassVarDict ForwardRefGenericIteratorListMappingOptionalTupleTypeTypeVarUnioncast)WeakKeyDictionaryWeakValueDictionary) AnnotatedLiteral)gather_all_validators) DeferredType) BaseModel create_model) JsonWrapper)display_as_typeget_all_type_hintsget_args get_origin typing_base) all_identicallenient_issubclass )_UnionGenericAliasr#)r GenericModelT GenericModel)bound.)r# c@seZdZUdZdZeeed<eree e dfed<de e de e ee e edffde efd d Zede ede e edfdefd d Zed edee efddZdS)r)F __concrete__.__parameters__clsparamsreturnc sdtdtffdd }t||}|dur|Sjr$tjvr$tdt|t s,|f}t ur=t dd|Dr=td t d sKtd j d t|ttj|}t||re|reS|}t}t}d d|D}fdd|D} t\} } ttt t|f| pjft |d|dd| } |t| <| rd} |}t j!| jj"}| | ur|#|| } |d7}| | usj$| _$t ddt%|D}| | _|r|| _| t||<t&|dkr| t||d<t'| | ||| S)aInstantiates a new class from a generic class `cls` and type variables `params`. :param params: Tuple of types the class . Given a generic class `Model` with 2 type variables and a concrete model `Model[str, int]`, the value `(str, int)` would be passed to `params`. :return: New model class inheriting from `cls` with instantiated types described by `params`. If no parameters are given, `cls` is returned as is. _paramsr1cs@t|}t|dkrt|dtrt|d|df}||fS)Nrr)rlen isinstancelisttuple)r2argsr/r,J/var/www/html/ai/venv/lib/python3.10/site-packages/pydantic/v1/generics.py _cache_keyWs z2GenericModel.__class_getitem__.._cache_keyNz?Cannot parameterize a concrete instantiation of a generic modelcss|]}t|tVqdSN)r5r.0paramr,r,r: ez1GenericModel.__class_getitem__..zDType parameters should be placed on typing.Generic, not GenericModelr.zType z< must inherit from typing.Generic before being parameterizedcSs"i|] \}}t|tur||qSr,)rr)r>kvr,r,r: us"z2GenericModel.__class_getitem__..cs*i|]}|jvr|tj|jfqSr,) __fields__r field_info)r>rBr9r,r:rDws*) __module____base__ __config____validators____cls_kwargs___cSsi|]}|dqSr<r,r=r,r,r:rDsrr)(rCacheKey_generic_types_cachegetr-r __bases__ TypeErrorr5r7r)anyhasattr__name__check_parameters_countdictzipr.r keysvalues__concrete_name__rritemsget_caller_frame_inforr rrG__parameterized_bases___assigned_parameterssysmodules__dict__ setdefaultConfigiter_contained_typevarsr4_prepare_model_fields)r/r0r;cached typevars_map model_name validators type_hintsinstance_type_hintsfields model_modulecalled_globally created_modelobject_by_referencereference_namereference_module_globals new_paramsr,r9r:__class_getitem__Ksr           zGenericModel.__class_getitem__cCs*dd|D}d|}|jd|dS)aCompute class name for child classes. :param params: Tuple of types the class . Given a generic class `Model` with 2 type variables and a concrete model `Model[str, int]`, the value `(str, int)` would be passed to `params`. :return: String representing a the new class where `params` are passed to `cls` as type variables. This method can be overridden to achieve a custom naming scheme for GenericModels. cSsg|]}t|qSr,)rr=r,r,r: sz2GenericModel.__concrete_name__..z, [])joinrT)r/r0 param_namesparams_componentr,r,r:rZs zGenericModel.__concrete_name__rgc#sdttdtdtttffdd }jD]6}t|tsqt|dds&qtvrE|tvr/qfdd tD}|||EdHq||EdHqdS) a Returns unbound bases of cls parameterised to given type variables :param typevars_map: Dictionary of type applications for binding subclasses. Given a generic class `Model` with 2 type variables [S, T] and a concrete model `Model[str, int]`, the value `{S: str, T: int}` would be passed to `typevars_map`. :return: an iterator of generic sub classes, parameterised by `typevars_map` and other assigned parameters of `cls` e.g.: ``` class A(GenericModel, Generic[T]): ... class B(A[V], Generic[V]): ... assert A[int] in B.__parameterized_bases__({V: int}) ``` base_model mapped_typesr1c3sBtfdd|jD}||}||us|urdS|VdS)Nc3s|]}|VqdSr<r,r=r|r,r:r@szQGenericModel.__parameterized_bases__..build_base_model..)r7r.rt)r{r|base_parametersparameterized_baser9r}r:build_base_models   z>GenericModel.__parameterized_bases__..build_base_modelr.Ncsi|] \}}|||qSr,)rO)r>keyvalue)rgr,r:rDsz8GenericModel.__parameterized_bases__..) r r)ParametrizationrrP issubclassgetattrr^r[)r/rgrr{r|r,)r/rgr:r]s,     z$GenericModel.__parameterized_bases__N)rTrG __qualname__ __slots__r-rbool__annotations__rr TypeVarTyper r(rrrt classmethodstrrZrrr]r,r,r,r:r)@s 6_( type_type_mapr1c ss|St|}t|}|tur|^}}tt|t|fS|tus,tjdkr2|tur2 ||S|r{tfdd|D}t ||rF|S|durbt |t rbt |t sbt |dddurbt t|j}|dushJtjdkrw|tjurwt||S||S|st|tr|js|j}tfdd|D}t ||r|S||St |ttfrtfdd|D}t ||r|S|S|st|trt|j|_|S ||}t |trt|S|S) aReturn type with all occurrences of `type_map` keys recursively replaced with their values. :param type_: Any type, class or generic alias :param type_map: Mapping from `TypeVar` instance to concrete types. :return: New type representing the basic structure of `type_` with all `typevar_map` keys recursively replaced. >>> replace_types(Tuple[str, Union[List[str], float]], {str: int}) Tuple[int, Union[List[int], float]] r&c3|]}t|VqdSr< replace_types)r>argrr,r:r@rAz replace_types..N_namer"c3rr<r)r>trr,r:r@0rAc3rr<r)r>elementrr,r:r@8rA)rrrrr7 ExtLiteralr_ version_inforrOr r5rrtypingrtypes UnionTyper%r!r)r-r.r r6r inner_typerr) rr type_args origin_typeannotated_type annotationsresolved_type_args resolved_listnew_typer,rr:rsT          rr/ parametersc CsPt|}t|j}||kr&||krdnd}td|d|jd|d|dS)NmanyfewzToo z parameters for z ; actual z , expected )r4r.rQrT)r/ractualexpected descriptionr,r,r:rUMs  "rU DictValuesrCccst|tr |VdSt|dr!t|s!t|tr!|jEdHdSt|ttfr6|D] }t |EdHq*dSt |}|D] }t |EdHqsZ D         <"R