o ɝiL @s8UddlZddlZddlZddlZddlmZddlmZddlmZm Z m Z mZ m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZm Z m!Z!m"Z#m$Z%zddlm&Z'Wne(yrddlm)Z'Ynwzddlm*Z+Wn e(yd Z+Ynwzdd l,m-Z.Wn e(yd Z.Ynwej/d krd ed e de de fddZ0n ej/dkrd ed e de de fddZ0n d ed e de de fddZ0ej/d kreZ1ndpde d e de de fddZ1edZ2e de fZ3e ge fZ4e de2fZ5ddhZ6e!hZ7ee e8d<e9edre7:ej!ej/dkr)dee deee fd d!Z;ndd"lm;Z<d#ee deee fd$d!Z;ej/dkrZdd%lm=Z=dee dee dffd&d'Z>n$dd(lm>Z?d#ee dee dffd)d*Z@d#ee dee dffd+d'Z>ej/d krd#ee dee fd,d-ZAn d#ee dee fd.d-ZAej/d/krd#eee deBfd0d1ZCe+fZDnddl,Z,ddlZd#eee deBfd2d1ZCej=e,j*e,j-fZDeeEefZFer9dd3lGmHZHeeeEe fddfZIe eEe fZJe e e fZKeeEZLeeEZMeeNeEfZOe eOZPe eOe fZQeeOe fZRee3ddfZSeeeeEe fZTd4ZUeUr2eVe ZWneVe e e fZWd5ZXdjYZZdeZe!dfZ[ee e e fe8d6<ej/dkr^d e deBfd7d8Z\nej/dd9dkrrd e deBfd:d8Z\n d e deBfd;d8Z\dZ]d?e eEee fd@eeEde eEee ffdAdBZ^d ee deBfdCdDZ_d ee deBfdEdFZ`d ee dee dffdGdHZad ee dee dffdIdJZbd ee deBfdKdLZcd ee deBfdMdNZdd e deBfdOdPZed e deBfdQdRZfedSeEZgd ee deBfdTdUZhd ee dee fdVdWZids " z#convert_generics..)rErUrQrr\rXrMr_TypesUnionType functoolsreduceoperatoror_setattrAttributeError)rForiginargs convertedr!rHr-r\s,       r$ cCs|tuSr*)rrHr!r!r-is_unionrocCs|tup|tjuSr*)rtypesr"rHr!r!r-ros) ModelFieldF)' AnyCallableNoArgAnyCallableNoneType is_none_typedisplay_as_typeresolve_annotationsis_callable_typeis_literal_typeall_literal_values is_namedtuple is_typeddictis_typeddict_special is_new_typenew_type_supertype is_classvar is_finalvarupdate_field_forward_refsupdate_model_forward_refsTupleGenerator DictStrAnyDictAnySetStrListStrIntStrAbstractSetIntStr DictIntStrAnyCallableGeneratorReprArgsAnyClassMethodr WithArgsTypesrQrE get_sub_types typing_baser7roStrPathMappingIntStrAny NONE_TYPEScC|tvSr*rr&r!r!r-rvXrprvcCs0tD] }||ur dSqt|rt|dkSdS)NTr*F)rrzr{)r& none_typer!r!r-rv]s cCrr*rrr!r!r-rvjrpvcCst|tst|tst|ts|j}tt|r&ddtt t |dSt|tr3t | ddSz|j WStyHt | ddYSw)NzUnion[z, ]ztyping.)rMrrrA __class__rorEjoinmaprwrQr^replacerBrirr!r!r-rwns   rwraw_annotations module_namec Csd}|rztj|}Wn tyYnw|j}i}|D]L\}}t|trFdtjkr2dks8ntjdkr@t|ddd}nt|dd}ztjd krUt ||dd d }nt ||d}Wn t yeYnw|||<q|S) z Partially taken from typing.get_type_hints. Resolve string or ForwardRef annotations into type objects if possible. Nrm)r$r%r>)r$rnr]FT) is_argumentis_class)r)r$ r!)r4) rVmodulesKeyError__dict__itemsrMr^rWr r NameError)rr base_globalsmodule annotationsnamevaluer!r!r-rxs0  "     rxcCs|tup t|tuSr*)rrErr!r!r-ryrycCstduo t|tvSr*)rrEr<rr!r!r-rzrrzcCst|Sr*rRrr!r!r-literal_valuesrprcCs(t|s|fSt|}tdd|DS)z This method is used to retrieve all Literal values as Literal can be used recursively (see https://www.python.org/dev/peps/pep-0586) e.g. `Literal[Literal[Literal[1, 2, 3], "foo"], 5, None]` css"|] }t|D]}|VqqdSr*)r{)r`rxr!r!r-rbs z%all_literal_values..)rzrrX)r&valuesr!r!r-r{sr{cC ddlm}||tot|dS)zy Check if a given class is a named tuple. It can be either a `typing.NamedTuple` or `collections.namedtuple` rlenient_issubclass_fields)pydantic.v1.utilsrrXrUr&rr!r!r-r| r|cCr)z Check if a given class is a typed dict (from `typing` or `typing_extensions`) In 3.10, there will be a public method (https://docs.python.org/3.10/library/typing.html#typing.is_typeddict) rr __total__)rrdictrUrr!r!r-r}rr}cCs|tup|tuSr*)TypedDictRequiredTypedDictNotRequiredrr!r!r-_check_typeddict_specialr8rcCt|p tt|S)zN Check if type is a TypedDict special form (Required or NotRequired). )rrErr!r!r-r~sr~ test_typecCst|tjo t|dS)z> Check whether type_ was created using typing.NewType __supertype__)rMrrrUrr!r!r-rsrcCst|dr |j}t|ds|S)Nr)rUrrr!r!r-rs  rcCs(|durdS|jtjkot|dddkS)NF_namer)rrrDrr!r!r-_check_classvarsrcCs2|durdS|jtjkotjdkpt|dddkS)z9 Check if a given type is a `typing.Final` type. NFr=rr)rrrVrWrDrr!r!r-_check_finalvars&rann_typecCs6t|s tt|r dS|jtkr|jdrdSdS)NTz ClassVar[F)rrErr __forward_arg__ startswithrr!r!r-rs rcCrr*)rrErr!r!r-rrrfieldrrcCsd}|jjtkrd}t|j||pd|_|jjtkr&d}t|j||p#d|_|r,||jr<|jD] }t|||dq2|jdurG| dSdS)z] Try to update ForwardRefs on fields based on this ModelField, globalns and localns. FTNr'r() r&rr r. outer_type_prepare sub_fieldsrdiscriminator_key&prepare_discriminated_union_sub_fields)rr'r(rsub_fr!r!r-r s     rmodelfields json_encodersrexc_to_suppressc Cs|jtjvrtj|jj}ni}||j||D]}z t|||dWq|y/Yqwt| D]0}t |t rBt |}n t |t rJ|}nq6z t |||pRd} Wn |y^Yq6w|||| <q6dS)zL Try to update model fields ForwardRefs based on model and localns. rN) __module__rVrrcopy setdefaultrBrr3keysrMr^r r.pop) rrrr(rr'fkeyfrnew_keyr!r!r-r"s.      rcCsB|turdSt|durdSt|}|rt|dtsdS|dS)z Tries to get the class of a Type[T] annotation. Returns True if Type is used without brackets. Otherwise returns None. TNr)rArErQrM)r&rkr!r!r- get_classIs rcCs@t|}|turtt|dSt|rddt|DS|gS)z~ Return all the types that are allowed by type `tp` `tp` can be a `Union` of allowed types or an `Annotated` type rcSsg|] }t|D]}|qqSr!)r)r`r?rr!r!r- dsz!get_sub_types..)rErrrQro)rFrjr!r!r-r[s r)NNrT)srdrfrVtypingcollections.abcrosrrrrTypingCallablerrr r r r r rrrrrrrrrrrtyping_extensionsrrrrrrrrr ImportErrorrr r_rqr"rcrWr.r7r9rsrtAnyArgTCallablerCr<__annotations__rUaddrErGrIrQr[rZr\boolrorr^rpydantic.v1.fieldsrrrrrrrintrrrrrrMYPY classmethodr__all__rrurrvrwrxryrzrr{r|r}rr~rrrrrrrr BaseExceptionrrrr!r!r!r-s   `                   )        +   2#       $'