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 s. | t ur |t ur tdt| |||||||S )a Return an object to identify dataclass fields.
default is the default value of the field. default_factory is a
0-argument function called to initialize a field's value. If init
is true, the field will be a parameter to the class's __init__()
function. If repr is true, the field will be included in the
object's repr(). If hash is true, the field will be included in the
object's hash(). If compare is true, the field will be used in
comparison functions. metadata, if specified, must be a mapping
which is stored but not otherwise examined by dataclass. If kw_only
is true, the field will become a keyword-only parameter to
__init__().
It is an error to specify both default and default_factory.
z/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 s0 f dd}
| du r|
S |
| S )a Returns 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. If match_args is true,
the __match_args__ tuple is added. If kw_only is true, then by
default all fields are keyword-only. If slots is true, an
__slots__ attribute is added.
c s t |
S r" )r )r? rV rY rE rH r rW r> r rX r r wrap s zdataclass..wrapNr ) 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 tddw tdd | D S )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 instanceNc 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 )z2Returns True if obj is an instance of a dataclass.)r r; r )objr r r _is_dataclass_instance s r c C s* t | tr t | ts | nt| }t|tS )zEReturns True if obj is a dataclass or an instance of a
dataclass.)r= r; r r r )r r? r r r r s
r dict_factoryc C t | stdt| |S )a Return 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)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 )a Return 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)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 du ri t }
i i |D ]^}t|tr|}d}n"t|dkr&|\}}nt|dkr6|\}}}||< ntd|t|trF| sMtd|t|rYtd|||
v rdtd||
| | |< q fd d
}t
| |i |}t ||||||| |
| | d
S ) a Return 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.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 | stdt| t D ];}|jtu rq|js)|j|v r(t d|j dq|j|vrJ|jt
u rA|j t u rAt d|jdt| |j||j< q| j
di |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 instancesr zC is declared with init=False, it cannot be specified with replace()zInitVar z! must be specified with replace()Nr )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