File: //usr/lib/python3/dist-packages/pyrsistent/__pycache__/_pclass.cpython-38.pyc
U
g �\�% � @ s� d dl Z d dlmZmZmZmZ d dlmZmZm Z m
Z
m Z m Z d dl
mZ dd� ZG dd� de�Ze� Zd d
� Ze �e�G d d � d e��ZG d
d� de�ZdS )� N)�InvariantException� CheckedType�_restore_pickle�store_invariants)�
set_fields�
check_type� is_field_ignore_extra_complaint�PFIELD_NO_INITIAL� serialize�check_global_invariants�� transformc C s t | �dko| d tkS )N� r )�lenr )�bases� r �4/usr/lib/python3/dist-packages/pyrsistent/_pclass.py�
_is_pclass s r c s e Zd Z� fdd�Z� ZS )�
PClassMetac sh t ||dd� t||dd� dtdd� |d D �� |d< t|�rR|d d 7 < tt| ��| |||�S )
N�_pclass_fields)�name�_pclass_invariantsZ
__invariant__)�_pclass_frozenc s s | ]
}|V qd S �Nr ��.0�keyr r r � <genexpr> s z%PClassMeta.__new__.<locals>.<genexpr>� __slots__)� __weakref__)r r �tupler �superr �__new__)Zmcsr r Zdct�� __class__r r r" s zPClassMeta.__new__)�__name__�
__module__� __qualname__r" �
__classcell__r r r# r r
s r c C s< t | |||� |�|�\}}|s,|�|� n t|||� d S r )r Z invariant�append�setattr)�cls�fieldr �value�result�invariant_errorsZis_okZ
error_coder r r �_check_and_set_attr s
r0 c s� e Zd ZdZ� fdd�Zdd� Zed"dd ��Zd#d
d �Zd d
� Z dd� Z
dd� Z dd� Z � fdd�Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z� ZS )$�PClassa�
A PClass is a python class with a fixed set of specified fields. PClasses are declared as python classes inheriting
from PClass. It is defined the same way that PRecords are and behaves like a PRecord in all aspects except that it
is not a PMap and hence not a collection but rather a plain Python object.
More documentation and examples of PClass usage is available at https://github.com/tobgu/pyrsistent
c sR t t| ��| �}|�dd �}|�dd �}g }g }| j�� D ]�\}}||kr�|d ksZ||kr�tt||�rz|j|| |d�} q�|�|| �} n|| } t| ||| ||� ||= q:|j t
k r�t |j �r�|� � n|j }
t| |||
||� q:|j r:|�
d�| j|�� q:|�s
|�r tt|�t|�d��|�r<td�d�|�| j���t|| j� d|_|S ) N�_factory_fields� ignore_extra)r3 z{0}.{1}zField invariant failedz0'{0}' are not among the specified fields for {1}�, T)r! r1 r" �popr �itemsr �factoryr0 �initialr �callableZ mandatoryr) �formatr% r r �AttributeError�joinr r r ) r+ �kwargsr. �factory_fieldsr3 Zmissing_fieldsr/ r r, r- r8 r# r r r" 0 sH
� � zPClass.__new__c O s` |r|d ||d < t |�}| jD ](}||kr"t| |t�}|tk r"|||< q"| jf d|i|��S )a
Set a field in the instance. Returns a new instance with the updated value. The original instance remains
unmodified. Accepts key-value pairs or single string representing the field name and a value.
>>> from pyrsistent import PClass, field
>>> class AClass(PClass):
... x = field()
...
>>> a = AClass(x=1)
>>> a2 = a.set(x=2)
>>> a3 = a.set('x', 3)
>>> a
AClass(x=1)
>>> a2
AClass(x=2)
>>> a3
AClass(x=3)
r r r2 )�setr �getattr�_MISSING_VALUEr$ )�self�argsr= r> r r- r r r r? T s
z
PClass.setNFc s: t � | �r� S |r&� fdd�| jD �� | f ||d�� ��S )a:
Factory method. Will create a new PClass 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 PClass.
c s i | ]}|� kr|� | �qS r r )r �k�r= r r �
<dictcomp>� s z!PClass.create.<locals>.<dictcomp>)r2 r3 )�
isinstancer )r+ r= r2 r3 r rE r �createt s
z
PClass.createc C s@ i }| j D ]0}t| |t�}|tk r
t| j | j||�||< q
|S )z�
Serialize the current PClass using custom serializer functions for fields where
such have been supplied.
)r r@ rA r
Z
serializer)rB r: r. r r- r r r r
� s
zPClass.serializec G s
t | |�S )aT
Apply transformations to the currency PClass. For more details on transformations see
the documentation for PMap. Transformations on PClasses do not support key matching
since the PClass is not a collection. Apart from that the transformations available
for other persistent types work as expected.
r )rB Ztransformationsr r r r
� s zPClass.transformc C s> t || j�r:| jD ]"}t| |t�t||t�kr dS qdS tS )NFT)rG r$ r r@ rA �NotImplemented)rB �otherr r r r �__eq__� s
z
PClass.__eq__c C s
| |k S r r )rB rJ r r r �__ne__� s z
PClass.__ne__c s t t� fdd�� jD ���S )Nc 3 s | ]}|t � |t�fV qd S r )r@ rA r �rB r r r � s z"PClass.__hash__.<locals>.<genexpr>)�hashr r rM r rM r �__hash__� s zPClass.__hash__c s2 t | dd�r td�||���tt| ��||� d S )Nr Fz'Can't set attribute, key={0}, value={1})r@ r; r: r! r1 � __setattr__�rB r r- r# r r rP � s zPClass.__setattr__c C s t d�|���d S )Nz-Can't delete attribute, key={0}, use remove())r; r: )rB r r r r � __delattr__� s zPClass.__delattr__c C s0 i }| j D ] }t| |t�}|tk r
|||< q
|S r )r r@ rA )rB r. r r- r r r �_to_dict� s
zPClass._to_dictc C s( d� | jjd�dd� | �� �� D ���S )Nz{0}({1})r4 c s s" | ]\}}d � |t|��V qdS )z{0}={1}N)r: �repr)r rD �vr r r r � s z"PClass.__repr__.<locals>.<genexpr>)r: r$ r% r< rS r6 rM r r r �__repr__� s
�zPClass.__repr__c s& t � fdd�� jD ��}t� j|ffS )Nc 3 s& | ] }t � |�r|t� |�fV qd S r )�hasattrr@ r rM r r r � s
z$PClass.__reduce__.<locals>.<genexpr>)�dictr r r$ )rB �datar rM r �
__reduce__� s zPClass.__reduce__c C s t | | �� �S )z5
Returns an evolver for this object.
)�_PClassEvolverrS rM r r r �evolver� s zPClass.evolverc C s | � � }||= |�� S )z�
Remove attribute given by name from the current instance. Raises AttributeError if the
attribute doesn't exist.
)r\ �
persistent)rB r r\ r r r �remove� s z
PClass.remove)NF)N)r% r& r' �__doc__r" r? � classmethodrH r
r
rK rL rO rP rR rS rV rZ r\ r^ r( r r r# r r1 & s" $
r1 c s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
� fdd�Z dd� Z � Z
S )r[ )�_pclass_evolver_original�_pclass_evolver_data� _pclass_evolver_data_is_dirtyr2 c C s || _ || _d| _t� | _d S )NF)ra rb rc r? r2 )rB ZoriginalZ initial_dictr r r �__init__� s z_PClassEvolver.__init__c C s
| j | S r )rb �rB �itemr r r � __getitem__� s z_PClassEvolver.__getitem__c C s2 | j �|t�|k r.|| j |< | j�|� d| _| S �NT)rb �getrA r2 �addrc rQ r r r r? � s
z_PClassEvolver.setc C s | � ||� d S r )r? rQ r r r � __setitem__� s z_PClassEvolver.__setitem__c C s4 || j kr(| j |= | j�|� d| _| S t|��d S rh )rb r2 �discardrc r; re r r r r^ � s
z_PClassEvolver.removec C s | � |� d S r )r^ re r r r � __delitem__� s z_PClassEvolver.__delitem__c C s& | j r | jjf d| ji| j��S | jS )Nr2 )rc ra r$ r2 rb rM r r r r] � s
�z_PClassEvolver.persistentc s. || j kr| �||� ntt| ��||� d S r )r r? r! r[ rP rQ r# r r rP s
z_PClassEvolver.__setattr__c C s | | S r r re r r r � __getattr__ s z_PClassEvolver.__getattr__)r% r& r' r rd rg r? rk r^ rm r] rP rn r( r r r# r r[ � s r[ )ZsixZpyrsistent._checked_typesr r r r Zpyrsistent._field_commonr r r r r
r Zpyrsistent._transformationsr
r �typer �objectrA r0 Z
add_metaclassr1 r[ r r r r �<module> s
1