B
���[�$ � @ s� d dl Z d dlmZmZmZmZ d dlmZmZm Z m
Z
m Z d dl m
Z
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�PFIELD_NO_INITIAL� serialize�check_global_invariants)� transformc C s t | �dko| d tkS )N� r )�lenr )�bases� r �E/opt/alt/python37/lib64/python3.7/site-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 � s z%PClassMeta.__new__..� __slots__)� __weakref__)r r �tupler �superr �__new__)Zmcsr r �dct)� __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 )N)r Z invariant�append�setattr)�cls�fieldr �value�result�invariant_errorsZis_okZ
error_coder r r �_check_and_set_attr s
r, 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
s& t t| ��| �}|�dd �}g }g }x�| j�� D ]�\}}||kr�|d ksP||kr`|�|| �}n|| }t| |||||� ||= q0|jt k r�t
|j�r�|�� n|j} t| ||| ||� q0|j r0|� d�
| j|�� q0W |s�|r�tt|�t|�d��|�rtd�
d�|�| j���t|| j� d|_|S )N�_factory_fieldsz{0}.{1}zField invariant failedz0'{0}' are not among the specified fields for {1}z, T)r r- r �popr �items�factoryr, �initialr �callable� mandatoryr% �formatr! r r �AttributeError�joinr
r r )
r' �kwargsr* �factory_fieldsZmissing_fieldsr+ r r( r) r2 )r r r r . s2
zPClass.__new__c O sd |r|d ||d < t |�}x2| jD ](}||kr$t| |t�}|tk r$|||< q$W | 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 r. )�setr �getattr�_MISSING_VALUEr )�self�argsr8 r9 r r) r r r r: N s z
PClass.setNFc s8 t � | �r� S |r&� fdd�| jD �� | f d|i� ��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)r8 r r �
{ s z!PClass.create..r. )�
isinstancer )r' r8 r. Z ignore_extrar )r8 r �createn s
z
PClass.createc C sD i }x:| j D ]0}t| |t�}|tk r t| j | j||�||< q W |S )z�
Serialize the current PClass using custom serializer functions for fields where
such have been supplied.
)r r; r< r Z
serializer)r= r5 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 )r= Ztransformationsr r r r � s zPClass.transformc C s@ t || j�r.)�hashr r )r= r )r= 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; r6 r5 r r- � __setattr__)r= r r) )r r r rI � s zPClass.__setattr__c C s t d�|���d S )Nz-Can't delete attribute, key={0}, use remove())r6 r5 )r= r r r r � __delattr__� s zPClass.__delattr__c C s4 i }x*| j D ] }t| |t�}|tk r |||< q W |S )N)r r; r< )r= r* r r) r r r �_to_dict� s zPClass._to_dictc C s( d� | jjd�dd� | �� �� D ���S )Nz{0}({1})z, c s s" | ]\}}d � |t|��V qdS )z{0}={1}N)r5 �repr)r r? �vr r r r � s z"PClass.__repr__..)r5 r r! r7 rK r0 )r= r r r �__repr__� s
zPClass.__repr__c s& t � fdd�� jD ��}t� j|ffS )Nc 3 s& | ] }t � |�r|t� |�fV qd S )N)�hasattrr; )r r )r= r r r � s z$PClass.__reduce__..)�dictr r r )r= �datar )r= r �
__reduce__� s zPClass.__reduce__c C s t | | �� �S )z5
Returns an evolver for this object.
)�_PClassEvolverrK )r= 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.
)rT �
persistent)r= r rT r r r �remove� s z
PClass.remove)NF)N)r! r" r# �__doc__r r: � classmethodrB r r rE rF rH rI rJ rK rN rR rT rV r$ r r )r r r- $ s"
r- c s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
� fdd�Z dd� Z � Z
S )rS )�_pclass_evolver_original�_pclass_evolver_data� _pclass_evolver_data_is_dirtyr. c C s || _ || _d| _t� | _d S )NF)rY rZ r[ r: r. )r= �originalZ initial_dictr r r �__init__� s z_PClassEvolver.__init__c C s
| j | S )N)rZ )r= �itemr r r � __getitem__� s z_PClassEvolver.__getitem__c C s2 | j �|t�|k r.|| j |< | j�|� d| _| S )NT)rZ �getr< r. �addr[ )r= r r) r r r r: � s
z_PClassEvolver.setc C s | � ||� d S )N)r: )r= r r) r r r � __setitem__� s z_PClassEvolver.__setitem__c C s4 || j kr(| j |= | j�|� d| _| S t|��d S )NT)rZ r. �discardr[ r6 )r= r^ r r r rV � s
z_PClassEvolver.removec C s | � |� d S )N)rV )r= r^ r r r � __delitem__� s z_PClassEvolver.__delitem__c C s& | j r | jjf d| ji| j��S | jS )Nr. )r[ rY r r. rZ )r= r r r rU � s
z_PClassEvolver.persistentc s. || j kr| �||� ntt| ��||� d S )N)r r: r rS rI )r= r r) )r r r rI � s
z_PClassEvolver.__setattr__c C s | | S )Nr )r= r^ r r r � __getattr__ s z_PClassEvolver.__getattr__)r! r" r# r r] r_ r: rb rV rd rU rI re r$ r r )r r rS � s rS )�sixZpyrsistent._checked_typesr r r r Zpyrsistent._field_commonr r r r r
Zpyrsistent._transformationsr r �typer �objectr< r, Z
add_metaclassr- rS r r r r � s
-