a
R�f�� � @ st d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z g d�Z
G dd� de �Z G dd� d�Z
e
� ZG dd � d �Ze� Ze�i �ZG d
d � d �Zed �Zed
�Zed�ZdZdZdZe �d�ZG dd� d�Z G dd� d�Z G dd� d�Z eedddddd�dd �Z d d � Z d d � Z!dded!�d"d#�Z"d$d%� Z#d&d'� Z$d(d)� Z%d*d+� Z&d,d-� Z'd.d/� Z(d0d1� Z)d2d3� Z*d4d5� Z+d6d7� Z,d8d9� Z-d:d;� Z.d<d=� Z/d>d?� Z0d@dA� Z1dBdC� Z2dddde0de1de1e2e1e2e1e2e1e2dD�Z3dEdF� Z4daddddGdGdGdH�dIdJ�Z5dKdL� Z6dMdN� Z7dOdP� Z8e9dQ�dRdS�Z:dTdU� Z;e<dV�dWdX�Z=dYdZ� Z>d[dddddGdGdGd\�d]d^�Z?d_d`� Z@dS )b� N)� GenericAlias) � dataclass�field�Field�FrozenInstanceError�InitVar�MISSING�fields�asdict�astuple�make_dataclass�replace� is_dataclassc @ s e Zd ZdS )r N��__name__�
__module__� __qualname__� r r �0/opt/alt/python39/lib64/python3.9/dataclasses.pyr � � r c @ s e Zd Zdd� ZdS )�_HAS_DEFAULT_FACTORY_CLASSc C s dS )Nz <factory>r ��selfr r r �__repr__� s z#_HAS_DEFAULT_FACTORY_CLASS.__repr__N)r r r r r r r r r � s r c @ s e Zd ZdS )�
_MISSING_TYPENr r r r r r � s r c @ s e Zd Zdd� Zdd� ZdS )� _FIELD_BASEc C s
|| _ d S �N��name�r r r r r �__init__� s 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( e Zd ZdZdd� Zdd� Zdd� ZdS ) r ��typec C s
|| _ d S r r$ )r r% r r r r � s zInitVar.__init__c C s8 t | jt�r"t | jt�s"| jj}n
t| j�}d|� d�S )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 @ s0 e Zd ZdZdd� Zdd� Zdd� Zee�Z dS ) r )
r r% �default�default_factoryr( �hash�init�compare�metadata� _field_typec C sR d | _ d | _|| _|| _|| _|| _|| _|| _|d u r<tnt �
|�| _ d | _ d S r )
r r% r, r- r/ r( r. r0 �_EMPTY_METADATA�types�MappingProxyTyper1 r2 )r r, r- r/ r( r. r0 r1 r r r r � s ��zField.__init__c C sV d| j �d| j�d| j�d| j�d| j�d| j�d| j�d| j�d | 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
,_field_type=�))
r r% r, r- r/ r( r. r0 r1 r2 r r r r r s* �������� �
�zField.__repr__c C s( t t| j�dd �}|r$|| j||� d S )N� __set_name__)�getattrr% r, )r �ownerr �funcr r r r8 s zField.__set_name__N)
r r r r+ r r r8 � classmethodr r* r r r r r � s
r c @ s e Zd ZdZdd� Zdd� ZdS )�_DataclassParams�r/ r( �eq�order� unsafe_hash�frozenc C s( || _ || _|| _|| _|| _|| _d S r r> )r r/ r( r? r@ rA rB r r r r , s z_DataclassParams.__init__c
C s6 d| j �d| j�d| j�d| j�d| j�d| j�d�
S )Nz_DataclassParams(init=r6 z,eq=z,order=z
,unsafe_hash=z,frozen=r7 r> r r r r r 4 s ������z_DataclassParams.__repr__N)r r r r+ r r r r r r r= # s r= T�r, r- r/ r( r. r0 r1 c C s, | t ur|t urtd��t| ||||||�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.
It is an error to specify both default and default_factory.
z/cannot specify both default and default_factory)r �
ValueErrorr rC r r r r B s
�r c s( |sdS dd� � fdd�|D ��� d�S )N�()�(�,c s g | ]}� � d |j � ��qS )�.r ��.0�f��obj_namer r �
<listcomp>a r z _tuple_str.<locals>.<listcomp>�,))�join)rM r r rL r �
_tuple_strX s rQ c s" t � � t���� �fdd��}|S )Nc sN t | �t�� f}|� v r dS � �|� z�| �}W � �|� n � �|� 0 |S )Nz...)�id�_thread� get_ident�add�discard)r �key�result�� repr_running�
user_functionr r �wrapperk s
z _recursive_repr.<locals>.wrapper)�set� functools�wraps)r[ r\ r rY r �_recursive_reprf s
r` )�globals�locals� return_typec
C s� |d u r i }d|vr t |d< d}|tur4||d< d}d�|�}d�dd� |D ��}d | � d
|� d |� d |� �}d
�|�� �}d|� d|� d| � �}i } t||| � | d f i |��S )NZBUILTINS� Z _return_typez->_return_typerG �
c s s | ]}d |� �V qdS )z Nr )rJ �br r r � <genexpr>� r z _create_fn.<locals>.<genexpr>z def rF r7 z:
�, zdef __create_fn__(z):
z
return Z
__create_fn__)�builtinsr rP �keys�exec)
r �args�bodyra rb rc Zreturn_annotationZtxtZ
local_vars�nsr r r �
_create_fny s
ro c C s0 | r d|� d|�d|� d�S |� d|� d|� �S )Nz BUILTINS.object.__setattr__(rG r7 rH �=r )rB r �value� self_namer r r �
_field_assign� s rs c C s� d| j � �}| jturV| jr@| j||< |� d| j � d| j � �}q�| j||< |� d�}n8| jr�| jtu rn| j }q�| jtur�| j||< | j }nd S | jtu r�d S t|| j ||�S )NZ_dflt_z() if z is _HAS_DEFAULT_FACTORY else rE )r r- r r/ r, r2 r# rs )rK rB ra rr Z default_namerq r r r � _field_init� s*
��
rt c C sV | j tu r| jtu rd}n&| j tur2d| j� �}n| jtur@d}| j� d| j� |� �S )Nrd z=_dflt_z=_HAS_DEFAULT_FACTORYz:_type_)r, r r- r )rK r, r r r � _init_param� s
ru c C s� d}| D ]:}|j r|jtu r&|jtu s,d}q|rtd|j�d���qdd� | D �}|�ttd�� g }| D ] }t||||�} | rj|� | � qj|r�d�
d d
� | D ��}
|� |� d t � d |
� d
�� |s�dg}t d|gdd� | D � |||d d�S )NFTznon-default argument z follows default argumentc S s i | ]}d |j � �|j�qS )Z_type_)r r% rI r r r �
<dictcomp>� r z _init_fn.<locals>.<dictcomp>)r �_HAS_DEFAULT_FACTORYrG c s s | ]}|j tu r|jV qd S r )r2 r# r rI r r r rg s
�z_init_fn.<locals>.<genexpr>rH rF r7 �passr c S s g | ]}|j rt|��qS r )r/ ru rI r r r rN r z _init_fn.<locals>.<listcomp>)rb ra rc )
r/ r, r r- � TypeErrorr �updaterw rt �appendrP �_POST_INIT_NAMEro ) r rB �
has_post_initrr ra Z seen_defaultrK rb Z
body_lines�lineZ
params_strr r r �_init_fn� s: � �r c C s2 t dddd�dd� | D �� d g|d�}t|�S ) Nr r z(return self.__class__.__qualname__ + f"(rh c S s g | ]}|j � d |j � d��qS )z={self.z!r}r rI r r r rN s �z _repr_fn.<locals>.<listcomp>z)"�ra )ro rP r` )r ra �fnr r r �_repr_fn s
����r� c C sp | t d�}|r,dd�dd� |D �� d }nd}tdd d
|� d �d d
f||d�tddd
|� d �ddf||d�fS )N)r) r rF rG c s s | ]}t |j�V qd S r )r( r rI r r r rg * r z'_frozen_get_del_attr.<locals>.<genexpr>rO rE � __setattr__)r r rq 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))rb ra � __delattr__r z; raise FrozenInstanceError(f"cannot delete field {name!r}")z"super(cls, self).__delattr__(name))r rP ro )r) r ra rb Z
fields_strr r r �_frozen_get_del_attr& s2 �
��
���r� c C s$ t | ddd|� |� |� �dg|d�S )N)r �otherz%if other.__class__ is self.__class__:z return zreturn NotImplementedr� )ro )r �op�
self_tuple� other_tuplera r r r �_cmp_fn? s ��r� c C s$ t d| �}tddd|� d�g|d�S )Nr �__hash__r z return hash(r7 r� )rQ ro )r ra r� r r r �_hash_fnM s
�r� c C s$ | |j u p"t| �|ju o"| j|j u S r )�ClassVarr% �
_GenericAlias�
__origin__)�a_type�typingr r r � _is_classvarU s
�r� c C s | |j u pt| �|j u S r )r r% )r� � dataclassesr r r � _is_initvar] s
�r� c C s� t �| �}|r�d }|�d�}|s2tj�|j�j}n2tj�|j�}|rd|j�|�|u rdtj�|j�j}|r�||�|�d��|�r�dS dS )N� � TF)�_MODULE_IDENTIFIER_RE�match�group�sys�modules�getr �__dict__) �
annotationr) Za_moduler� Zis_type_predicater� rn Z module_name�moduler r r �_is_typed s )
r� c C s8 t | |t�}t|t�r |}nt|tj�r,t}t|d�}||_||_t |_
t j �
d�}|r�t||�s�t|jt�r�t|j| ||jt�r�t|_
|j
t u r�t j t }t||�s�t|jt�r�t|j| ||jt�r�t|_
|j
ttfv r�|jtur�td|j� d���|j
t u �r4t|jttt f��r4t dt|j�� d|j� d���|S )N)r, r� �field z cannot have a default factoryzmutable default z for field z$ is not allowed: use default_factory) r9 r r' r r4 �MemberDescriptorTyper r r% r! r2 r� r� r� r� �strr� r� r" r r� r r# r- ry r, �list�dictr] rD )r) Za_namer� r, rK r� r� r r r �
_get_field� sJ
���
���
�
r� c C s || j v rdS t| ||� dS )NTF)r� �setattr)r) r rq r r r �_set_new_attribute� s
r� c C s d S r r �r) r ra r r r �_hash_set_none� s r� c C s dd� |D �}t ||�S )Nc S s( g | ] }|j d u r|jrn|j r|�qS r )r. r0 rI r r r rN r z _hash_add.<locals>.<listcomp>)r� )r) r ra �fldsr r r � _hash_add s r� c C s t d| j� ���d S )Nz-Cannot overwrite attribute __hash__ in class )ry r r� r r r �_hash_exception s �r� ))FFFF)FFFT)FFTF)FFTT)FTFF)FTFT)FTTF)FTTT)TFFF)TFFT)TFTF)TFTT)TTFF)TTFT)TTTF)TTTTc s� i }� j tjv r tj� j j}ni }t� tt||||||�� d} d}
� jddd� D ]D} t| t d �} | d urVd}
| �
� D ]}
|
||
j < qzt| t�j rVd} qV� j�
di �}� fdd�|�� D �}|D ]L}
|
||
j < tt� |
j d �t�r�|
jtu �r t� |
j � q�t� |
j |
j� qĈ j�� D ].\}}t|t��r ||v�r t|�d����q |
�rz| �rf|�sftd ��| �sz|�rztd
��t� t |� � j�
d t�}|tu �p�|d u �o�d � jv }|�r�|�s�td
��|�rt� t�}dd� |�
� D �}t� dt|||d|v �rdnd|�� dd� |�
� D �}|�rHdd� |D �}t� dt||�� |�r�dd� |D �}td|�}td|�}t� d t d d|||d�� |�r�dd� |D �}td|�}td|�}dD ]>\}}t� |t |||||d���r�td|� d � j � d ����q�|�r8t � ||�D ].}t� |j |��rtd|j � d � j � ����qt t |�t |�t |�|f }|�rh|� ||�� _!t� d ��s�� j t"t#�$� ���%d d � � _&� S )!NF���r T�__annotations__c s g | ]\}}t � ||��qS r )r� )rJ r r% �r) r r rN _ s �z"_process_class.<locals>.<listcomp>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 truec S s g | ]}|j ttfv r|�qS r )r2 r! r# rI r r r rN � s �r r Z__dataclass_self__c S s g | ]}|j tu r|�qS r �r2 r! rI r r r rN � r c S s g | ]}|j r|�qS r )r( rI r r r rN � r r c S s g | ]}|j r|�qS r �r0 rI r r r rN � r r� z==r� c S s g | ]}|j r|�qS r r� rI r r r rN � r ))�__lt__�<)�__le__z<=)�__gt__�>)�__ge__z>=zCannot overwrite attribute z
in class z). Consider using functools.total_ordering�__doc__z -> Nonerd )'r r� r� r� r� �_PARAMSr= �__mro__r9 �_FIELDS�valuesr rB r� �itemsr' r r, r �delattrry rD �hasattrr| r� r r� rQ r� r r� � _hash_action�boolr� r� �inspectZ signaturer
r� ) r) r/ r( r? r@ rA rB r ra Zany_frozen_baseZhas_dataclass_basesrf Z base_fieldsrK Zcls_annotationsZ
cls_fieldsr rq Z
class_hashZhas_explicit_hashr} r� Z
field_listr� r� r� r� Z hash_actionr r� r �_process_class) s� �
�
�
��
��
��
� � � �r� Fr> c s* � �����fdd�}| 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.
c s t | ��� ����S r )r� r� �r? rB r/ r@ r( rA r r �wrap� s zdataclass.<locals>.wrapNr )r) r/ r( r? r@ rA rB r� r r� r r � s r c C s@ zt | t�}W n ty( td��Y n0 tdd� |�� D ��S )z�Return 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 instancec s s | ]}|j tu r|V qd S r r� rI r r r rg r zfields.<locals>.<genexpr>)r9 r� �AttributeErrorry �tupler� )Zclass_or_instancer r r r r s
r c C s t t| �t�S )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 | t�rt | t�s| nt| �}t|t�S )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 s t | �std��t| |�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� ry �
_asdict_inner)r� r� r r r r
s r
c s� t | �rDg }t| �D ]&}tt| |j�� �}|�|j|f� q� |�S t| t�rrt| d�rrt | �� fdd�| D �� S t| t
tf�r�t | �� fdd�| D ��S t| t �r�t | �� fdd�| � � D ��S t
�| �S d S )N�_fieldsc s g | ]}t |� ��qS r �r� �rJ �vr� r r rN Q r z!_asdict_inner.<locals>.<listcomp>c 3 s | ]}t |� �V qd S r r� r� r� r r rg V r z _asdict_inner.<locals>.<genexpr>c 3 s& | ] \}}t |� �t |� �fV qd S r r� �rJ �kr� r� r r rg X s ��)r� r r� r9 r r{ r' r� r� r% r� r� r� �copy�deepcopy)r� r� rX rK rq r r� r r� 6 s
�r� ��
tuple_factoryc C s t | �std��t| |�S )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� ry �_astuple_inner)r� r� r r r r _ s r c s� t | �r>g }t| �D ] }tt| |j�� �}|�|� q� |�S t| t�rlt| d�rlt | �� fdd�| D �� S t| t
tf�r�t | �� fdd�| D ��S t| t �r�t | �� fdd�| � � D ��S t
�| �S d S )Nr� c s g | ]}t |� ��qS r �r� r� r� r r rN � r z"_astuple_inner.<locals>.<listcomp>c 3 s | ]}t |� �V qd S r r� r� r� r r rg � r z!_astuple_inner.<locals>.<genexpr>c 3 s& | ] \}}t |� �t |� �fV qd S r r� r� r� r r rg � s �)r� r r� r9 r r{ r' r� r� r% r� r� r� r� r� )r� r� rX rK rq r r� r r� w s
�r� r )�bases� namespacer/ r( r? r@ rA rB c s � du ri � n� � � � t� }
i } |D ]�} t| t�r<| }
d}nDt| �dkrR| \}
}n.t| �dkrr| \}
}}|� |
< ntd| ����t|
t�r�|
�� s�td|
����t�|
�r�td|
����|
|
v r�td|
����|
� |
� || |
< q$| � d < t
� | |i � fd
d ��}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: r� c s
| � � �S r )rz )rn �r� r r �<lambda>� r z make_dataclass.<locals>.<lambda>r> )
r� r] r' r� �lenry � isidentifier�keyword� iskeywordrU r4 � new_classr )�cls_namer r� r� r/ r( r? r@ rA rB �seenZanns�itemr �tp�specr) r r� r r � s8
�r c K s� t | �std��t| t��� D ]v}|jtu r.q |jsR|j|v r t d|j� d���q |j|vr |jt
u r�|j t u r�t d|j�d���t| |j�||j< q | j
f i |��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())r� ry r9 r� r� r2 r" r/ r rD r# r, r � __class__)r� ZchangesrK r r r r
� s
r
)N)A�rer� r� r4 r� r� ri r^ rS r �__all__r� r r rw r r r5 r3 r r! r" r# r� r� r| �compiler� r r r= r rQ r` ro rs rt ru 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 �<module> s�
< � � 62 ;R� >
� )� B