a d@stddlZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z gdZ Gddde Z GdddZ e ZGdd d ZeZeiZGd d d Zed Zed ZedZdZdZdZedZGdddZGdddZGdddZeeddddddddZddZ dd Z!dded!d"d#Z"d$d%Z#d&d'Z$d(d)Z%d*d+Z&d,d-Z'd.d/Z(d0d1Z)d2d3Z*d4d5Z+d6d7Z,d8d9Z-d:d;Z.dd?Z0d@dAZ1dBdCZ2dddde0de1de1e2e1e2e1e2e1e2dDZ3dEdFZ4daddddGdGdGdHdIdJZ5dKdLZ6dMdNZ7dOdPZ8e9dQdRdSZ:dTdUZ;ed[dddddGdGdGd\d]d^Z?d_d`Z@dS)bN) GenericAlias) dataclassfieldFieldFrozenInstanceErrorInitVarMISSINGfieldsasdictastuplemake_dataclassreplace is_dataclassc@s eZdZdS)rN__name__ __module__ __qualname__rr#/usr/lib64/python3.9/dataclasses.pyrrc@seZdZddZdS)_HAS_DEFAULT_FACTORY_CLASScCsdS)Nz rselfrrr__repr__sz#_HAS_DEFAULT_FACTORY_CLASS.__repr__N)rrrrrrrrrsrc@s eZdZdS) _MISSING_TYPENrrrrrrsrc@seZdZddZddZdS) _FIELD_BASEcCs ||_dSNnamerrrrr__init__sz_FIELD_BASE.__init__cCs|jSrrrrrrrsz_FIELD_BASE.__repr__N)rrrr rrrrrrsr_FIELD_FIELD_CLASSVAR_FIELD_INITVAR__dataclass_fields____dataclass_params__ __post_init__z^(?:\s*(\w+)\s*\.)?\s*(\w+)c@s(eZdZdZddZddZddZdS) rtypecCs ||_dSrr')rr(rrrr szInitVar.__init__cCs8t|jtr"t|jts"|jj}n t|j}d|dS)Nzdataclasses.InitVar[]) isinstancer(rrrepr)r type_namerrrrs  zInitVar.__repr__cCst|Sr)r)clsr(rrr__class_getitem__szInitVar.__class_getitem__N)rrr __slots__r rr.rrrrrsrc@s0eZdZdZddZddZddZeeZ dS) r) rr(defaultdefault_factoryr+hashinitcomparemetadata _field_typecCsRd|_d|_||_||_||_||_||_||_|durarz_tuple_str..,))join)rQr rrPr _tuple_strXsrUcs"ttfdd}|S)Nc sNt|tf}|vrdS|z|}W|n |0|S)Nz...)id_thread get_identadddiscard)rkeyresult repr_running user_functionrrwrapperks  z _recursive_repr..wrapper)set functoolswraps)r_r`rr]r_recursive_reprfs rd)globalslocals return_typec Cs|dur i}d|vrt|d<d}|tur4||d<d}d|}ddd|D}d |d |d |d |}d |}d|d|d|}i} t||| | dfi|S)NBUILTINS _return_typez->_return_typerK css|]}d|VqdS)z Nr)rNbrrr rz_create_fn..z def rJr;z: , zdef __create_fn__(z): z return __create_fn__)builtinsrrTkeysexec) rargsbodyrerfrgreturn_annotationtxt local_varsnsrrr _create_fnys   rycCs0|rd|d|d|dS|d|d|S)NzBUILTINS.object.__setattr__(rKr;rL=r)rFrvalue self_namerrr _field_assignsr}cCsd|j}|jturV|jr@|j||<|d|jd|j}q|j||<|d}n8|jr|jturn|j}q|jtur|j||<|j}ndS|jturdSt||j||S)N_dflt_z() if z is _HAS_DEFAULT_FACTORY else rI)rr1rr3r0r6r#r})rOrFrer| default_namer{rrr _field_inits*         rcCsV|jtur|jturd}n&|jtur2d|j}n|jtur@d}|jd|j|S)Nriz=_dflt_z=_HAS_DEFAULT_FACTORYz:_type_)r0rr1r)rOr0rrr _init_params  rc Csd}|D]:}|jr|jtur&|jtus,d}q|rtd|jdqdd|D}|ttdg}|D] }t||||} | rj| | qj|rd d d |D} | |d t d | d |sdg}t d|gdd|D|||ddS)NFTznon-default argument z follows default argumentcSsi|]}d|j|jqS)_type_)rr(rMrrr rz_init_fn..)r_HAS_DEFAULT_FACTORYrKcss|]}|jtur|jVqdSr)r6r#rrMrrrrm s z_init_fn..rLrJr;passr cSsg|]}|jrt|qSr)r3rrMrrrrRrz_init_fn..)rfrerg) r3r0rr1 TypeErrorrupdaterrappendrT_POST_INIT_NAMEry) r rF has_post_initr|re seen_defaultrOrf body_linesline params_strrrr_init_fns: rcCs2tdddddd|Ddg|d}t|S) Nrrz(return self.__class__.__qualname__ + f"(rncSs g|]}|jd|jdqS)z={self.z!r}rrMrrrrRsz_repr_fn..z)"re)ryrTrd)r refnrrr_repr_fns rcCsp|td}|r,dddd|Dd}nd}tdd d |d d d f||dtddd |d ddf||dfS)N)r-rrJrKcss|]}t|jVqdSr)r+rrMrrrrm*rz'_frozen_get_del_attr..rSrI __setattr__)rrr{z if type(self) is cls or name in :z> raise FrozenInstanceError(f"cannot assign to field {name!r}")z)super(cls, self).__setattr__(name, value))rfre __delattr__rz; raise FrozenInstanceError(f"cannot delete field {name!r}")z"super(cls, self).__delattr__(name))rrTry)r-r rerf fields_strrrr_frozen_get_del_attr&s2  rcCs$t|ddd|||dg|dS)N)rotherz%if other.__class__ is self.__class__:z return zreturn NotImplementedr)ry)rop self_tuple other_tuplererrr_cmp_fn?srcCs$td|}tddd|dg|dS)Nr__hash__rz return hash(r;r)rUry)r rerrrr_hash_fnMs   rcCs$||jup"t||juo"|j|juSr)ClassVarr( _GenericAlias __origin__)a_typetypingrrr _is_classvarUs  rcCs||jupt||juSr)rr()r dataclassesrrr _is_initvar]s  rc Cst|}|rd}|d}|s2tj|jj}n2tj|j}|rd|j||urdtj|jj}|r|||d|rdSdS)NTF)_MODULE_IDENTIFIER_REmatchgroupsysmodulesgetr__dict__) annotationr-a_moduleris_type_predicaterrx module_namemodulerrr_is_typeds)  rcCs8t||t}t|tr|}nt|tjr,t}t|d}||_||_t |_ t j d}|rt||st|jtrt|j|||jtrt|_ |j t urt j t}t||st|jtrt|j|||jtrt|_ |j ttfvr|jturtd|jd|j t ur4t|jtttfr4tdt|jd|jd|S)N)r0rfield z cannot have a default factoryzmutable default z for field z$ is not allowed: use default_factory)r=rr*rr8MemberDescriptorTyperrr(r!r6rrrrstrrrr"rrrr#r1rr0listdictrarH)r-a_namerr0rOrrrrr _get_fieldsJ              rcCs||jvrdSt|||dS)NTF)rsetattr)r-rr{rrr_set_new_attributes  rcCsdSrrr-r rerrr_hash_set_nonesrcCsdd|D}t||S)NcSs(g|] }|jdur|jrn|jr|qSr)r2r4rMrrrrRrz_hash_add..)r)r-r refldsrrr _hash_addsrcCstd|jdS)Nz-Cannot overwrite attribute __hash__ in class )rrrrrr_hash_exceptionsr))FFFF)FFFT)FFTF)FFTT)FTFF)FTFT)FTTF)FTTT)TFFF)TFFT)TFTF)TFTT)TTFF)TTFT)TTTF)TTTTc si}jtjvr tjjj}ni}ttt||||||d} d} jdddD]D} t| t d} | durVd} | D]} | || j <qzt| tj rVd} qVj di}fdd|D}|D]L} | || j <tt| j dtr| jturt| j qt| j | jqĈjD].\}}t|tr||vrt|dq| rz| rf|sftd | sz|rztd tt |j d t}|tup|duod jv }|r|std |rtt}dd| D}tdt|||d|vrdnd|dd| D}|rHdd|D}tdt|||rdd|D}td|}td|}td td d|||d|rdd|D}td|}td|}dD]>\}}t|t|||||drtd|djdq|r8t||D].}t|j|rtd|jdjqtt |t |t ||f}|rh|||_!tdsjt"t#$%dd _&S)!NFrT__annotations__csg|]\}}t||qSr)r)rNrr(r-rrrR_sz"_process_class..z& is a field but has no type annotationz5cannot inherit non-frozen dataclass from a frozen onez5cannot inherit frozen dataclass from a non-frozen oner__eq__z eq must be true if order is truecSsg|]}|jttfvr|qSr)r6r!r#rMrrrrRsr r__dataclass_self__cSsg|]}|jtur|qSrr6r!rMrrrrRrcSsg|]}|jr|qSr)r+rMrrrrRrrcSsg|]}|jr|qSrr4rMrrrrRrrz==rcSsg|]}|jr|qSrrrMrrrrRr))__lt__<)__le__z<=)__gt__>)__ge__z>=zCannot overwrite attribute z in class z). Consider using functools.total_ordering__doc__z -> Noneri)'rrrrr_PARAMSrA__mro__r=_FIELDSvaluesrrFritemsr*rr0rdelattrrrHhasattrrrrrrUrrr _hash_actionboolrrinspect signaturer r)r-r3r+rCrDrErFr reany_frozen_basehas_dataclass_basesrl base_fieldsrOcls_annotations cls_fieldsrr{ class_hashhas_explicit_hashrr field_listrrrr hash_actionrrr_process_class)s                        rFrBcs*fdd}|dur"|S||S)aReturns the same class as was passed in, with dunder methods added based on the fields defined in the class. Examines PEP 526 __annotations__ to determine fields. If init is true, an __init__() method is added to the class. If repr is true, a __repr__() method is added. If order is true, rich comparison dunder methods are added. If unsafe_hash is true, a __hash__() method function is added. If frozen is true, fields may not be assigned to after instance creation. cst|Sr)rrrCrFr3rDr+rErrwrapszdataclass..wrapNr)r-r3r+rCrDrErFrrrrrsrcCs@zt|t}Wnty(tdYn0tdd|DS)zReturn a tuple describing the fields of this dataclass. Accepts a dataclass or an instance of one. Tuple elements are of type Field. z0must be called with a dataclass type or instancecss|]}|jtur|VqdSrrrMrrrrmrzfields..)r=rAttributeErrorrtupler)class_or_instancer rrrr s  r cCstt|tS)z2Returns True if obj is an instance of a dataclass.)rr(r)objrrr_is_dataclass_instancesrcCs*t|trt|ts|nt|}t|tS)zEReturns True if obj is a dataclass or an instance of a dataclass.)r*r(rrr)rr-rrrrs r dict_factorycCst|stdt||S)aReturn the fields of a dataclass instance as a new dictionary mapping field names to field values. Example usage: @dataclass class C: x: int y: int c = C(1, 2) assert asdict(c) == {'x': 1, 'y': 2} If given, 'dict_factory' will be used instead of built-in dict. The function applies recursively to field values that are dataclass instances. This will also look into built-in containers: tuples, lists, and dicts. z0asdict() should be called on dataclass instances)rr _asdict_inner)rrrrrr sr cst|rDg}t|D]&}tt||j}||j|fq|St|trrt|drrt |fdd|DSt|t tfrt |fdd|DSt|t rt |fdd| DSt |SdS)N_fieldscsg|]}t|qSrrrNvrrrrRQrz!_asdict_inner..c3s|]}t|VqdSrrrrrrrmVrz _asdict_inner..c3s&|]\}}t|t|fVqdSrrrNkrrrrrmXs)rr rr=rrr*rrr(rrrcopydeepcopy)rrr\rOr{rrrr6s  r tuple_factorycCst|stdt||S)aReturn the fields of a dataclass instance as a new tuple of field values. Example usage:: @dataclass class C: x: int y: int c = C(1, 2) assert astuple(c) == (1, 2) If given, 'tuple_factory' will be used instead of built-in tuple. The function applies recursively to field values that are dataclass instances. This will also look into built-in containers: tuples, lists, and dicts. z1astuple() should be called on dataclass instances)rr_astuple_inner)rrrrrr _sr cst|r>g}t|D] }tt||j}||q|St|trlt|drlt |fdd|DSt|t tfrt |fdd|DSt|t rt |fdd| DSt |SdS)Nrcsg|]}t|qSrrrrrrrRrz"_astuple_inner..c3s|]}t|VqdSrrrrrrrmrz!_astuple_inner..c3s&|]\}}t|t|fVqdSrrrrrrrms)rr rr=rrr*rrr(rrrrr)rrr\rOr{rrrrws   rr)bases namespacer3r+rCrDrErFc sdurint} i} |D]} t| tr<| } d}nDt| dkrR| \} }n.t| dkrr| \} }}|| <ntd| t| tr| std| t| rtd| | | vrtd| | | || | <q$| d <t ||ifd d }t ||||||| d S) aReturn a new dynamically created dataclass. The dataclass name will be 'cls_name'. 'fields' is an iterable of either (name), (name, type) or (name, type, Field) objects. If type is omitted, use the string 'typing.Any'. Field objects are created by the equivalent of calling 'field(name, type [, Field-info])'. C = make_dataclass('C', ['x', ('y', int), ('z', int, field(init=False))], bases=(Base,)) is equivalent to: @dataclass class C(Base): x: 'typing.Any' y: int z: int = field(init=False) For the bases and namespace parameters, see the builtin type() function. The parameters init, repr, eq, order, unsafe_hash, and frozen are passed to dataclass(). Nz typing.AnyrzInvalid field: z'Field names must be valid identifiers: z"Field names must not be keywords: zField name duplicated: rcs |Sr)r)rxrrrrz make_dataclass..rB) rrar*rlenr isidentifierkeyword iskeywordrYr8 new_classr)cls_namer rrr3r+rCrDrErFseenannsitemrtpspecr-rrrr s8          r cKst|stdt|tD]v}|jtur.q|jsR|j|vrt d|jdq|j|vr|jt ur|j t urt d|jdt||j||j<q|j fi|S)a,Return a new object replacing specified fields with new values. This is especially useful for frozen classes. Example usage: @dataclass(frozen=True) class C: x: int y: int c = C(1, 2) c1 = replace(c, x=3) assert c1.x == 3 and c1.y == 2 z1replace() should be called on dataclass instancesrzC is declared with init=False, it cannot be specified with replace()zInitVar z! must be specified with replace())rrr=rrr6r"r3rrHr#r0r __class__)rchangesrOrrrr s   r )N)Arerrr8rrrprbrWr__all__rrrrrrr9r7rr!r"r#rrrcompilerrrrArrUrdryr}rrrrrrrrrrrrrrrrrrr rrrr rrr rr r rrrrs   <   62 ;R>  ) B