o
0iF @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dlm
Z
m Z g dZ G dd de
ZG dd dZe ZG dd d Ze ZG d
d d Ze Zei ZG d d
d
ZedZedZedZdZdZ dZ e dZ dd Z G dd dZ!G dd dZ"G dd d Z#eed d dd ded d d Z$d!d" Z%d#d$ Z&dded%d&d'Z'd(d) Z(d*d+ Z)d,d- Z*d.d/ Z+d0d1 Z,d2d3 Z-d4d5 Z.d6d7 Z/d8d9 Z0d:d; Z1dd? Z3d@dA Z4dBdC Z5dDdE Z6dFdG Z7dHdI Z8dJdK Z9i dLddMddNddOddPe7dQddRe8dSddTe8dUe9dVe8dWe9dXe8dYe9dZe8d[e9Z:d\d] Z;d^d_ Zd~d d d ddddddd ddddde dfdgZ?dhdi Z@djdk ZAdldm ZBeCdndodpZDdqdr ZEeFdsdtduZGdvdw ZHdxdd d d ddddddd dddddy dzd{ZId|d} ZJdS ) N) FunctionType GenericAlias)
dataclassfieldFieldFrozenInstanceErrorInitVarKW_ONLYMISSINGfieldsasdictastuplemake_dataclassreplace is_dataclassc @ e Zd ZdS )r N__name__
__module__ __qualname__ r r 2/opt/alt/python310/lib64/python3.10/dataclasses.pyr s r c @ s e Zd Zdd ZdS )_HAS_DEFAULT_FACTORY_CLASSc C s dS )Nz r selfr r r __repr__ z#_HAS_DEFAULT_FACTORY_CLASS.__repr__N)r r r r r r r r r s r c @ r )
_MISSING_TYPENr r r r r r r c @ r )
_KW_ONLY_TYPENr r r r r r r r c @ s e Zd Zdd Zdd ZdS ) _FIELD_BASEc C
|| _ d S Nnamer r$ r r r __init__
z_FIELD_BASE.__init__c C s | j S r" r# r r r r r s z_FIELD_BASE.__repr__N)r r r r& r r r r r r s r _FIELD_FIELD_CLASSVAR_FIELD_INITVARZ__dataclass_fields__Z__dataclass_params__Z
__post_init__z^(?:\s*(\w+)\s*\.)?\s*(\w+)c s" t t fdd}|S )Nc sL t | t f}| v rdS | z | }W | |S | w )Nz...)id_thread get_identadddiscard)r keyresult repr_running
user_functionr r wrapper s
z _recursive_repr..wrapper)set functoolswraps)r4 r5 r r2 r _recursive_repr s
r9 c @ s( e Zd ZdZdd Zdd Zdd ZdS ) r typec C r! r" r: )r r; r r r r& r' zInitVar.__init__c C s8 t | jtrt | jts| jj}nt| j}d| dS )Nzdataclasses.InitVar[])
isinstancer; r r repr)r Z type_namer r r r s
zInitVar.__repr__c C s t |S r" )r )clsr; r r r __class_getitem__ s zInitVar.__class_getitem__N)r r r __slots__r& r r@ r r r r r s
r c @ s4 e Zd ZdZdd Zedd Zdd Zee Z
dS ) r ) r$ r; defaultdefault_factoryr> hashinitcomparemetadatakw_only _field_typec C sX d | _ d | _|| _|| _|| _|| _|| _|| _|d u r tnt
|| _ || _ d | _
d S r" )r$ r; rB rC rE r> rD rF _EMPTY_METADATAtypesMappingProxyTyperG rH rI ) r rB rC rE r> rD rF rG rH r r r r& s
zField.__init__c C s^ d| j d| jd| jd| jd| jd| jd| jd| jd | jd
| j d | j
d S )
Nz Field(name=z,type=z ,default=z,default_factory=z,init=,repr=z,hash=z ,compare=z
,metadata=z ,kw_only=z
,_field_type=)) r$ r; rB rC rE r> rD rF rG rH rI r r r r r . s.
zField.__repr__c C s, t t| jdd }|r|| j|| d S d S )N __set_name__)getattrr; rB )r ownerr$ funcr r r rO F s zField.__set_name__N) r r r rA r& r9 r rO classmethodr r@ r r r r r s
r c @ s e Zd ZdZdd Zdd ZdS )_DataclassParamsrE r> eqorder unsafe_hashfrozenc C s( || _ || _|| _|| _|| _|| _d S r" rU )r rE r> rV rW rX rY r r r r& Y s
z_DataclassParams.__init__c
C s6 d| j d| jd| jd| jd| jd| jd
S )Nz_DataclassParams(init=rM z,eq=z,order=z
,unsafe_hash=z,frozen=rN rU r r r r r a s z_DataclassParams.__repr__N)r r r rA r& r r r r r rT P s rT TrB rC rE r> rD rF rG rH c C s0 | t ur
|t ur
tdt| |||||||S )Nz/cannot specify both default and default_factory)r
ValueErrorr rZ r r r r o s r c C s$ t dd | D t dd | D fS )Nc s s | ] }|j r|js|V qd S r" rE rH .0fr r r z(_fields_in_init_order..c s s | ] }|j r|jr|V qd S r" r\ r] r r r r` ra )tupler r r r _fields_in_init_order s rd c s( |sdS dd fdd|D dS )N()(,c s g | ]
} d |j qS ).r# r] obj_namer r
z _tuple_str..,))join)rj r r ri r
_tuple_str s ro )globalslocals return_typec
C s |d u ri }d}|t ur||d< d}d|}ddd |D }d| d | d
| d | }d | }d
| d| d| }i } t||| | d di |S )N Z _return_typez->_return_typerg
c s s | ]}d | V qdS )z Nr )r^ br r r r` z _create_fn..z def rf rN z:
, zdef __create_fn__(z):
z
return Z
__create_fn__r )r
rn keysexec)
r$ argsbodyrp rq rr Zreturn_annotationZtxtZ
local_varsnsr r r
_create_fn s
r} c C s0 | rd| d|d| dS | d| d| S )Nz*__dataclass_builtins_object__.__setattr__(rg rN rh =r )rY r$ value self_namer r r
_field_assign s r c C s d| j }| jtur+| jr | j||< | d| j d| j }n6| j||< | d}n+| jrE| jtu r7| j }n | jturD| j||< | j }n|rT| jturT| j||< |}nd S | jtu r]d S t|| j ||S )NZ_dflt_z() if z is _HAS_DEFAULT_FACTORY else re )r$ rC r
rE rB rI r* r )r_ rY rp r slotsZ default_namer r r r _field_init s2
r c C sV | j tu r
| jtu r
d}n| j turd| j }n| jtur d}| j d| j | S )Nrs z=_dflt_z=_HAS_DEFAULT_FACTORYz:_type_)rB r
rC r$ )r_ rB r r r _init_param s
r c C s d}|D ] } | j r!| jtu r| jtu sd}q|r!td| jdqdd | D }
|
tttd g } | D ]} t | ||
||} | rG|
| q6|rbd d d
| D }
|
| d t d |
d
| sgdg} dd |D }|r~|dg7 }|dd |D 7 }t
d|g| | |
|d dS )NFTznon-default argument z follows default argumentc S s i | ]
}d |j |jqS )Z_type_)r$ r; r] r r r
# rl z _init_fn..)r
_HAS_DEFAULT_FACTORYZ __dataclass_builtins_object__rg c s s | ] }|j tu r|jV qd S r" )rI r* r$ r] r r r r` 4 s
z_init_fn..rh rf rN passc S g | ]}t |qS r r r] r r r rk < z _init_fn..*c S r r r r] r r r rk B r r& )rq rp rr )rE rB r
rC TypeErrorr$ updater objectr appendrn _POST_INIT_NAMEr} )r Z
std_fieldsZkw_only_fieldsrY
has_post_initr rp r Z seen_defaultr_ rq Z
body_lineslineZ
params_strZ _init_paramsr r r _init_fn sH
r c C s2 t dddddd | D d g|d}t|S ) Nr r z(return self.__class__.__qualname__ + f"(rw c S s g | ] }|j d |j dqS )z={self.z!r}r# r] r r r rk O s z _repr_fn..z)"rp )r} rn r9 )r rp fnr r r _repr_fnK s
r c C sp | t d}|rdddd |D d }nd}tdd d
| d d d
f||dtddd
| d ddf||dfS )N)r? r rf rg c s s | ]}t |jV qd S r" )r> r$ r] r r r r` Z rv z'_frozen_get_del_attr..rm re __setattr__)r r$ r 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))rq rp __delattr__r% z; raise FrozenInstanceError(f"cannot delete field {name!r}")z"super(cls, self).__delattr__(name))r rn r} )r? r rp rq Z
fields_strr r r _frozen_get_del_attrV s2
r c C s$ t | ddd| | | dg|dS )N)r otherz%if other.__class__ is self.__class__:z return zreturn NotImplementedr )r} )r$ op
self_tuple other_tuplerp r r r _cmp_fno s r c C s$ t d| }tddd| dg|dS )Nr __hash__r z return hash(rN r )ro r} )r rp r r r r _hash_fn} s
r c C s$ | |j u pt| |ju o| j|j u S r" )ClassVarr; Z
_GenericAlias
__origin__)a_typetypingr r r _is_classvar s
r c C s | |j u p t| |j u S r" )r r; r dataclassesr r r _is_initvar s
r c C s
| |j u S r" )r r r r r _is_kw_only r' r c C s t | }|rAd }|d}|stj|jj}ntj|j}|r2|j||u r2tj|jj}|rA|||d|rAdS dS )N TF)_MODULE_IDENTIFIER_REmatchgroupsysmodulesgetr __dict__)
annotationr? Za_moduler Zis_type_predicater r| Z module_namemoduler r r _is_type s
)
r c C sp t | |t}t|tr|}n
t|tjrt}t|d}||_||_t |_
t j
d}|rDt||sAt|jtrDt|j| ||jtrDt|_
|j
t u rft j t }t||sct|jtrft|j| ||jtrft|_
|j
ttfv r{|jtur{td|j d|j
t tfv r|jtu r||_n|jturtd|j d|j
t u rt|jtt t frt dt|j d|j d|S ) N)rB r field z cannot have a default factoryz$ is a ClassVar but specifies kw_onlyzmutable default z for field z$ is not allowed: use default_factory) rP r
r= r rK MemberDescriptorTyper r$ r; r( rI r r r r strr r r) r r r r* rC r rH rB listdictr6 r[ )r? Za_namer Zdefault_kw_onlyrB r_ r r r r r
_get_field sV
r c C s" t |tr| j d|j |_|S )Nrh )r= r r r )r? r r r r
_set_qualname1 s
r c C s( || j v rdS t| | t| || dS )NTF)r r setattr)r? r$ r r r r _set_new_attribute8 s
r c C s d S r" r r? r rp r r r _hash_set_noneG r r c C s dd |D }t | t||S )Nc S s( g | ]}|j d u r
|jrn|j r|qS r" )rD rF r] r r r rk K s ( z _hash_add..)r r )r? r rp fldsr r r _hash_addJ s r c C s t d| j )Nz-Cannot overwrite attribute __hash__ in class )r r r r r r _hash_exceptionN s r )FFFF)FFFT)FFTF)FFTT)FTFF)FTFT)FTTF)FTTT)TFFF)TFFT)TFTF)TFTT)TTFF)TTFT)TTTF)TTTTc
% C s0 i }
| j tjv rtj| j j} ni } t| tt|||||| d} d}
| jddd D ]"}t|t d }|d urMd}
|
D ]}||
|j < q=t|tj rMd} q+| j
di }g }d}tjt }| D ]/\}}t||syt|trt|| ||jtr|rt|dd}d}qb|t| ||| qb|D ]%}||
|j < tt| |j d tr|jtu rt| |j qt| |j |j q| j D ]\}}t|tr||vrt|dq|
r| r|std| s|rtd t| t |
| j
d
t}|tu p|d u od | jv }|r |s t d d
d |
D }t |\}} |r8t | t } t | dt!||| || d|
v r2dnd| | dd |
D } |rTdd | D } t | dt"| | |rudd | D } t#d| } t#d| }!t | d t$d d| |!| d |rdd | D } t#d| } t#d| }!dD ] \}}"t | |t$||"| |!| drtd| d | j d q|rt%| | | D ]}#t | |#j|#rtd|#j d | j qt&t'|t'|t'||f }$|$r|$| | | | _(t| d s| jtt)*| +d d | _,|r t | d!t-d"d# |D | rt.| |} t/0| | S )$NFr T__annotations__z3 is KW_ONLY, but KW_ONLY has already been specifiedz& 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 truec S s g | ] }|j ttfv r|qS r )rI r( r* r] r r r rk s z"_process_class..r& r Z__dataclass_self__c S s g | ] }|j tu r|qS r rI r( r] r r r rk s c S g | ]}|j r|qS r )r> r] r r r rk r c S r r rF r] r r r rk r r z==r c S r r r r] r r r rk # r ))__lt__<)__le__z<=)__gt__>)__ge__z>=zCannot overwrite attribute z
in class z). Consider using functools.total_ordering__doc__z -> Noners __match_args__c s | ]}|j V qd S r" r# r] r r r r` J z!_process_class..)1r r r r r _PARAMSrT __mro__rP _FIELDSvaluesr$ rY r r itemsr r= r r r r r r r rB r
delattrr[ rd hasattrr r r r ro r r _hash_actionboolr inspectZ signaturer r rb
_add_slotsabcupdate_abstractmethods)%r? rE r> rV rW rX rY
match_argsrH r r rp Zany_frozen_baseZhas_dataclass_basesru Z base_fieldsr_ Zcls_annotationsZ
cls_fieldsZ KW_ONLY_seenr r$ r; r Z
class_hashZhas_explicit_hashZall_init_fieldsZstd_init_fieldsZkw_only_init_fieldsr Z
field_listr r r r r Z hash_actionr r r _process_classq s
r c s fddt D S )Nc s g | ]}t |jqS r )rP r$ r] r r r rk Y s z'_dataclass_getstate..rc r r r r _dataclass_getstateX s r c C s, t t| |D ] \}}t| |j| qd S r" )zipr r r r$ )r stater r r r r _dataclass_setstate\ s r c C s d| j v r
t| j dt| j }tdd t| D }||d< |D ]}||d q#|dd t| dd }t| | j| j |} |d urI|| _
|rQt | _ t
| _| S )NrA z already specifies __slots__c s r r" r# r] r r r r` l r z _add_slots..r r )r r r r rb r poprP r; __bases__r r __getstate__r __setstate__)r? is_frozenZcls_dict field_namesZ
field_namequalnamer r r r b s
r F rE r> rV rW rX rY r rH r c s2 f dd}
| d u r|
S |
| S )Nc s t |
S r" )r )r? rV rY rE rH r rW r> r rX r r wrap s zdataclass..wrapr ) r? rE r> rV rW rX rY r rH r r r r r r s
r c C s@ zt | t}W n ty tdd w tdd | D S )Nz0must be called with a dataclass type or instancec s s | ]
}|j tu r|V qd S r" r r] r r r r` s zfields..)rP r AttributeErrorr rb r )Zclass_or_instancer r r r r s
r c C s t t| tS r" )r r; r )objr r r _is_dataclass_instance s r c C s, t | tr
t | ts
| nt| }t|tS r" )r= r; r r r )r r? r r r r s
r dict_factoryc C t | s tdt| |S )Nz0asdict() should be called on dataclass instances)r r
_asdict_inner)r r r r r r
r c s t | r"g }t| D ]}tt| |j }||j|f q
|S t| tr9t| dr9t | fdd| D S t| t
tfrMt | fdd| D S t| t rat | fdd| D S t
| S )N_fieldsc g | ]}t | qS r r r^ vr r r rk r z!_asdict_inner..c 3 | ]}t | V qd S r" r r r r r r` rv z _asdict_inner..c 3 ( | ]\}}t | t | fV qd S r" r r^ kr r r r r` s )r r r rP r$ r r= rb r r; r r r copydeepcopy)r r r1 r_ r r r r r s
r
tuple_factoryc C r )Nz1astuple() should be called on dataclass instances)r r _astuple_inner)r r r r r r
r r
c s t | r g }t| D ]}tt| |j }|| q
|S t| tr6t| dr6t | fdd| D S t| t
tfrJt | fdd| D S t| t r^t | fdd| D S t
| S )Nr c r r r r r r r rk ( r z"_astuple_inner..c 3 r r" r r r r r r` - rv z!_astuple_inner..c 3 r r" r r r r r r` / s )r r r rP r$ r r= rb r r; r r r r r )r r r1 r_ r r r r r s
r r ) bases namespacerE r> rV rW rX rY r rH r c s d u ri t }
i i |D ]^}t|tr |}d}n"t|dkr'|\}}nt|dkr7|\}}}||< ntd|t|trG| sNtd|t|rZtd|||
v retd||
| | |< q fdd }t
| |i |}t ||||||| |
| | d
S ) Nz
typing.Anyr zInvalid field: z'Field names must be valid identifiers: z"Field names must not be keywords: zField name duplicated: c s | | | d< d S )Nr )r )r| Z annotationsdefaultsr r r exec_body_callbackn s
z*make_dataclass..exec_body_callbackr ) r6 r= r lenr isidentifierkeyword iskeywordr. rK new_classr )cls_namer r
r rE r> rV rW rX rY r rH r seenitemr$ tpspecr r? r r
r r 5 s<
r c K s t | s tdt| t D ];}|jtu rq|js*|j|v r)t d|j dq|j|vrK|jt
u rB|j t u rBt d|jdt| |j||j< q| j
di |S )Nz1replace() should be called on dataclass instancesr zC is declared with init=False, it cannot be specified with replace()zInitVar z! must be specified with replace()r )r r rP r r rI r) rE r$ r[ r* rB r
__class__)r Zchangesr_ r r r r } s
r r" )Krer r rK r r builtinsr7 r r, r r __all__r r r r r r
r r rL rJ r r( r) r* r r r compiler r9 r r rT r rd ro r} r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r rb r
r r r r r r r s
@ << ;_
h
" ) H