a f=ic?@sUddlZddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZddlmZdd lmZmZmZmZmZdd l m!Z!m"Z"m#Z#e d d d Z$eZ%e e%e efZ&e!Z'e!e e eeee edfffe efe(d<e!Z)e!e ee&fe(d<Gdd d eZ*ee eefedddZ+e e*e edfddddZ,i-j.Z/e ee(d<eee%dddZ0e e e1e2fdddZ3e e*e e1efe e1e4fe ee4fddd d!Z5dS)"N) TYPE_CHECKINGAnyClassVarDictGenericIteratorListMappingOptionalTupleTypeTypeVarUnioncast) Annotated)gather_all_validators) DeferredType) BaseModel create_model) JsonWrapper)display_as_typeget_all_type_hintsget_args get_origin typing_base) LimitedDict all_identicallenient_issubclass GenericModelT GenericModel)bound._generic_types_cache_assigned_parametersc@seZdZUdZdZeeed<er6ee e dfed<e e e e ee e edffe edddZee ee e edfedd d Zeeee ed d d ZdS)r F __concrete__.__parameters__)clsparamsreturnc stttttttdffdfdd }t||}|durD|Sjr\tjvr\t dt |t sl|f}t urt dd|Drt d td st d jd t|ttj|}t||r|rވS|}t}t}d d|D}fdd|D} t\} } ttt t|f| p>jft |d|dd| } |t | <| rd} |}t!j"| jj#}| | ur|$|| } |d7}qj%| _%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. .)_paramsr)cs|t|fSN)r)r*r'r$b/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pydantic/generics.py _cache_keyBsz2GenericModel.__class_getitem__.._cache_keyNz?Cannot parameterize a concrete instantiation of a generic modelcss|]}t|tVqdSr+) isinstancer .0paramr$r$r- Lz1GenericModel.__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)r1kvr$r$r- \r4z2GenericModel.__class_getitem__..cs*i|]"}|jvr|tj|jfqSr$) __fields__rZ field_info)r1r5r,r$r-r7^r4) __module____base__Z __config__Z__validators__Z__cls_kwargs___cSsi|] }|dqSr+r$r0r$r$r-r7r4rr))rr r rr"getr%r __bases__ TypeErrorr/tupler anyhasattr__name__check_parameters_countdictzipr&rkeysvalues__concrete_name__rritemsget_caller_frame_inforrr9__parameterized_bases__r#sysmodules__dict__ setdefaultZConfigiter_contained_typevarslen_prepare_model_fields)r'r(r.cached typevars_mapZ model_name validatorsZ type_hintsinstance_type_hintsfieldsZ model_moduleZcalled_globally created_modelZobject_by_referenceZreference_nameZreference_module_globals new_paramsr$r,r-__class_getitem__6sp *            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$)rr0r$r$r- r4z2GenericModel.__concrete_name__..z, [])joinrB)r'r(Z param_namesZparams_componentr$r$r-rHs  zGenericModel.__concrete_name__)rTr)c#sttttttdfdd }jD]r}t|ts:q(q(t|ddsJq(q(tvr|tvr^q(qfddtD}|||EdHq(||EdHq(dS)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_typesr)c3s@tfdd|jD}||}||us2|ur6dS|VdS)Nc3s|]}|VqdSr+r$r0r`r$r-r3r4zQGenericModel.__parameterized_bases__..build_base_model..)r?r&rZ)r_r`Zbase_parametersZparameterized_baser,rar-build_base_models  z>GenericModel.__parameterized_bases__..build_base_modelr&Ncsi|]\}}|||qSr$)r<)r1keyvalue)rTr$r-r7sz8GenericModel.__parameterized_bases__..) r r Parametrizationrr= issubclassgetattrr#rI)r'rTrbr_r`r$)r'rTr-rKs       z$GenericModel.__parameterized_bases__N)rBr9 __qualname__ __slots__r%rbool__annotations__rr TypeVarTyper rrrrZ classmethodstrrHrerrKr$r$r$r-r +s 2[$)type_type_mapr)csbs|St|}t|}|tur>|^}}tt|t|fS|rtfdd|D}t||rf|S|durt|trt|tst|dddurtt |j }|dusJ||S|st |t r|j s|j}tfdd|D}t||r|S||St|ttfr2tfdd|D}t||r.|S|S|sVt |trVt|j|_|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]] c3s|]}t|VqdSr+ replace_types)r1argrpr$r-r3r4z replace_types..N_namec3s|]}t|VqdSr+rq)r1trtr$r-r3r4c3s|]}t|VqdSr+rq)r1elementrtr$r-r3r4)rrrrrr?rr/rrgtypingrurr r%r&rlistrZ inner_typer<)rorpZ type_argsZ origin_typeannotated_type annotationsZresolved_type_argsZ resolved_listr$rtr-rrsJ      rr)r' parametersr)c CsPt|}t|j}||krL||kr&dnd}td|d|jd|d|dS)NmanyfewzToo z parameters for z ; actual z , expected )rQr&r>rB)r'r|actualexpected descriptionr$r$r-rC)s  rC DictValues)r6r)ccst|tr|Vnrt|drs6 @     68G