a
`�]_W � @ s� d dl mZmZmZmZ d dlmZmZmZm Z m
Z
m Z d dl m
Z
mZ G dd� de�ZG dd� de
eed�ZG d d
� d
e
j�Zd S ) � )� CheckedType�_restore_pickle�InvariantException�store_invariants)�
set_fields�
check_type� is_field_ignore_extra_complaint�PFIELD_NO_INITIAL� serialize�check_global_invariants)�PMap�pmapc s e Zd Z� fdd�Z� ZS )� _PRecordMetac sv t ||dd� t||dd� tdd� |d �� D ��|d< tdd� |d �� D ��|d < d
|d < tt| ��| |||�S ) N�_precord_fields)�name�_precord_invariantsZ
__invariant__c s s | ]\}}|j r|V qd S �N)Z mandatory)�.0r �field� r �9/usr/lib64/python3.9/site-packages/pyrsistent/_precord.py� <genexpr> � z'_PRecordMeta.__new__.<locals>.<genexpr>�_precord_mandatory_fieldsc s s& | ] \}}|j tur||j fV qd S r )�initialr )r �kr r r r r r �_precord_initial_valuesr � __slots__)r r �set�items�dict�superr �__new__)Zmcsr �basesZdct�� __class__r r r" s ��z_PRecordMeta.__new__)�__name__�
__module__� __qualname__r" �
__classcell__r r r$ r r s r c s\ e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Zedd d
��Z dd� Z
ddd�Z � Z S )�PRecordau
A PRecord is a PMap with a fixed set of specified fields. Records are declared as python classes inheriting
from PRecord. Because it is a PMap it has full support for all Mapping methods such as iteration and element
access using subscript notation.
More documentation and examples of PRecord usage is available at https://github.com/tobgu/pyrsistent
c s� d|v r,d|v r,t t| ��| |d |d �S |�dd �}|�dd�}|}| jrptdd� | j�� D ��}|�|� t| t � ||d�}|�� D ]\}}|||< q�|�
� S ) N�
_precord_size�_precord_buckets�_factory_fields�
_ignore_extraFc s s( | ] \}}|t |�r|� n|fV qd S r )�callable�r r �vr r r r - s �z"PRecord.__new__.<locals>.<genexpr>�r- r. ) r! r* r" �popr r r �update�_PRecordEvolverr
�
persistent)�cls�kwargsZfactory_fields� ignore_extraZinitial_values�er r1 r$ r r r" ! s �
zPRecord.__new__c s( |r t t| ��|d |d �S | �|�S )a
Set a field in the record. This set function differs slightly from that in the PMap
class. First of all it accepts key-value pairs. Second it accepts multiple key-value
pairs to perform one, atomic, update of multiple fields.
r � )r! r* r r4 )�self�argsr8 r$ r r r 7 s z PRecord.setc C s t | j| �S )z4
Returns an evolver of this object.
)r5 r% �r< r r r �evolverE s zPRecord.evolverc C s$ d� | jjd�dd� | �� D ���S )Nz{0}({1})z, c s s" | ]\}}d � |t|��V qdS )z{0}={1}N)�format�reprr0 r r r r M r z#PRecord.__repr__.<locals>.<genexpr>)r@ r% r& �joinr r> r r r �__repr__K s
�zPRecord.__repr__NFc s: t � | �r� S |r&� fdd�| jD �� | f ||d�� ��S )a<
Factory method. Will create a new PRecord of the current type and assign the values
specified in kwargs.
:param ignore_extra: A boolean which when set to True will ignore any keys which appear in kwargs that are not
in the set of fields on the PRecord.
c s i | ]}|� v r|� | �qS r r )r r �r8 r r �
<dictcomp>\ r z"PRecord.create.<locals>.<dictcomp>r2 )�
isinstancer )r7 r8 r- r9 r rD r �createO s
zPRecord.createc C s t | jt| �ffS r )r r% r r> r r r �
__reduce__` s zPRecord.__reduce__c s t � �fdd���� D ��S )z�
Serialize the current PRecord using custom serializer functions for fields where
such have been supplied.
c 3 s* | ]"\}}|t �j| j� |�fV qd S r )r
r Z
serializerr0 �r@ r< r r r i r z$PRecord.serialize.<locals>.<genexpr>)r r )r< r@ r rI r r
d s zPRecord.serialize)NF)N)
r& r' r( �__doc__r" r r? rC � classmethodrG rH r
r) r r r$ r r* s r* )� metaclassc sB e Zd ZdZd � fdd� Zdd� Z� fdd �Z� fd
d �Z� ZS )
r5 )�_destination_cls�_invariant_error_codes�_missing_fieldsr- r. NFc s2 t t| ��|� || _g | _g | _|| _|| _d S r )r! r5 �__init__rM rN rO r- r. )r< r7 Z
original_pmapr- r. r$ r r rP o s z_PRecordEvolver.__init__c C s | � ||� d S r )r )r< �key�original_valuer r r � __setitem__w s z_PRecordEvolver.__setitem__c
s� | j j�|�}|r�| jd u s&|| jv r�z.tt|| j�rH|j|| jd�}n
|�|�}W q� ty� } z0| j |j
7 _ | j |j 7 _ | W Y d }~S d }~0 0 n|}t
| j |||� |�|�\}}|s�| j �|� tt| ��||�S td�|| j j���d S )N)r9 z/'{0}' is not among the specified fields for {1})rM r �getr- r r* r. �factoryr rN Zinvariant_errorsrO Zmissing_fieldsr Z invariant�appendr! r5 r �AttributeErrorr@ r& )r< rQ rR r �valuer: Zis_okZ
error_coder$ r r r z s$ z_PRecordEvolver.setc s� | j � | �� }tt| ��� }|s*t|� �s<� |j|jd�}n|}� jrt| j t
� fdd�� jt |� � � D ��7 _ | j
s�| j r�tt
| j
�t
| j �d��t|� j� |S )N)r, r+ c 3 s | ]}d � � j|�V qdS )z{0}.{1}N)r@ r& )r �f�r7 r r r � r z-_PRecordEvolver.persistent.<locals>.<genexpr>zField invariant failed)rM �is_dirtyr! r5 r6 rF Z_buckets�_sizer rO �tupler �keysrN r r r )r< r[ Zpm�resultr$ rZ r r6 � s � � z_PRecordEvolver.persistent)NF) r& r' r( r rP rS r r6 r) r r r$ r r5 l s
r5 N)Zpyrsistent._checked_typesr r r r Zpyrsistent._field_commonr r r r r
r Zpyrsistent._pmapr r
�typer r* Z_Evolverr5 r r r r �<module> s
S