U
    ꥡc%                     @   s   d dl mZmZmZmZ d dlmZmZmZm	Z	m
Z
mZ d dlmZ dd ZG dd deZe Zdd	 ZG d
d deedZG dd deZdS )    )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 | dko| d tkS )N   r   )lenr   )bases r   6/tmp/pip-unpacked-wheel-og78g_cf/pyrsistent/_pclass.py
_is_pclass   s    r   c                       s   e Zd Z fddZ  ZS )
PClassMetac                    sh   t ||dd t||dd dtdd |d D  |d< t|rR|d  d	7  < tt| | |||S )
N_pclass_fields)name_pclass_invariantsZ__invariant__)_pclass_frozenc                 s   s   | ]
}|V  qd S Nr   .0keyr   r   r   	<genexpr>   s     z%PClassMeta.__new__.<locals>.<genexpr>	__slots__)__weakref__)r   r   tupler   superr   __new__)Zmcsr   r   dct	__class__r   r   r"      s    zPClassMeta.__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_errorsZis_okZ
error_coder   r   r   _check_and_set_attr   s
    r1   c                       s   e Zd ZdZ fddZdd Zed"dd	Zd#d
dZdd Z	dd Z
dd Zdd Z fddZdd Zdd Zdd Zdd Zdd Zd d! Z  ZS )$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| | }|dd }|dd }g }g }| j D ]\}}||kr|d ksZ||krtt||rz|j|| |d}	q||| }	n|| }	t| |||	|| ||= q:|j	t
k	rt|j	r|	 n|j	}
t| |||
|| q:|jr:|d| j| q:|s
|rtt|t|d|r<tdd|| jt|| j d|_|S )	N_factory_fieldsignore_extra)r4   z{0}.{1}zField invariant failedz0'{0}' are not among the specified fields for {1}, T)r!   r2   r"   popr   itemsr   factoryr1   initialr	   callable	mandatoryr*   formatr&   r   r    AttributeErrorjoinr   r   r   )r,   kwargsr/   factory_fieldsr4   Zmissing_fieldsr0   r   r-   r.   r9   r$   r   r   r"   .   sH    
      zPClass.__new__c                 O   s`   |r|d ||d < t |}| jD ](}||kr"t| |t}|tk	r"|||< q"| jf 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   r3   )setr   getattr_MISSING_VALUEr%   )selfargsr?   r@   r   r.   r   r   r   rA   R   s    

z
PClass.setNFc                    s:   t  | r S |r& fdd| jD  | 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 | ]}| kr| | qS r   r   )r   kr?   r   r   
<dictcomp>   s       z!PClass.create.<locals>.<dictcomp>)r3   r4   )
isinstancer   )r,   r?   r3   r4   r   rG   r   creater   s
    	
zPClass.createc                 C   s@   i }| j D ]0}t| |t}|tk	r
t| j | j||||< q
|S )z
        Serialize the current PClass using custom serializer functions for fields where
        such have been supplied.
        )r   rB   rC   r
   
serializer)rD   r<   r/   r   r.   r   r   r   r
      s    
zPClass.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   )rD   Ztransformationsr   r   r   r      s    zPClass.transformc                 C   s>   t || jr:| jD ]"}t| |tt||tkr dS qdS tS )NFT)rI   r%   r   rB   rC   NotImplemented)rD   otherr   r   r   r   __eq__   s    
zPClass.__eq__c                 C   s
   | |k S r   r   )rD   rM   r   r   r   __ne__   s    zPClass.__ne__c                    s   t t fdd jD S )Nc                 3   s   | ]}|t  |tfV  qd S r   )rB   rC   r   rD   r   r   r      s     z"PClass.__hash__.<locals>.<genexpr>)hashr    r   rP   r   rP   r   __hash__   s    zPClass.__hash__c                    s2   t | ddrtd||tt| || d S )Nr   Fz'Can't set attribute, key={0}, value={1})rB   r=   r<   r!   r2   __setattr__rD   r   r.   r$   r   r   rS      s    zPClass.__setattr__c                 C   s   t d|d S )Nz-Can't delete attribute, key={0}, use remove())r=   r<   )rD   r   r   r   r   __delattr__   s    zPClass.__delattr__c                 C   s0   i }| j D ] }t| |t}|tk	r
|||< q
|S r   )r   rB   rC   )rD   r/   r   r.   r   r   r   _to_dict   s    

zPClass._to_dictc                 C   s(   d | jjddd |   D S )Nz{0}({1})r5   c                 s   s"   | ]\}}d  |t|V  qdS )z{0}={1}N)r<   repr)r   rF   vr   r   r   r      s     z"PClass.__repr__.<locals>.<genexpr>)r<   r%   r&   r>   rV   r7   rP   r   r   r   __repr__   s    
zPClass.__repr__c                    s&   t  fdd jD }t j|ffS )Nc                 3   s&   | ]}t  |r|t |fV  qd S r   )hasattrrB   r   rP   r   r   r      s     
 z$PClass.__reduce__.<locals>.<genexpr>)dictr   r   r%   )rD   datar   rP   r   
__reduce__   s    zPClass.__reduce__c                 C   s   t | |  S )z5
        Returns an evolver for this object.
        )_PClassEvolverrV   rP   r   r   r   evolver   s    zPClass.evolverc                 C   s   |   }||= | S )z
        Remove attribute given by name from the current instance. Raises AttributeError if the
        attribute doesn't exist.
        )r_   
persistent)rD   r   r_   r   r   r   remove   s    zPClass.remove)NF)N)r&   r'   r(   __doc__r"   rA   classmethodrJ   r
   r   rN   rO   rR   rS   rU   rV   rY   r]   r_   ra   r)   r   r   r$   r   r2   %   s"   $ 
	
	r2   )	metaclassc                       s`   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
 fddZdd Z  ZS )r^   )_pclass_evolver_original_pclass_evolver_data_pclass_evolver_data_is_dirtyr3   c                 C   s   || _ || _d| _t | _d S )NF)re   rf   rg   rA   r3   )rD   originalZinitial_dictr   r   r   __init__   s    z_PClassEvolver.__init__c                 C   s
   | j | S r   )rf   rD   itemr   r   r   __getitem__   s    z_PClassEvolver.__getitem__c                 C   s2   | j |t|k	r.|| j |< | j| d| _| S NT)rf   getrC   r3   addrg   rT   r   r   r   rA      s
    
