o dd@sxUddlZddlZddlZddlmZmZmZddlmZddl m Z m Z ddl m Z mZmZmZmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'dd l(m)Z)dd lm*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1dd l2m3Z3erdd l4m5Z5dd l6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?ddlm@Z@mAZAmBZBmCZCmDZDee%ee"ee"eEefe"eEeeffZFdZGdZHeIeJeKeEeLeMeNe*ee eejOeeePjQeRjQhZSe!e#eeTd<eUeVeWeXeYeeehZZe!e#eeTd<deEdefddZ[ddde%eEdeIdeEfdd Z\dedeLfd!d"Z]d#ee#d$d%eEddfd&d'Z^d(ed)e%e#ee"e#ed*fdfdeLfd+d,Z_d-ed)e%e#ee"e#ed*fdfdeLfd.d/Z`deLfd0d1Zad2eEdeLfd3d4Zbe$d5Zcd6eecefd7eecefdeeceffd8d9Zdd6eeefd:eddfd;d<Zed=d>d?eJd@eJdAeJdeLfdBdCZfdDedEdFeeEdGfdHe#dIddJfdKdLZgdMe%e#d$e#dNfde#d$fdOdPZhdQeEdeEfdRdSZidQeEdeEfdTdUZje$dVZkeEdWdXe%eeke"ekd*ffdYeekgeEfdeekfdZd[ZlGd\d]d]eEZmGd^d_d_ZnGd`dadaenZoGdbdcdcenZpGdddedeZqdfdgdhdidjdkdldmdnZrdodpdeEfdqdrZse$dsZtdMetdetfdtduZudveEdeLfdwdxZvhdyZwdveEdeLfdzd{ZxeyZzd|eed}eedeLfd~dZ{dMedeEdefddZ|deeEdeEdeEfddZ}dedeEde"eEe"eEd*fffddZ~de#edeEde"eEe"e"eEd*fd*fffddZdS)N) OrderedDict defaultdictdeque)deepcopy)islice zip_longest)BuiltinFunctionTypeCodeType FunctionType GeneratorType LambdaType ModuleType) TYPE_CHECKING AbstractSetAnyCallable CollectionDict GeneratorIterableIteratorListMappingNoReturnOptionalSetTupleTypeTypeVarUnion) Annotated) ConfigError)NoneType WithArgsTypesall_literal_valuesdisplay_as_typeget_args get_originis_literal_typeis_union) version_info) Signature)Path) BaseConfig) Dataclass) ModelField BaseModel)AbstractSetIntStr DictIntStrAnyIntStrMappingIntStrAnyReprArgs) import_string sequence_likevalidate_field_namelenient_isinstancelenient_issubclass in_ipythonis_valid_identifier deep_updateupdate_not_nonealmost_equal_floats get_modelto_camelis_valid_fieldsmart_deepcopy PyObjectStrRepresentation GetterDict ValueItemsr+ClassAttribute path_typeROOT_KEYget_unique_discriminator_alias"get_discriminator_alias_and_valuesDUNDER_ATTRIBUTES__root__IMMUTABLE_NON_COLLECTIONS_TYPESBUILTIN_COLLECTIONS dotted_pathreturnc Csddlm}z |ddd\}}Wnty(}z td|d|d}~ww||}zt||WStyK}z td |d |d |d}~ww) z Stolen approximately from django. Import a dotted module path and return the attribute/class designated by the last name in the path. Raise ImportError if the import fails. r) import_module .r!"z!" doesn't look like a module pathNzModule "z" does not define a "z " attribute) importlibrUstriprsplit ValueError ImportErrorgetattrAttributeError)rSrU module_path class_nameemodulerd9/usr/local/lib/python3.10/dist-packages/pydantic/utils.pyr8ys  r8P)max_lenvrgcCstdtt|trt||dkr|d|ddSz|}Wnty4|j|}Ynwt||krE|d|dd}|S)z` Truncate a value and add a unicode ellipsis (three dots) to the end if it was too long z:`truncate` is no-longer used by pydantic and is deprecatedNu…r!) warningswarnDeprecationWarning isinstancestrlen__repr__ TypeError __class__)rhrgrdrdretruncates    rtcCst|ttttttfSN)rnlisttupleset frozensetr rrhrdrdrer9sr9basesr2 field_namecCs0|D]}t||drtd|d|dqdS)zZ Ensure that the field's name does not shadow an existing attribute of the model. Nz Field name "zI" shadows a BaseModel attribute; use a different field name with "alias='z'".)r^ NameError)r{r|baserdrdrer:s r:oclass_or_tuple.cCs"zt||WStyYdSwNF)rnrr)rrrdrdrer;s   r;clscCs8z t|to t||WStyt|trYdSwr)rntype issubclassrrr$)rrrdrdrer<s  r<cCs$ztdWdStyYdSw)zU Check whether we're in an ipython environment, including jupyter notebooks. __IPYTHON__FT)evalr}rdrdrdrer=s   r= identifiercCs|o t| S)z Checks that a string is a valid identifier and not a Python keyword. :param identifier: The identifier to test. :return: True if the identifier is valid. ) isidentifierkeyword iskeyword)rrdrdrer>sr>KeyTypemappingupdating_mappingscGsd|}|D])}|D]"\}}||vr*t||tr*t|tr*t|||||<q |||<q q|Sru)copyitemsrndictr?)rrupdated_mappingupdating_mappingkrhrdrdrer?s  r?updatecKs|dd|DdS)NcSsi|] \}}|dur||qSrurd).0rrhrdrdre sz#update_not_none..)rr)rrrdrdrer@sr@g:0yE>)deltavalue_1value_2rcCst|||kS)z4 Return True if two floats are almost equal )abs)rrrrdrdrerAsrAinit).Nfieldsr0configr.r,cCstddlm}m}m}ddlm}||j}i}d} d} t|ddD]} | j | j ur.| } q#| || j <q#| rw|j } | D]9\} }|j}| |vsL||vrMq=t|s]| rZt| rZ| }nd} q=|jsed|jini}|||jfd |ji|||<q=|j|jurd} | r| rd |jfd |j fg}d d |D|krd}n| j }||vr|d7}||vs| j|d||<|t|ddS)z: Generate signature for model based on its fields r) Parameterr, signaturer!)ExtraNFTdefault annotation__pydantic_self__datacSsg|]}|j|jfqSrd)namekind)rprdrdre z,generate_model_signature.. extra_data_r) parametersreturn_annotation)inspectrr,rrrrvaluesrr VAR_KEYWORDrallow_population_by_field_nameraliasr>requiredr KEYWORD_ONLYrextraallowPOSITIONAL_OR_KEYWORDreplacerv)rrrrr,rrpresent_params merged_paramsvar_kw use_var_kwparam allow_namesr|field param_namekwargsdefault_model_signature var_kw_namerdrdregenerate_model_signaturesX     robjr/cCsDddlm}z|j}Wn ty|}Ynwt||s td|S)Nr!r1z7Unsupported type, must be either BaseModel or dataclass)mainr2__pydantic_model__r_rrr)rr2 model_clsrdrdrerB,s    rBstringcCsddd|dDS)Ncss|]}|VqdSru) capitalize)rwordrdrdre :szto_camel..r)joinsplit)rrdrdrerC9rCcCs4t|dkrt|}|d|ddS|S)Nr!r)rprClower)r pascal_stringrdrdreto_lower_camel=s rT) name_factory input_listrcCsJg}g}|D]}||}||vr||||q||||<q|S)z Make a list unique while maintaining order. We update the list if another one with the same name is set (e.g. root validator overridden in subclass) )appendindex)rrresult result_namesrhv_namerdrdre unique_listGs   rc@seZdZdZdefddZdS)rFz String class where repr doesn't include quotes. Useful with Representation when you want to return a string representation of something that valid (or pseudo-valid) python. rTcCt|Sru)roselfrdrdrerqdszPyObjectStr.__repr__N)__name__ __module__ __qualname____doc__rorqrdrdrdrerF^srFc @seZdZUdZeZeedfed<dddZ defdd Z d edefd d Z d e e ge fde dee ddffddZdefddZdefddZdddZdS)rGz Mixin to provide __str__, __repr__, and __pretty__ methods. See #884 for more details. __pretty__ is used by [devtools](https://python-devtools.helpmanual.io/) to provide human readable representations of objects. . __slots__rTr7cs"fddjD}dd|DS)a6 Returns the attributes to show in __str__, __repr__, and __pretty__ this is generally overridden. Can either return: * name - value pairs, e.g.: `[('foo_name', 'foo'), ('bar_name', ['b', 'a', 'r'])]` * or, just values, e.g.: `[(None, 'foo'), (None, ['b', 'a', 'r'])]` c3s|] }|t|fVqdSru)r^)rsrrdrerzz/Representation.__repr_args__..cSs g|] \}}|dur||fqSrurdrarhrdrdrer{s z0Representation.__repr_args__..)r)rattrsrdrre __repr_args__rszRepresentation.__repr_args__cCs|jjS)zA Name of the instance's class, used in __repr__. )rsrrrdrdre __repr_name__}szRepresentation.__repr_name__join_strcCs|dd|DS)Ncss2|]\}}|durt|n|d|VqdS)N=)reprrrdrdrers0z.Representation.__repr_str__..)rr)rrrdrdre __repr_str__szRepresentation.__repr_str__fmtrNcks`|dVdV|D]\}}|dur|dV||VdVdVqdVdVdS) z Used by devtools (https://python-devtools.helpmanual.io/) to provide a human readable representations of objects (r!Nr,r))rr)rrrrvaluerdrdre __pretty__s   zRepresentation.__pretty__cCs |dS)NrV)rrrdrdre__str__ zRepresentation.__str__cCs|d|ddS)Nr, r)rrrrdrdrerqrzRepresentation.__repr__RichReprResultccs2|D]\}}|dur|Vq||fVqdS)zGet fields for Rich libraryN)r)rr field_reprrdrdre __rich_repr__s  zRepresentation.__rich_repr__rTr7)rTr)rrrrrwrrro__annotations__rrrrrrrrrqrrdrdrdrerGhs   *rGc@seZdZdZdZdefddZdedefdd Zd%ded edefd d Z de efddZ de efddZ de efddZdeeeeffddZdeefddZdefddZdedefddZdedefddZd&d!d"Zdefd#d$Zd S)'rHz Hack to make object's smell just enough like dicts for validate_model. We can't inherit from Mapping[str, Any] because it upsets cython so we have to implement all methods ourselves. _objrcCs ||_dSrur)rrrdrdre__init__rzGetterDict.__init__keyrTc Cs2zt|j|WSty}zt||d}~wwru)r^rr_KeyError)rrrbrdrdre __getitem__s  zGetterDict.__getitem__NrcCst|j||Sru)r^r)rrrrdrdregetzGetterDict.getcCstS)zn We don't want to get any other attributes of obj if the model didn't explicitly ask for them )rxrrdrdre extra_keysszGetterDict.extra_keyscCr)z Keys of the pseudo dictionary, uses a list not set so order information can be maintained like python dictionaries. )rvrrdrdrekeysszGetterDict.keyscsfddDS)Ncsg|]}|qSrdrdrrrrdrersz%GetterDict.values..rdrrdrrerzGetterDict.valuesccs |D] }|||fVqdSru)r)rrrdrdrerszGetterDict.itemsccs&t|jD] }|ds|VqdS)Nr)dirr startswith)rrrdrdre__iter__s  zGetterDict.__iter__cCstdd|DS)Ncss|]}dVqdS)r!Nrd)rrrdrdrersz%GetterDict.__len__..)sumrrdrdre__len__r zGetterDict.__len__itemcCs ||vSru)r rrrdrdre __contains__ zGetterDict.__contains__othercCst|t|kSru)rr)rrrdrdre__eq__szGetterDict.__eq__r7cCsdt|fgSru)rrrdrdrerrzGetterDict.__repr_args__cCsdt|jdS)Nz GetterDict[])r&rrrdrdrerr zGetterDict.__repr_name__rur)rrrrrrrrorrrrrr rrrrrintrboolrrrrrdrdrdrerHs  rHc @seZdZdZdZdededddfdd Zd edefd d Z d edefd dZ ddde edfddZ ddde ddfddZed(dedededefddZededddfddZededefd d!Zed"edefd#d$Zd)d&d'ZdS)*rIzY Class for more convenient calculation of excluded or included fields on values. )_items_typerr)r3r6rTNcCs2||}t|ttfr||t|}||_dSru) _coerce_itemsrnrvrw_normalize_indexesrpr)rrrrdrdrers  zValueItems.__init__rcCs||j|S)z` Check if item is fully excluded. :param item: key or index of a value )is_truerrrrdrdre is_excludedszValueItems.is_excludedcCs ||jvS)zh Check if value is contained in self._items :param item: key or index of value rrrdrdre is_includeds zValueItems.is_includedrbr5cCs|j|}||s |SdS)z :param e: key or index of element on value :return: raw values for element if self._items is dict and contain needed element N)rrr)rrbrrdrdre for_elements zValueItems.for_elementr6v_lengthr4c Csi}d}|D]F\}}t|ts&t|ts&||s&td|d|j|dkr0||}qt|ts9td|dkrA||n|}| || |||<q|sS|S||rgt |D]}| |dq\|St |D]}| |i}||s| ||||<qk|S)af :param items: dict or set of indexes which will be normalized :param v_length: length of sequence indexes of which will be >>> self._normalize_indexes({0: True, -2: True, -1: True}, 4) {0: True, 2: True, 3: True} >>> self._normalize_indexes({'__all__': True}, 4) {0: True, 1: True, 2: True, 3: True} Nz,Unexpected type of exclude value for index "z" __all__zExcluding fields from a sequence of sub-models or dicts must be performed index-wise: expected integer keys or keyword "__all__"r.) rrnrrrrrrs _coerce_valuermergerrange setdefault) rrr#normalized_items all_itemsirh normalized_inormalized_itemrdrdrer s4        zValueItems._normalize_indexesFr~override intersectcs||durS|sdurS|r&|r$SS|r;fddDfddD}n tfddD}i}|D]}|j|||d}|durd|||<qL|S)ae Merge a ``base`` item with an ``override`` item. Both ``base`` and ``override`` are converted to dictionaries if possible. Sets are converted to dictionaries with the sets entries as keys and Ellipsis as values. Each key-value pair existing in ``base`` is merged with ``override``, while the rest of the key-value pairs are updated recursively with this function. Merging takes place based on the "union" of keys if ``intersect`` is set to ``False`` (default) and on the intersection of keys if ``intersect`` is set to ``True``. Ncg|]}|vr|qSrdrdr )r.rdrerLrz$ValueItems.merge..cr0rdrdr r~rdrerLrcsg|]}|vr|qSrdrdr r1rdrerNr)r/)r%rrvr&r)rr~r.r/ merge_keysmergedr merged_itemrd)r~r.rer&1s$    &zValueItems.mergecCsJt|tr |St|trt|d}|St|dd}t|d||S)N.rsz???z!Unexpected type of exclude value )rnrrrfromkeysr^ assert_never)rrardrdrerXs   zValueItems._coerce_itemscCs |dus ||r |S||Sru)rr)rrrdrdrer%fs zValueItems._coerce_valuerhcCs|dup|duS)NT.rdrzrdrdrerlszValueItems.is_truer7cCs d|jfgSrur rrdrdrerprzValueItems.__repr_args__)Fr)rrrrrrrrrrr!rr"rr classmethodr& staticmethodrr%rrrdrdrdrerIs" '& rIc@sDeZdZdZdZdededdfddZd ed eeddfd d Z dS) rJz1 Hide class attribute from its instances rrrrrTNcCs||_||_dSrur9)rrrrdrdrer~s zClassAttribute.__init__instanceownercCs&|dur|jSt|jd|jd)Nz attribute of z is class-only)rr_rr)rr:r;rdrdre__get__szClassAttribute.__get__) rrrrrrorrrr<rdrdrdrerJts rJ directoryfilez mount pointsymlinkz block devicez char deviceFIFOsocket)is_diris_fileis_mount is_symlinkis_block_deviceis_char_deviceis_fifo is_socketrr-cCs:|sJdtD]\}}t||r|Sq dS)z0 Find out what sort of thing a path is. zpath does not existunknown)exists path_typesrr^)rmethodrrdrdrerKs  rKObjc Csj|j}|tvr |Sz|s|tvr!|tur|WS|WSWt|SWt|Stttfy4Yt|Sw)z Return type as is for immutable built-in types Use obj.copy() for built-in empty collections Use copy.deepcopy() for non-empty collections and unknown objects ) rsrQrRrwrrrr\ RuntimeErrorr)robj_typerdrdrerEs rErcCs|dsdSt|kS)NrT)r rLrrdrdrerDs rD>rrr __classcell____orig_bases____orig_class__rcCst| o|tvSru)rDrOrrdrdreis_valid_private_namer rTleftrightcCs*t||tdD] \}}||urdSqdS)a Check that the items of `left` are the same objects as those in `right`. >>> a, b = object(), object() >>> all_identical([a, b, a], [a, b, a]) True >>> all_identical([a, b, [a]], [a, b, [a]]) # new list object, while "equal" is not "identical" False ) fillvalueFT)r_EMPTY)rUrV left_item right_itemrdrdre all_identicals  r[msgcCst|)z Helper to make sure that we have covered all possible types. This is mostly useful for ``mypy``, docs: https://mypy.readthedocs.io/en/latest/literal_types.html#exhaustive-checks )rr)rr\rdrdrer6sr6 all_aliasesdiscriminator_keycCs<t|}t|dkrtd|ddt|d|S)zNValidate that all aliases are the same and if that's the case return the aliasr!zAliases for discriminator z must be the same (got rr)rxrpr"rsortedpop)r]r^unique_aliasesrdrdrerMs  rMtpc CsVt|dd}t|turt|d}t|dr|j}tt|r2t||\}}|tdd|DfS|r[|j t j }t||\}}t t |dkrUtd|d t|||dfSz|j |j }Wn-tyy}z td |jd |d }~wty}z td |jd||d }~wwt|std|d|jd|j |jt|fS)z Get alias and all valid values in the `Literal` type of the discriminator field `tp` can be a `BaseModel` class or directly an `Annotated` `Union` of many. __custom_root_type__Frrcss|] }|D]}|VqqdSrurd)rrrhrdrdrerrz5get_discriminator_alias_and_values..r!zField z& is not the same for all submodels of zType z* is not a valid `BaseModel` or `dataclass`NzModel z% needs a discriminator field for key z of model z needs to be a `Literal`)r^r(r r'hasattrrr*_get_union_alias_and_all_valuesrw __fields__rLtype_rprxr"r&r_rrrrr)rr%)rbr^ is_root_modelr all_values union_typet_discriminator_typerbrdrdrerNs8       rNrjcs0fddt|D}t|\}}t||fS)Ncsg|]}t|qSrd)rN)rtr^rdrer sz3_get_union_alias_and_all_values..)r'ziprM)rjr^zipped_aliases_valuesr]rirdrmreres re)rrkweakref collectionsrrrrr itertoolsrrtypesrr r r r r typingrrrrrrrrrrrrrrrrrrtyping_extensionsr errorsr"r#r$r%r&r'r(r)r*versionr+rr,pathlibr-rr. dataclassesr/rr0rr2r3r4r5r6r7rorr$rLrfloatcomplexrbytesrrefNotImplementedrsEllipsisrQrrvrxrwryrrRr8rtr9r:r;r<r=r>rr?r@rArrBrCrrrrFrGrHrIrJrLrKrNrErDrOrTobjectrXr[r6rMrNrerdrdrdres   P  (       * 0 0   .   &A    <>    & )