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� |dk s t �x,| jjD ] }t|d�r|jt@ rP qW t}|tkrHd }n || jkr`td|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)�AssertionError� __class__�__mro__�hasattrr � _HEAPTYPEr r �__name__� __getstate__�AttributeError�getattr�__dict__r )�self�protor r �args�getstate�dictr r r
�
_reduce_ex6 s0
r/ c G s | j | f|�� S )N)r )r r, r r r
�
__newobj__W s r0 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 r2 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) r4 ) r) �getr# r" �
isinstance�str�
startswith�endswithr% �lstrip�appendr3 )r �namesr �slots�name�strippedr r r
�
_slotnames` s2
rB 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_registryr7 �_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)rE r7 rF rD �_extension_cache)rG r@ rH rI r r r
r � s c C s t j� d S )N)rJ �clearr r r r
r � s )Ni )�__doc__�__all__r r r r � NameErrorr r r$ r/ r0 r2 rB rE rF rJ r r r r r r r
�<module> s.
!<