Google
Edit File: _field_common.cpython-39.pyc
a `�]_ - � @ sd d dl Z d dlmZmZmZmZmZmZmZm Z m Z d dlm Z d dlm Z d dlZe jd dk Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� ZG dd� de�ZdZdd� Zdd� Ze� Z dd� Z eee dee fdd�Z d d � Z G d d � d e �Z!ed ed!iZ"i Z#d"d#� Z$d$d%� Z%d&d'� Z&d(d)� Z'd5d*d+�Z(d6d,d-�Z)d.d� Z*i Z+d/d0� Z,d1d2� Z-defd3d4�Z.dS )7� N) � CheckedPMap� CheckedPSet�CheckedPVector� CheckedType�InvariantException�_restore_pickle�get_type�maybe_parse_user_type�maybe_parse_many_user_types)�optional)�wrap_invariant� c sV t t� fdd�|D �g ��| � < t| �� �D ]$\}}t|t�r,|| � |< | |= q,d S )Nc s" g | ]}t |j�� i ��� ��qS � )�list�__dict__�get�items)�.0�b��namer �>/usr/lib64/python3.9/site-packages/pyrsistent/_field_common.py� <listcomp> � z set_fields.<locals>.<listcomp>)�dict�sumr r � isinstance�_PField)Zdct�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 � <genexpr> s �z*check_global_invariants.<locals>.<genexpr>c 3 s | ]}|� �V qd S r" r )r � invariant��subjectr r r# r r zGlobal invariant failed)�tupler )r& Z invariantsZ error_codesr r% r �check_global_invariants s �r( c C s& t |t�r | tu r |�|�S | ||�S r" )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 r" )r r �r �t�r- r r r# - r z check_type.<locals>.<genexpr>z'Invalid type for field {0}.{1}, was {2})�type�anyr, �__name__� PTypeError)Zdestination_cls�fieldr r- � actual_type�messager r0 r � check_type, s r8 c C s: t |�tu rdS t|�}t|�dkr(dS tt|d �| �S )NTr F)r1 �setr' �len� issubclassr )�type_clsZ field_type�typesr r r � is_type_cls3 s r> c C sD |sdS t | |j�sdS tr.dt�|j�jv S dt�|j�jv S d S )NF� ignore_extra) r> r1 �PY2�inspectZ getargspec�factory�argsZ signature� parameters)r< r5 r? r r r � is_field_ignore_extra_complaint< s rE c @ s$ e Zd ZdZdd� Zedd� �ZdS )r �r1 r$ �initial� mandatory�_factoryr+ c C s( || _ || _|| _|| _|| _|| _d S r" rF )�selfr1 r$ rG rH rB r+ r r r �__init__N s z_PField.__init__c C s@ | j tu r:t| j�dkr:tt| j�d �}t|t�r:|jS | j S )N� r ) rI �PFIELD_NO_FACTORYr: r1 r r' r; r �create)rJ �typr r r rB V s z_PField.factoryN)r3 � __module__� __qualname__� __slots__rK �propertyrB r r r r r K s r r c C s dS )N)TNr )�_r r r �<lambda>a r rU c C s | S r" r )�xr r r rU b r c C s |S r" r )rT r- r r r rU d r 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 )r1 r$ rG rH rB r+ ) r r r9 r' r r �PFIELD_NO_INVARIANT�callabler r �_check_field_parameters) r1 r$ rG rH rB r+ r= Zinvariant_functionr5 r r r r5 g s �r5 c s� � j D ]*}t|t �st|t�std�t |����q� jturxt� j�sx� j rxt� fdd�� j D ��sxtd�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 r" )r rG r. �r5 r r r# � r z*_check_field_parameters.<locals>.<genexpr>z Initial has invalid type {0}zInvariant must be callablezFactory must be callablezSerializer must be callable) r1 r �str� TypeErrorr, rG �PFIELD_NO_INITIALrX r2 r$ rB r+ )r5 r/ r rZ r rY � s" ��� rY c s e Zd ZdZ� fdd�Z� ZS )r4 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 s2 t t| �j|i |�� || _|| _|| _|| _d S r" )�superr4 rK � source_classr5 �expected_typesr6 )rJ r_ r5 r` r6 rC �kwargs�� __class__r r rK � s zPTypeError.__init__)r3 rP rQ �__doc__rK � __classcell__r r rb r r4 � s r4 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 rk 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 r" )r r3 � capitalize)r rO r r r r# � r z"_types_to_names.<locals>.<genexpr>)�join)r= r r r �_types_to_names� s ro c sX t �� �f�}|dur|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.<locals>.TheTypec s t � �t| �ffS r" )rk r �rJ �rg rh r r � __reduce__� s �z0_make_seq_field_type.<locals>.TheType.__reduce__N)r3 rP rQ Z__type__rr r rq r r �TheType� s rs )rf r �SEQ_FIELD_TYPE_SUFFIXESro Z_checked_typesr3 )rg rh rj rs �suffixr rq r �_make_seq_field_type� s rv c sD t | |�� |r d� fdd� }n� j}t|r2t� �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. NFc s | d u r d S � j | ||d�S d S )N)�_factory_fieldsr? �rN )�argumentrw r? �rs r r rB � s z _sequence_field.<locals>.factoryT)r1 rB rH rG )NF)rv rN r5 � optional_type)rg rh r rG rB r rz r �_sequence_field� s �r| 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. )r| r �rh r rG r r r � pset_field� s �r~ 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. )r| r r} r r r � pvector_field s �r c C s dS )N)Trl r )�itemr r r rU r c C s t | |f }t||�S )z8Unpickling function for auto-generated PMap field types.)�_pmap_field_typesr )�key_type� value_typeri rj r r r �_restore_pmap_field_pickle s r� c sZ t �� �f�}|dur|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.<locals>.TheMapc S s t | j| jt| �ffS r" )r� � __key_type__�__value_type__r rp r r r rr % s �z0_make_pmap_field_type.<locals>.TheMap.__reduce__N)r3 rP rQ r� r� rr r �r� r� r r �TheMap! s r� z {0}To{1}PMap)r� r r r, ro Z_checked_key_typesZ_checked_value_typesr3 )r� r� rj r� r r� r �_make_pmap_field_type s � r� 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 u r d S � � | �S d S r" rx )ry �r� r r rB ? s zpmap_field.<locals>.factoryT)rH rG r1 rB r$ )r� rN r5 r{ )r� r� r r$ rB r r� r � pmap_field0 s �r� )Fr )Fr )/�sysZpyrsistent._checked_typesr r r r r r r r r r r{ r rA � version_infor@ r! r( r* r8 r> rE �objectr ZPFIELD_NO_TYPErW rM r] r) r5 rY r\ r4 rt rf rk ro rv r| r~ r Z_validr� r� r� r� r r r r �<module> sJ , � $