3
\_ � @ s� d Z dddddgZi Zddd�Zdd� Zye W n ek
rD Y nX d d
� Zeeee� d d � ZdZ dd� Z
dd� Z dd� Z dd� Z
i Zi Zi Zdd� Zdd� Zdd� ZdS ) z�Helper to provide extensibility for pickle.
This is only useful to add pickle support for extension types defined in
C, not for instances of user-defined classes.
�pickle� constructor�
add_extension�remove_extension�clear_extension_cacheNc C s, t |�std��|t| < |d k r(t|� d S )Nz$reduction functions must be callable)�callable� TypeError�dispatch_tabler )�ob_type�pickle_function�constructor_ob� r � /usr/lib64/python3.6/copyreg.pyr s
c C s t | �std��d S )Nz constructors must be callable)r r )�objectr r r
r s c C s t | j| jffS )N)�complex�real�imag)�cr r r
�pickle_complex" s r c C s< |t krt j| �}n$|j| |�}|jt jkr8|j||� |S )N)r �__new__�__init__)�cls�base�state�objr r r
�_reconstructor) s r � � c C s� x,| j jD ] }t|d�r
|jt@ r
P q
W t}|tkr<d }n || j krTtd|j ��|| �}| j ||f}y
| j}W nL t k
r� t
| dd �r�td��y
| j }W n t k
r� d }Y nX Y nX |� }|r�t ||fS t |fS d S )N� __flags__zcan't pickle %s objects� __slots__zNa class that defines __slots__ without defining __getstate__ cannot be pickled)
� __class__�__mro__�hasattrr � _HEAPTYPEr r �__name__� __getstate__�AttributeError�getattr�__dict__r )�self�protor r �args�getstate�dictr r r
�
_reduce_ex6 s.
r- c G s | j | f|�� S )N)r )r r* r r r
�
__newobj__W s r. c C s | j | f|�|�S )zUsed by pickle protocol 4, instead of __newobj__ to allow classes with
keyword-only arguments to be pickled correctly.
)r )r r* �kwargsr r r
�
__newobj_ex__Z s r0 c C s� | j jd�}|dk r|S g }t| d�s(n�x�| jD ]�}d|j kr0|j d }t|t�rX|f}xh|D ]`}|d krnq^q^|jd�r�|jd� r�|jj d�}|r�|j
d||f � q�|j
|� q^|j
|� q^W q0W y
|| _ W n Y nX |S )
a� Return a list of slot names for a given class.
This needs to find slots defined by the class and its bases, so we
can't simply return the __slots__ attribute. We must walk down
the Method Resolution Order and concatenate the __slots__ of each
class found there. (This assumes classes don't modify their
__slots__ attribute to misrepresent their slots after the class is
defined.)
�
__slotnames__Nr r' � __weakref__�__�_z_%s%s)r' r2 ) r' �getr! r �
isinstance�str�
startswith�endswithr# �lstrip�appendr1 )r �namesr �slots�name�strippedr r r
�
_slotnames` s2
r@ c C s� t |�}d| kodkn s(td��| |f}tj|�|krPtj|�|krPdS |tkrltd|t| f ��|tkr�td|t| f ��|t|< |t|< dS )zRegister an extension code.r i���zcode out of rangeNz)key %s is already registered with code %sz$code %s is already in use for key %s)�int�
ValueError�_extension_registryr5 �_inverted_registry)�moduler> �code�keyr r r
r � s c C sR | |f}t j|�|ks$tj|�|kr4td||f ��t |= t|= |tkrNt|= dS )z0Unregister an extension code. For testing only.z%key %s is not registered with code %sN)rC r5 rD rB �_extension_cache)rE r> rF rG r r r
r � s c C s t j� d S )N)rH �clearr r r r
r � s )Ni )�__doc__�__all__r r r r � NameErrorr r r" r- r. r0 r@ rC rD rH r r r r r r r
�<module> s.
!<