z_PClassEvolver.setc                 C   s   |  || d S r   )rA   rT   r   r   r   __setitem__   s    z_PClassEvolver.__setitem__c                 C   s4   || j kr(| j |= | j| d| _| S t|d S rm   )rf   r3   discardrg   r=   rj   r   r   r   ra      s    
z_PClassEvolver.removec                 C   s   |  | d S r   )ra   rj   r   r   r   __delitem__   s    z_PClassEvolver.__delitem__c                 C   s&   | j r | jjf d| ji| jS | jS )Nr3   )rg   re   r%   r3   rf   rP   r   r   r   r`      s
    z_PClassEvolver.persistentc                    s.   || j kr| || ntt| || d S r   )r   rA   r!   r^   rS   rT   r$   r   r   rS      s    
z_PClassEvolver.__setattr__c                 C   s   | | S r   r   rj   r   r   r   __getattr__  s    z_PClassEvolver.__getattr__)r&   r'   r(   r   ri   rl   rA   rp   ra   rr   r`   rS   rs   r)   r   r   r$   r   r^      s   	r^   N)Zpyrsistent._checked_typesr   r   r   r   Zpyrsistent._field_commonr   r   r   r	   r
   r   Zpyrsistent._transformationsr   r   typer   objectrC   r1   r2   r^   r   r   r   r   <module>   s    	 1