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 )Nz/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�}| d u r"|S || �S )Nc s t | ��� ����S r )r� r� �r? rB r/ r@ r( rA r r �wrap� s zdataclass.<locals>.wrapr )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 )Nz0must 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 r )r� r% r� )�objr r r �_is_dataclass_instance s r� c C s* t | t�rt | t�s| nt| �}t|t�S r )r' r% r r� r� )r� r) r r r r s r �� dict_factoryc C s t | �std��t| |�S )Nz0asdict() 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 )Nz1astuple() 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 � d u 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 ) 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 )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� 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