B
+�[4* � @ sR d dl Z d dlmZmZmZmZmZmZmZm Z m
Z
d dlm Z d dlm
Z
d dlmZ dd� Zdd � Zd
d � Zd d
� ZG dd� de�ZdZdd� Zdd� Ze� Zdd� Zdd� Zeeedeefdd�Zdd� Z G dd � d e �Z ed ed iZ i Z d d � Z!d!d"� Z"d#d$� Z#d%d&� Z$d2d'd(�Z%d3d)d*�Z&d+d� Z'i Z(d,d-� Z)d.d/� Z*defd0d1�Z+dS )4� N) � CheckedPMap� CheckedPSet�CheckedPVector� CheckedType�InvariantException�_restore_pickle�get_type�maybe_parse_user_type�maybe_parse_many_user_types)�optional)�wrap_invariant)�Enumc sZ t t� fdd�|D �g ��| � < x4t| �� �D ]$\}}t|t�r.|| � |< | |= q.W d S )Nc s" g | ]}t |j�� i ��� ��qS � )�list�__dict__�get�items)�.0�b)�namer �K/opt/alt/python37/lib64/python3.7/site-packages/pyrsistent/_field_common.py�
s z set_fields..)�dict�sumr r �
isinstance�_PField)�dct�basesr �k�vr )r r �
set_fields s
r c s4 t dd� � fdd�|D �D ��}|r0t|dd��d S )Nc s s | ]\}}|s|V qd S )Nr )r Zis_okZ
error_coder r r � s z*check_global_invariants..c 3 s | ]}|� �V qd S )Nr )r � invariant)�subjectr r r! s r zGlobal invariant failed)�tupler )r# Z
invariantsZ error_codesr )r# r �check_global_invariants s r% c C s& t |t�r | tkr |�|�S | ||�S )N)r r �PFIELD_NO_SERIALIZER� serialize)�
serializer�format�valuer r r r' # s
r' c sN |j rJt� fdd�|j D ��sJt � �}d�| j||j�}t| ||j ||��d S )Nc 3 s | ]}t � t|��V qd S )N)r r )r �t)r* r r r! + s z check_type..z'Invalid type for field {0}.{1}, was {2})�type�anyr) �__name__�
PTypeError)Zdestination_cls�fieldr r* � actual_type�messager )r* r �
check_type* s r3 c @ s$ e Zd ZdZdd� Zedd� �ZdS )r )r, r" �initial� mandatory�_factoryr( c C s( || _ || _|| _|| _|| _|| _d S )N)r, r" r4 r5 r6 r( )�selfr, r" r4 r5 �factoryr( r r r �__init__4 s z_PField.__init__c C s@ | j tkr:t| j�dkr:tt| j�d �}t|t�r:|jS | j S )N� r ) r6 �PFIELD_NO_FACTORY�lenr, r r$ �
issubclassr �create)r7 �typr r r r8 < s
z_PField.factoryN)r. �
__module__� __qualname__� __slots__r9 �propertyr8 r r r r r 1 s r r c C s dS )N)TNr )�_r r r �G � rE c C s | S )Nr )�xr r r rE H rF c C s |S )Nr )rD r* r r r rE J rF c C s |S )Nr )rD r* r r r rE K rF Fc C sf t | tttf�r tt| ��}n tt| ��}|tkrBt|�rBt|�n|}t ||||||d�}t
|� |S )a
Field specification factory for :py:class:`PRecord`.
:param type: a type or iterable with types that are allowed for this field
:param invariant: a function specifying an invariant that must hold for the field
:param initial: value of field if not specified when instantiating the record
:param mandatory: boolean specifying if the field is mandatory or not
:param factory: function called when field is set.
:param serializer: function that returns a serialized version of the field
)r, r" r4 r5 r8 r( ) r r �setr$ r
r �PFIELD_NO_INVARIANT�callabler r �_check_field_parameters) r, r" r4 r5 r8 r( �typesZinvariant_functionr0 r r r r0 M s r0 c s� x6� j D ],}t|t �st|tj�std�t |����qW � jtk r~t� j�s~� j r~t � fdd�� j D ��s~td�t � j����t� j
�s�td��t� j �s�td��t� j �s�td��d S )Nz Type parameter expected, not {0}c 3 s | ]}t � j|�V qd S )N)r r4 )r r+ )r0 r r r! x s z*_check_field_parameters..z Initial has invalid type {0}zInvariant must be callablezFactory must be callablezSerializer must be callable)
r, r �six� string_types� TypeErrorr) r4 �PFIELD_NO_INITIALrJ r- r" r8 r( )r0 r+ r )r0 r rK q s
rK c s e Zd ZdZ� fdd�Z� ZS )r/ a
Raised when trying to assign a value with a type that doesn't match the declared type.
Attributes:
source_class -- The class of the record
field -- Field name
expected_types -- Types allowed for the field
actual_type -- The non matching type
c s. t t| �j||� || _|| _|| _|| _d S )N)�superr/ r9 � source_classr0 �expected_typesr1 )r7 rR r0 rS r1 �args�kwargs)� __class__r r r9 � s
zPTypeError.__init__)r. r@ rA �__doc__r9 �
__classcell__r r )rV r r/ � s r/ ZPVectorZPSetc C s t | |f }t||�S )z=Unpickling function for auto-generated PVec/PSet field types.)�_seq_field_typesr )�
checked_class� item_type�data�type_r r r �_restore_seq_field_pickle� s r^ c C s d� dd� | D ��S )z4Convert a tuple of types to a human-readable string.� c s s | ]}t |�j�� V qd S )N)r r. �
capitalize)r r? r r r r! � s z"_types_to_names..)�join)rL r r r �_types_to_names� s rb c sX t �� �f�}|dk r|S G � �fdd�d� �}t� }t|j�| |_|t � �f< |S )zFCreate a subclass of the given checked class with the given item type.Nc s e Zd Z�Z� �fdd�ZdS )z%_make_seq_field_type..TheTypec s t � �t| �ffS )N)r^ r )r7 )rZ r[ r r �
__reduce__� s z0_make_seq_field_type..TheType.__reduce__N)r. r@ rA Z__type__rc r )rZ r[ r r �TheType� s rd )rY r �SEQ_FIELD_TYPE_SUFFIXESrb Z_checked_typesr. )rZ r[ r] rd �suffixr )rZ r[ r �_make_seq_field_type� s rg c sB t | |�� |r � fdd�}n� j}t|r0t� �n� |d||�d�S )a�
Create checked field for either ``PSet`` or ``PVector``.
:param checked_class: ``CheckedPSet`` or ``CheckedPVector``.
:param item_type: The required type for the items in the set.
:param optional: If true, ``None`` can be used as a value for
this field.
:param initial: Initial value to pass to factory.
:return: A ``field`` containing a checked class.
c s | d kr d S � � | �S d S )N)r> )�argument)rd r r r8 � s z _sequence_field..factoryT)r, r8 r5 r4 )rg r> r0 �
optional_type)rZ r[ r r4 r8 r )rd r �_sequence_field� s
rj c C s t t| ||�S )al
Create checked ``PSet`` field.
:param item_type: The required type for the items in the set.
:param optional: If true, ``None`` can be used as a value for
this field.
:param initial: Initial value to pass to factory if no value is given
for the field.
:return: A ``field`` containing a ``CheckedPSet`` of the given type.
)rj r )r[ r r4 r r r �
pset_field� s rk c C s t t| ||�S )au
Create checked ``PVector`` field.
:param item_type: The required type for the items in the vector.
:param optional: If true, ``None`` can be used as a value for
this field.
:param initial: Initial value to pass to factory if no value is given
for the field.
:return: A ``field`` containing a ``CheckedPVector`` of the given type.
)rj r )r[ r r4 r r r �
pvector_field� s rl c C s dS )N)Tr_ r )�itemr r r rE � rF c C s t | |f }t||�S )z8Unpickling function for auto-generated PMap field types.)�_pmap_field_typesr )�key_type�
value_typer\ r] r r r �_restore_pmap_field_pickle� s rq c sZ t �� �f�}|dk r|S G � �fdd�dt�}d�t|j�t|j��|_|t � �f< |S )zDCreate a subclass of CheckedPMap with the given key and value types.Nc s e Zd Z� Z�Zdd� ZdS )z%_make_pmap_field_type..TheMapc S s t | j| jt| �ffS )N)rq � __key_type__�__value_type__r )r7 r r r rc s z0_make_pmap_field_type..TheMap.__reduce__N)r. r@ rA rr rs rc r )ro rp r r �TheMap s rt z {0}To{1}PMap)rn r r r) rb Z_checked_key_typesZ_checked_value_typesr. )ro rp r] rt r )ro rp r �_make_pmap_field_type s ru c sB t | |�� |r � fdd�}n� j}td� � |r6t� �n� ||d�S )ad
Create a checked ``PMap`` field.
:param key: The required type for the keys of the map.
:param value: The required type for the values of the map.
:param optional: If true, ``None`` can be used as a value for
this field.
:param invariant: Pass-through to ``field``.
:return: A ``field`` containing a ``CheckedPMap``.
c s | d kr d S � � | �S d S )N)r> )rh )rt r r r8 % s zpmap_field..factoryT)r5 r4 r, r8 r" )ru r> r0 ri )ro rp r r" r8 r )rt r �
pmap_field s
rv )Fr )Fr ),rM Zpyrsistent._checked_typesr r r r r r r r r
r ri r Zpyrsistent._compatr
r r% r' r3 �objectr ZPFIELD_NO_TYPErI r; rP r& r0 rK rO r/ re rY r^ rb rg rj rk rl Z_validrn rq ru rv r r r r � sB , #