U
yi @ s d Z dddddgZi ZdddZdd Zze W n ek
rD Y nX d d
Zeeee d d Zd
Z dd Z
dd Z dd Z dd Z
i Zi Zi Zdd Zdd Zdd ZdS )zHelper 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_extensionremove_extensionclear_extension_cacheNc C s, t |std|t| < |d k r(t| d S )Nz$reduction functions must be callable)callable TypeErrordispatch_tabler )ob_typepickle_functionconstructor_ob r /usr/lib/python3.8/copyreg.pyr s
c C s t | stdd S )Nz constructors must be callable)r r )objectr r r
r s c C s t | j| jffS N)complexrealimag)cr r r
pickle_complex" s r c C s< |t krt | }n$|| |}|jt jkr8||| |S r )r __new____init__)clsbasestateobjr r r
_reconstructor) s r i c C s |dk s t | j}|jD ] }t|dr|jt@ s q:qt}|tkrHd }n"||krbtd|jd|| }|||f}z
| j }W n\ t
k
r t | dd rtd|jd| d z
| j }W n t
k
r d }Y nX Y nX | }|rt
||fS t
|fS d S )N __flags__zcannot pickle z object __slots__zf object: a class that defines __slots__ without defining __getstate__ cannot be pickled with protocol )AssertionError __class____mro__hasattrr _HEAPTYPEr r __name__ __getstate__AttributeErrorgetattr__dict__r )selfprotor r r argsgetstatedictr r r
_reduce_ex6 s6
r. c G s | j | f| S r r )r r+ r r r
__newobj__Z 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__] s r2 c C s | j d}|dk r|S g }t| ds(n| jD ]}d|j kr.|j d }t|trV|f}|D ]^}|dkrjqZqZ|dr|ds|j d}|r|
d||f q|
| qZ|
| qZq.z
|| _ 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( getr" r!
isinstancestr
startswithendswithr$ lstripappendr3 )r namesr slotsnamestrippedr r r
_slotnamesc s2
rB c C s t |}d| kr dks&n td| |f}t||krNt||krNdS |tkrjtd|t| f |tkrtd|t| f |t|< |t|< dS )zRegister an extension code. izcode 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_registrymoduler@ codekeyr r r
r s$
c C sR | |f}t ||ks$t||kr4td||f t |= t|= |tkrNt|= dS )z0Unregister an extension code. For testing only.z%key %s is not registered with code %sN)rF r7 rG rE _extension_cacherH r r r
r s c C s t d S r )rL clearr r r r
r s )N)__doc____all__r r r r NameErrorr r r# r. r0 r2 rB rF rG rL r r r r r r r