o dD @s:UddlZddlZddlZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZddlmZmZddlmZddlmZddlmZddlmZmZdd lmZdd lmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%ej&d krvdd lm'Z'edddZ(eZ)e e eee edffZ*e e)e efZ+ej&dkree*e efZ,ee ee+fZ-neZ,eZ-e,Z.e-Z/GdddeZ0dede eefdefddZ1de e0de edfddfddZ2i3j4Z5e ee6d<dedee)fdd Z7de e e8e9ffd!d"Z:d#e e0d$e e8efd%e e8e;fd&e ee;fddf d'd(Z._cache_keyNz?Cannot parameterize a concrete instantiation of a generic modelcss|]}t|tVqdSN)r1r .0paramr(r(r6 bz1GenericModel.__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(r6 rs"z2GenericModel.__class_getitem__..cs*i|]}|jvr|tj|jfqSr() __fields__r field_info)r:r>r5r(r6r@ts*) __module____base__ __config____validators____cls_kwargs___cSsi|]}|dqSr8r(r9r(r(r6r@srr)(rCacheKey_generic_types_cachegetr)r __bases__ TypeErrorr1r3r%anyhasattr__name__check_parameters_countdictzipr*rkeysvalues__concrete_name__rritemsget_caller_frame_inforr rrC__parameterized_bases___assigned_parameterssysmodules__dict__ setdefaultConfigiter_contained_typevarsr0_prepare_model_fields)r+r,r7cached typevars_map model_name validators type_hintsinstance_type_hintsfields model_modulecalled_globally created_modelobject_by_referencereference_namereference_module_globals new_paramsr(r5r6__class_getitem__Hsr           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()rr9r(r(r6 sz2GenericModel.__concrete_name__..z, [])joinrP)r+r, param_namesparams_componentr(r(r6rVs zGenericModel.__concrete_name__rcc#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_typesr-c3sBtfdd|jD}||}||us|urdS|VdS)Nc3s|]}|VqdSr8r(r9rxr(r6r<szQGenericModel.__parameterized_bases__..build_base_model..)r3r*rp)rwrxbase_parametersparameterized_baser5ryr6build_base_models   z>GenericModel.__parameterized_bases__..build_base_modelr*Ncsi|] \}}|||qSr()rK)r:keyvalue)rcr(r6r@sz8GenericModel.__parameterized_bases__..) r r%ParametrizationrrL issubclassgetattrrZrW)r+rcr|rwrxr()r+rcr6rYs,     z$GenericModel.__parameterized_bases__N)rPrC __qualname__ __slots__r)rbool__annotations__rr TypeVarTyper r$rrrp classmethodstrrVrrrYr(r(r(r6r%=s 6_( type_type_mapr-csxs|St|}t|}|tur|^}}tt|t|fS|rhtfdd|D}t||r3|S|durOt|trOt|tsOt|dddurOtt |j }|dusUJt j dkrd|t jurdt||S||S|st|tr|js|j}tfdd|D}t||r|S||St|ttfrtfdd|D}t||r|S|S|st|trt|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]] c3|]}t|VqdSr8 replace_types)r:argrr(r6r<r=z replace_types..N_namer c3rr8r)r:trr(r6r<+r=c3rr8r)r:elementrr(r6r<3r=)rrrrr3rr1rrtypingrr[ version_infotypes UnionTyper#rr%r)r*rr2r inner_typerK)rr type_args origin_typeannotated_type annotationsresolved_type_args resolved_listr(rr6rsN       rr+ parametersc CsPt|}t|j}||kr&||krdnd}td|d|jd|d|dS)NmanyfewzToo z parameters for z ; actual z , expected )r0r*rMrP)r+ractualexpected descriptionr(r(r6rQCs  "rQ DictValuesr?ccst|tr |VdSt|dr!t|s!t|tr!|jEdHdSt|ttfr6|D] }t |EdHq*dSt |}|D] }t |EdHqsV @        <"K