3
\�� � @ s d dl mZ d dlmZ d dlmZmZmZ d dlmZ d dl m
Z
d dl Z d dl m Z d dl
mZmZ d dlZd dlZd dlZd dlZd d
d d d
ddddg ZeefZdZdddddddgZdZdZG dd � d e �Z G d d
� d
e �Z G d d � d e �Z G d d � d e �Z yd d l!m"Z" W n e#k
�r. dZ"Y nX d!Z$d"Z%d#Z&d$Z'd%Z(d&Z)d'Z*d(Z+d)Z,d*Z-d+Z.d,Z/d-Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9Z<d:Z=d;Z>d<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZHdFZIdGZJdHZKdIZLdJZMdKZNdLZOdMZPdNZQdOZRdPZSdQZTdRZUdSZVdTZWdUZXdVZYdWZZeJeTeUeVgZ[dXZ\dYZ]dZZ^d[Z_d\Z`d]Zad^Zbd_Zcd`ZddaZedbZfdcZgejhddde� ei� D �� G dfdg� dg�ZjG dhdi� di�Zkdjdk� Zldldm� Zmdndo� Zndpdq� ZoG drds� ds�ZpG dtdu� du�Zqd�dvdw�dxdy�Zrd�dvdw�dzd{�Zsdvd|d}d~�dd��Ztdvd|d}d~�d�d��Zuy0d d�lvm Z m Z m Z mwZwmxZxmyZymzZzm{Z{m|Z| W n4 e#k
�r^ epeq ZwZxereseteuf\ZyZzZ{Z|Y nX d�d�� Z}e~d�k�rd dlZej�d�d��Z�e�j�d�ej�d��d�d�d�� e�j�d�d�d�d�d�� e�j�d�d�d�d�� e�j�� Z�e�j��r�e}� n>e�j��s�e�j�� n,d dl�Z�x"e�j�D ]Z�e{e��Z�e�j�e�� �q�W dS )�� )� FunctionType)�dispatch_table)�_extension_registry�_inverted_registry�_extension_cache)�islice)�partialN)�maxsize)�pack�unpack� PickleError�
PicklingError�UnpicklingError�Pickler� Unpickler�dump�dumps�load�loadsz4.0z1.0z1.1z1.2z1.3z2.0z3.0� � c @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r � /usr/lib64/python3.6/pickle.pyr @ s c @ s e Zd ZdS )r
N)r r r r r r r r
D s c @ s e Zd ZdS )r N)r r r r r r r r K s c @ s e Zd Zdd� ZdS )�_Stopc C s
|| _ d S )N)�value)�selfr r r r �__init__Y s z_Stop.__init__N)r r r r r r r r r X s r )� PyStringMap� (� .� 0� 1� 2� F� I� J� K� L� M� N� P� Q� R� S� T� U� V� X� a� b� c� d� }� e� g� h� i� j� l� ]� o� p� q� r� s� t� )� u� Gs I01
s I00
� �� �� �� �� �� �� �� �� �� �� �� �� B� C� �� �� �� �� �� �� �� �� �� �c C s g | ]}t jd |�r|�qS )z[A-Z][A-Z0-9_]+$)�re�match)�.0�xr r r �
<listcomp>� s rf c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd
d �Zd d
� ZdS )�_Framer�@ i c C s || _ d | _d S )N)�
file_write�
current_frame)r ri r r r r � s z_Framer.__init__c C s t j� | _d S )N)�io�BytesIOrj )r r r r �
start_framing� s z_Framer.start_framingc C s* | j r&| j j� dkr&| jdd� d | _ d S )Nr T)�force)rj �tell� commit_frame)r r r r � end_framing� s z_Framer.end_framingFc C st | j rp| j }|j� | jks |rp|j� �2}t|�}| j}|t� |td|�� ||� W d Q R X |jd� |j � d S )Nz<Qr )
rj ro �_FRAME_SIZE_TARGET� getbuffer�lenri �FRAMEr
�seek�truncate)r rn �f�data�n�writer r r rp � s
z_Framer.commit_framec C s | j r| j j|�S | j|�S d S )N)rj r{ ri )r ry r r r r{ � s z
_Framer.writeNi )F) r r r rr r rm rq rp r{ r r r r rg � s
rg c @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) � _UnframerNc C s || _ || _d | _d S )N)� file_read�
file_readlinerj )r r} r~ Z file_tellr r r r � s z_Unframer.__init__c C sV | j rH| j j|�}| r0|dkr0d | _ | j|�S t|�|k rDtd��|S | j|�S d S )Nr z$pickle exhausted before end of frame)rj �readr} rt r )r rz ry r r r r � s
z_Unframer.readc C sF | j r:| j j� }|s"d | _ | j� S |d dkr6td��|S | j� S d S )N� �
r z$pickle exhausted before end of frame����
)rj �readliner~ r )r ry r r r r� � s
z_Unframer.readlinec C s2 | j r | j j� dkr td��tj| j|��| _ d S )N� z4beginning of a new frame before end of current frame)rj r r rk rl r} )r �
frame_sizer r r �
load_frame� s z_Unframer.load_frame)N)r r r r r r� r� r r r r r| � s
r| c C sl xb|j d�D ]T}|dkr(tdj|| ���y| }t| |�} W q tk
r^ tdj|| ���Y q X q W | |fS )N�.z<locals>z&Can't get local attribute {!r} on {!r}z Can't get attribute {!r} on {!r})�split�AttributeError�format�getattr)�obj�nameZsubpath�parentr r r �
_getattribute s
r� c C sz t | dd �}|d k r|S x\ttjj� �D ]J\}}|dks(|d krBq(yt||�d | krZ|S W q( tk
rp Y q(X q(W dS )Nr �__main__r )r� �list�sys�modules�itemsr� r� )r� r� � module_name�moduler r r � whichmodule s
r� c C sh | dkr dS | j � d? d }| j|ddd�}| dk rd|dkrd|d dkrd|d d
@ dkrd|d d
� }|S )Nr r� r r� �littleT)� byteorder�signed� � � r� ���r� )�
bit_length�to_bytes)re �nbytes�resultr r r � encode_long$ s r� c C s t j| ddd�S )Nr� T)r� r� )�int�
from_bytes)ry r r r � decode_longB s r� c @ sh e Zd Zd7dd�dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zd8dd�Z dd� Z
dd� Z d9dd�Z i Z
dd� Zee
ed�< dd� Zee
e< d d � Zee
e< d d � Zee
e< d d!� Zee
e< d"d#� Zee
e< d$d%� Zee
e< d&d'� Z e e
e < d(Z d)d*� Z d+d,� Z e e
e!< e"dk �re e
e"< d-d.� Z#d/d0� Z$e$e
e%< d1d2� Z&e&e
e'< d:d3d4�Z(d5d6� Z)e(e
e*< e)e
e< dS );�_PicklerNT)� fix_importsc C s� |d kr t }|dk rt}n$d| ko,tkn s>tdt ��y |j| _W n tk
rf td��Y nX t| j�| _| jj| _i | _ t
|�| _ |dk| _ d| _
|o�|dk | _d S )Nr z pickle protocol must be <= %dz"file must have a 'write' attributer� r )�DEFAULT_PROTOCOL�HIGHEST_PROTOCOL�
ValueErrorr{ � _file_writer� � TypeErrorrg �framer�memor� �proto�bin�fastr� )r �file�protocolr� r r r r [ s"
z_Pickler.__init__c C s | j j� d S )N)r� �clear)r r r r �
clear_memo� s z_Pickler.clear_memoc C sr t | d�s td| jjf ��| jdkr<| jttd| j� � | jdkrP| jj � | j
|� | jt � | jj � d S )Nr� z2Pickler.__init__() was not called by %s.__init__()r� z<Br )
�hasattrr
� __class__r r� r{ �PROTOr
r� rm �save�STOPrq )r r� r r r r � s
z
_Pickler.dumpc C s: | j r
d S t| j�}| j| j|�� ||f| jt|�< d S )N)r� rt r� r{ �put�id)r r� �idxr r r �memoize� s
z_Pickler.memoizec C sT | j dkrtS | jr:|dk r*ttd|� S ttd|� S ntt|�jd� d S d S )Nr � z<Bz<I�asciir� ) r� �MEMOIZEr� �BINPUTr
� LONG_BINPUT�PUT�repr�encode)r r� r r r r� � s
z _Pickler.putc C s@ | j r*|dk r ttd|� S ttd|� S tt|�jd� d S )Nr� z<Bz<Ir� r� )r� �BINGETr
� LONG_BINGET�GETr� r� )r �ir r r �get� s
z _Pickler.getc C s� | j j� | j|�}|d k r.|r.| j|� d S | jjt|��}|d k r^| j| j|d �� d S t|�}| j j|�}|d k r�|| |� d S t
| dt �j|�}|d k r�||�}n�yt |t�} W n t
k
r� d} Y nX | r�| j|� d S t
|dd �}|d k �r|| j�}n0t
|dd �}|d k �r&|� }ntd|j|f ��t|t��rT| j||� d S t|t��sltd| ��t|�}
d|
k�o�d kn �s�td
| ��| j|d |i� d S ) Nr r F�
__reduce_ex__�
__reduce__zCan't pickle %r object: %rz %s must return string or tupler� � z3Tuple returned by %s must have two to five elementsr� )r� rp �
persistent_id� save_persr� r� r� r{ �type�dispatchr� r �
issubclassr� � save_globalr� r
r �
isinstance�str�tuplert � save_reduce) r r� �save_persistent_id�pidre �trx �reduce�rvZissc�lr r r r� � sT
z
_Pickler.savec C s d S )Nr )r r� r r r r� s z_Pickler.persistent_idc C sb | j r | j|dd� | jt� n>y | jtt|�jd� d � W n tk
r\ td��Y nX d S )NF)r� r� r� z2persistent IDs in protocol 0 must be ASCII strings) r� r� r{ � BINPERSID�PERSIDr� r� �UnicodeEncodeErrorr
)r r� r r r r� s z_Pickler.save_persc C s t |t�std��t|�s"td��| j}| j}t|dd�} | jdkr�| dkr�|\}
}} t|
d�sntdj | ���|d k r�|
|j
k r�td j | ���| jd
kr�||
� ||� || � |t � n,t |
j
|
f|�| �}||� |f � |t� n�| jdko�| d k�r\|d }
t|
d��std
��|d k �r6|
|j
k �r6td��|dd � }||
� ||� |t� n||� ||� |t� |d k �r�t|�| jk�r�|t| j| jt|� d � � n
| j|� |d k �r�| j|� |d k �r�| j|� |d k �r�||� |t� d S )Nz'args from save_reduce() must be a tuplez(func from save_reduce() must be callabler � r� �
__newobj_ex__�__new__z#args[0] from {} args has no __new__z(args[0] from {} args has the wrong classr �
__newobj__r z+args[0] from __newobj__ args has no __new__z0args[0] from __newobj__ args has the wrong classr� )r� r� r
�callabler� r{ r� r� r� r� r� � NEWOBJ_EXr r� �REDUCE�NEWOBJr� r� �POPr� r� �_batch_appends�_batch_setitems�BUILD) r �func�args�stateZ listitemsZ dictitemsr� r� r{ Z func_name�cls�kwargsr r r r� sd
"
z_Pickler.save_reducec C s | j t� d S )N)r{ �NONE)r r� r r r � save_none� s z_Pickler.save_nonec C s4 | j dkr | j|rtnt� n| j|r*tnt� d S )Nr� )r� r{ �NEWTRUE�NEWFALSE�TRUE�FALSE)r r� r r r � save_bool� s
z_Pickler.save_boolc C s� | j r~|dkrN|dkr.| jttd|� � d S |dkrN| jttd|� � d S d
| ko`dkn r~| jttd|� � d S | jd kr�t|�}t|�}|d
k r�| jt td|� | � n| jt
td|� | � d S | jt t |�j
d � d � d S )Nr r� z<Bi�� z<Hl i���z<ir� r� r� s L
i �)r� r{ �BININT1r
�BININT2�BININTr� r� rt �LONG1�LONG4�LONGr� r� )r r� �encodedrz r r r � save_long� s&
z_Pickler.save_longc C s< | j r | jttd|� � n | jtt|�jd� d � d S )Nz>dr� r� )r� r{ �BINFLOATr
�FLOATr� r� )r r� r r r �
save_float� s z_Pickler.save_floatc C s� | j dk r@|s | jtf |d� n | jtjt|d�df|d� d S t|�}|dkrj| jtt d|� | � nD|dkr�| j dkr�| jt
t d|� | � n| jt t d |� | � | j |� d S )
Nr )r� �latin1r� z<Bl �� r z<Qz<I)
r� r� �bytes�codecsr� r� rt r{ �SHORT_BINBYTESr
� BINBYTES8�BINBYTESr� )r r� rz r r r �
save_bytes� s
z_Pickler.save_bytesc C s� | j r�|jdd�}t|�}|dkrF| jdkrF| jttd|� | � q�|dkrr| jdkrr| jttd|� | � q�| jttd|� | � n0|j d d
�}|j d d �}| jt
|jd
� d � | j |� d S )Nzutf-8�
surrogatepassr� r z<Bl �� z<Qz<I�\z\u005c�
z\u000azraw-unicode-escaper� ) r� r� rt r� r{ �SHORT_BINUNICODEr
� BINUNICODE8�
BINUNICODE�replace�UNICODEr� )r r� r rz r r r �save_str� s z_Pickler.save_strc C sB |s(| j r| jt� n| jtt � d S t|�}| j}| j}|dkr�| jdkr�x|D ] }||� qTW t |�|kr�| j
|t |� d �}| jt | | � n| jt | � | j
|� d S | j}|t� x|D ] }||� q�W t |�|k�r,| j
|t |� d �}| j �r|t| � n|t |d | � d S |t� | j
|� d S )Nr r� r r� )r� r{ � EMPTY_TUPLE�MARK�TUPLErt r� r� r� r� r� r� �_tuplesize2coder� �POP_MARK)r r� rz r� r� �elementr� r{ r r r �
save_tuple� s:
z_Pickler.save_tuplec C s8 | j r| jt� n| jtt � | j|� | j|� d S )N)r� r{ �
EMPTY_LISTr �LISTr� r� )r r� r r r � save_list s
z_Pickler.save_listi� c C s� | j }| j}| js4x |D ]}||� |t� qW d S t|�}xrtt|| j��}t|�}|dkr�|t � x|D ] }||� qlW |t
� n|r�||d � |t� || jk r>d S q>W d S )Nr� r ) r� r{ r� �APPEND�iterr� r �
_BATCHSIZErt r �APPENDS)r r� r� r{ re �it�tmprz r r r r� s*
z_Pickler._batch_appendsc C s< | j r| jt� n| jtt � | j|� | j|j� � d S )N)r� r{ �
EMPTY_DICTr �DICTr� r� r� )r r� r r r � save_dict. s
z_Pickler.save_dictc C s� | j }| j}| js@x(|D ] \}}||� ||� |t� qW d S t|�}x�tt|| j��}t|�}|dkr�|t � x |D ]\}}||� ||� qxW |t
� n(|r�|d \}}||� ||� |t� || jk rJd S qJW d S )Nr� r ) r� r{ r� �SETITEMr" r� r r# rt r �SETITEMS) r r� r� r{ �k�vr% r& rz r r r r� ; s2
z_Pickler._batch_setitemsc C s� | j }| j}| jdk r0| jtt|�f|d� d S |t� | j|� t|�}xXtt || j
��}t |�}|dkr�|t � x|D ] }||� qzW |t
� || j
k rLd S qLW d S )Nr )r� r )r� r{ r� r� �setr� � EMPTY_SETr� r" r r# rt r �ADDITEMS)r r� r� r{ r% Zbatchrz �itemr r r �save_setZ s$
z_Pickler.save_setc C s� | j }| j}| jdk r0| jtt|�f|d� d S |t� x|D ] }||� q>W t|�| jkr�|t | j
| jt|� d � � d S |t � | j |� d S )Nr )r� r )
r� r{ r� r� � frozensetr� r r� r� r r� � FROZENSETr� )r r� r� r{ r1 r r r �save_frozensetr s
z_Pickler.save_frozensetc
C sT | j }| j}|d kr t|dd �}|d kr.|j}t||�}y(t|dd� tj| }t||�\}}W n, t t
t fk
r� t d|||f ��Y n X ||k r�t d|||f ��| j
dk�rtj||f�} | �r| dkr�|ttd| � � n0| d k�r|ttd
| � � n|ttd | � � d S |jd �d }
||k�r4|
}| j
d
k�r^| j|� | j|� |t� n�||k �rz| jt||
f� n�| j
dk�r�|tt|d� d t|d� d � n�| j�r�tj } tj } ||f| k�r�| ||f \}}n|| k�r�| | }y(|tt|d� d t|d� d � W n* t k
�rD t d||| j
f ��Y nX | j |� d S )Nr r )�levelz(Can't pickle %r: it's not found as %s.%sz2Can't pickle %r: it's not the same object as %s.%sr� r� z<Bi�� z<Hz<ir� r r zutf-8r� r� z?can't pickle global identifier '%s.%s' using pickle protocol %i) r{ r� r� r r� �
__import__r� r� r� � ImportError�KeyErrorr� r
r� r r� �EXT1r
�EXT2�EXT4�
rpartitionr� � STACK_GLOBALr� �GLOBALr r� �_compat_pickleZREVERSE_NAME_MAPPINGZREVERSE_IMPORT_MAPPINGr� r� )
r r� r� r{ r� r� r� Zobj2r� �codeZlastnameZr_name_mappingZr_import_mappingr r r r� � sn
$
&z_Pickler.save_globalc C s` |t d �kr | jt d|d�S |t t�kr:| jt tf|d�S |t d�krV| jt d|d�S | j|�S )N)r� .)N).)r� r� �NotImplementedr� )r r� r r r � save_type� s z_Pickler.save_type)N)T)NNNN)N)+r r r r r� r r� r� r� r� r� r� r� r� r� r� r� �boolr r� r �floatr
r r r� r r� r r� r# r� r) �dictr r� r2 r. r5 r3 r� rC r r r r r r� Y sV )
B
c 1
B r� c @ sn e Zd Zdddd�dd�Zdd� Zd d
� Zd d � Zi Zd
d� Zeee d < dd� Z
e
ee d < dd� Z e ee
d < dd� Zeeed < dd� Zeeed < dd� Zeeed < dd� Zeeed < d d � Zeeed < d d � Zeeed < d d!� Zeeed < d"d#� Z e ee d < d$d%� Z e ee d < d&d'� Z e ee!d < d(d)� Z"e"ee#d < d*d+� Z$e$ee%d < d,d-� Z&e&ee'd < d.d/� Z(d0d1� Z)e)ee*d < d2d3� Z+e+ee,d < d4d5� Z-e-ee.d < d6d7� Z/e/ee0d < d8d9� Z1e1ee2d < d:d;� Z3e3ee4d < d<d=� Z5e5ee6d < d>d?� Z7e7ee8d < d@dA� Z9e9ee:d < dBdC� Z;e;ee<d < dDdE� Z=e=ee>d < dFdG� Z?e?ee@d < dHdI� ZAeAeeBd < dJdK� ZCeCeeDd < dLdM� ZEeEeeFd < dNdO� ZGeGeeHd < dPdQ� ZIeIeeJd < dRdS� ZKeKeeLd < dTdU� ZMeMeeNd < dVdW� ZOeOeePd < dXdY� ZQeQeeRd < dZd[� ZSd\d]� ZTeTeeUd < d^d_� ZVeVeeWd < d`da� ZXeXeeYd < dbdc� ZZeZee[d < ddde� Z\e\ee]d < dfdg� Z^e^ee_d < dhdi� Z`e`eead < djdk� Zbebeecd < dldm� Zdedeeed < dndo� Zfdpdq� Zgdrds� Zheheeid < dtdu� Zjejeekd < dvdw� Zleleemd < dxdy� Zneneeod < dzd{� Zpepeeqd < d|d}� Zrereesd < d~d� Zteteeud < d�d�� Zveveewd < d�d�� Zxexeeyd < d�d�� Zzezee{d < d�d�� Z|e|ee}d < d�d�� Z~e~eed < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�S )��
_UnpicklerT�ASCII�strict)r� �encoding�errorsc C s2 |j | _|j| _i | _|| _|| _d| _|| _d S )Nr ) r� �_file_readliner �
_file_readr� rJ rK r� r� )r r� r� rJ rK r r r r � s z_Unpickler.__init__c C s� t | d�s td| jjf ��t| j| j�| _| jj| _| jj | _ g | _
g | _ | j j | _ d| _
| j}| j}y*x$|d�}|s|t�||d | � qlW W n" tk
r� } z|jS d }~X nX d S )NrM z4Unpickler.__init__() was not called by %s.__init__()r r� )r� r r� r r| rM rL � _unframerr r� � metastack�stack�appendr� r� �EOFErrorr r )r r r� �keyZstopinstr r r r s(
z_Unpickler.loadc C s | j }| jj� | _ | j j| _|S )N)rP rO �poprQ )r r� r r r �pop_mark s
z_Unpickler.pop_markc C s t d��d S )Nz%unsupported persistent id encountered)r )r r� r r r �persistent_load% s z_Unpickler.persistent_loadc C s< | j d�d }d| ko tkn s2td| ��|| _d S )Nr� r z unsupported pickle protocol: %d)r r� r� r� )r r� r r r �
load_proto* s z_Unpickler.load_protor c C s8 t d| jd��\}|tjkr(td| ��| jj|� d S )Nz<Q� z frame size > sys.maxsize: %d)r r r� r r� rN r� )r r� r r r r� 1 s
z_Unpickler.load_framec C sL y| j � d d� jd�}W n tk
r6 td��Y nX | j| j|�� d S )Nr� r� z2persistent IDs in protocol 0 must be ASCII stringsr� )r� �decode�UnicodeDecodeErrorr rQ rV )r r� r r r � load_persid8 s z_Unpickler.load_persidc C s | j j� }| j| j|�� d S )N)rP rT rQ rV )r r� r r r �load_binpersidA s
z_Unpickler.load_binpersidc C s | j d � d S )N)rQ )r r r r � load_noneF s z_Unpickler.load_nonec C s | j d� d S )NF)rQ )r r r r �
load_falseJ s z_Unpickler.load_falsec C s | j d� d S )NT)rQ )r r r r � load_trueN s z_Unpickler.load_truec C sL | j � }|tdd � kr d}n |tdd � kr4d}n
t|d�}| j|� d S )Nr� FTr )r� r� r� r� rQ )r ry �valr r r �load_intR s
z_Unpickler.load_intc C s | j td| jd��d � d S )Nz<ir r )rQ r r )r r r r � load_binint] s z_Unpickler.load_binintc C s | j | jd�d � d S )Nr� r )rQ r )r r r r � load_binint1a s z_Unpickler.load_binint1c C s | j td| jd��d � d S )Nz<Hr� r )rQ r r )r r r r � load_binint2e s z_Unpickler.load_binint2c C s@ | j � d d� }|r,|d dkr,|d d� }| jt|d�� d S )Nr� r* r r� r� �L r� )r� rQ r� )r r` r r r � load_longi s z_Unpickler.load_longc C s* | j d�d }| j |�}| jt|�� d S )Nr� r )r rQ r� )r rz ry r r r �
load_long1p s
z_Unpickler.load_long1c C s> t d| jd��\}|dk r"td��| j|�}| jt|�� d S )Nz<ir r z#LONG pickle has negative byte count)r r r rQ r� )r rz ry r r r �
load_long4v s
z_Unpickler.load_long4c C s | j t| j� d d� �� d S )Nr� r� )rQ rE r� )r r r r �
load_float s z_Unpickler.load_floatc C s | j td| jd��d � d S )Nz>drX r )rQ r r )r r r r �
load_binfloat� s z_Unpickler.load_binfloatc C s" | j dkr|S |j| j | j�S d S )Nr )rJ rY rK )r r r r r �_decode_string� s
z_Unpickler._decode_stringc C sl | j � d d� }t|�dkrF|d |d krF|d dkrF|dd� }ntd��| j| jtj|�d �� d S ) Nr� r� r s "'z)the STRING opcode argument must be quotedr� r� r� )r� rt r rQ rk r �
escape_decode)r ry r r r � load_string� s
(z_Unpickler.load_stringc C s@ t d| jd��\}|dk r"td��| j|�}| j| j|�� d S )Nz<ir r z(BINSTRING pickle has negative byte count)r r r rQ rk )r rt ry r r r �load_binstring� s
z_Unpickler.load_binstringc C s: t d| jd��\}|tkr&tdt ��| j| j|�� d S )Nz<Ir z2BINBYTES exceeds system's maximum size of %d bytes)r r r r rQ )r rt r r r �
load_binbytes� s
z_Unpickler.load_binbytesc C s | j t| j� d d� d�� d S )Nr� zraw-unicode-escaper� )rQ r� r� )r r r r � load_unicode� s z_Unpickler.load_unicodec C sB t d| jd��\}|tkr&tdt ��| jt| j|�dd�� d S )Nz<Ir z4BINUNICODE exceeds system's maximum size of %d byteszutf-8r )r r r r rQ r� )r rt r r r �load_binunicode� s
z_Unpickler.load_binunicodec C sB t d| jd��\}|tkr&tdt ��| jt| j|�dd�� d S )Nz<QrX z5BINUNICODE8 exceeds system's maximum size of %d byteszutf-8r )r r r r rQ r� )r rt r r r �load_binunicode8� s
z_Unpickler.load_binunicode8c C s: t d| jd��\}|tkr&tdt ��| j| j|�� d S )Nz<QrX z3BINBYTES8 exceeds system's maximum size of %d bytes)r r r r rQ )r rt r r r �load_binbytes8� s
z_Unpickler.load_binbytes8c C s, | j d�d }| j |�}| j| j|�� d S )Nr� r )r rQ rk )r rt ry r r r �load_short_binstring� s
z _Unpickler.load_short_binstringc C s" | j d�d }| j| j |�� d S )Nr� r )r rQ )r rt r r r �load_short_binbytes� s z _Unpickler.load_short_binbytesc C s* | j d�d }| jt| j |�dd�� d S )Nr� r zutf-8r )r rQ r� )r rt r r r �load_short_binunicode� s z _Unpickler.load_short_binunicodec C s | j � }| jt|�� d S )N)rU rQ r� )r r� r r r �
load_tuple� s z_Unpickler.load_tuplec C s | j f � d S )N)rQ )r r r r �load_empty_tuple� s z_Unpickler.load_empty_tuplec C s | j d f| j d< d S )Nr� r� r� )rP )r r r r � load_tuple1� s z_Unpickler.load_tuple1c C s$ | j d | j d fg| j dd �< d S )Nr� r� r� r� r� )rP )r r r r � load_tuple2� s z_Unpickler.load_tuple2c C s, | j d | j d | j d fg| j dd �< d S )Nr r� r� ���r� r� r{ )rP )r r r r � load_tuple3� s z_Unpickler.load_tuple3c C s | j g � d S )N)rQ )r r r r �load_empty_list� s z_Unpickler.load_empty_listc C s | j i � d S )N)rQ )r r r r �load_empty_dictionary� s z _Unpickler.load_empty_dictionaryc C s | j t� � d S )N)rQ r. )r r r r �load_empty_set� s z_Unpickler.load_empty_setc C s | j � }| jt|�� d S )N)rU rQ r3 )r r� r r r �load_frozenset� s z_Unpickler.load_frozensetc C s | j � }| j|� d S )N)rU rQ )r r� r r r � load_list� s z_Unpickler.load_listc s4 | j � � � fdd�tdt� �d�D �}| j|� d S )Nc s i | ]}� |d � | �qS )r� r )rd r� )r� r r �
<dictcomp> s z(_Unpickler.load_dict.<locals>.<dictcomp>r r� )rU �rangert rQ )r �dr )r� r � load_dict s
z_Unpickler.load_dictc C s� |st |t� st|d�rly ||� }W qv tk
rh } z&td|jt|�f tj� d ��W Y d d }~X qvX n
|j|�}| j |� d S )NZ__getinitargs__zin constructor for %s: %sr� )
r� r� r� r� r r� r� �exc_infor� rQ )r �klassr� r �errr r r � _instantiate s
0
z_Unpickler._instantiatec C sL | j � d d� jd�}| j � d d� jd�}| j||�}| j|| j� � d S )Nr� r� r� r� )r� rY �
find_classr� rU )r r� r� r� r r r � load_inst s z_Unpickler.load_instc C s" | j � }|jd�}| j||� d S )Nr )rU rT r� )r r� r� r r r �load_obj! s
z_Unpickler.load_objc C s2 | j j� }| j j� }|j|f|�� }| j|� d S )N)rP rT r� rQ )r r� r� r� r r r � load_newobj( s
z_Unpickler.load_newobjc C s> | j j� }| j j� }| j j� }|j|f|�|�}| j|� d S )N)rP rT r� rQ )r r� r� r� r� r r r �load_newobj_ex/ s
z_Unpickler.load_newobj_exc C sF | j � d d� jd�}| j � d d� jd�}| j||�}| j|� d S )Nr� zutf-8r� r� )r� rY r� rQ )r r� r� r� r r r � load_global7 s z_Unpickler.load_globalc C sJ | j j� }| j j� }t|�tk s,t|�tk r4td��| j| j||�� d S )NzSTACK_GLOBAL requires str)rP rT r� r� r rQ r� )r r� r� r r r �load_stack_global> s
z _Unpickler.load_stack_globalc C s | j d�d }| j|� d S )Nr� r )r �
get_extension)r rA r r r � load_ext1F s z_Unpickler.load_ext1c C s t d| jd��\}| j|� d S )Nz<Hr� )r r r� )r rA r r r � load_ext2K s z_Unpickler.load_ext2c C s t d| jd��\}| j|� d S )Nz<ir )r r r� )r rA r r r � load_ext4P s z_Unpickler.load_ext4c C sp g }t j||�}||k r&| j|� d S tj|�}|sP|dkrDtd��td| ��| j|� }|t |< | j|� d S )Nr zEXT specifies code <= 0z unregistered extension code %d)r r� rQ r r r� r� )r rA Znilr� rS r r r r� U s
z_Unpickler.get_extensionc C s� | j dk rF| jrF||ftjkr2tj||f \}}n|tjkrFtj| }t|dd� | j dkrpttj| |�d S t tj| |�S d S )Nr r )r6 r )
r� r� r@ Z NAME_MAPPINGZIMPORT_MAPPINGr7 r� r� r� r� )r r� r� r r r r� e s
z_Unpickler.find_classc C s& | j }|j� }|d }||� |d< d S )Nr� r� r� )rP rT )r rP r� r� r r r � load_reducer s z_Unpickler.load_reducec C s | j r| j d= n| j� d S )Nr� r� )rP rU )r r r r �load_popy s
z_Unpickler.load_popc C s | j � d S )N)rU )r r r r �
load_pop_mark� s z_Unpickler.load_pop_markc C s | j | jd � d S )Nr� r� )rQ rP )r r r r �load_dup� s z_Unpickler.load_dupc C s( t | j� d d� �}| j| j| � d S )Nr� r� )r� r� rQ r� )r r� r r r �load_get� s z_Unpickler.load_getc C s" | j d�d }| j| j| � d S )Nr� r )r rQ r� )r r� r r r � load_binget� s z_Unpickler.load_bingetc C s& t d| jd��\}| j| j| � d S )Nz<Ir )r r rQ r� )r r� r r r �load_long_binget� s z_Unpickler.load_long_bingetc C s8 t | j� d d� �}|dk r$td��| jd | j|< d S )Nr� r znegative PUT argumentr� r� )r� r� r� rP r� )r r� r r r �load_put� s z_Unpickler.load_putc C s2 | j d�d }|dk r td��| jd | j|< d S )Nr� r znegative BINPUT argumentr� )r r� rP r� )r r� r r r � load_binput� s z_Unpickler.load_binputc C s6 t d| jd��\}|tkr"td��| jd | j|< d S )Nz<Ir z negative LONG_BINPUT argumentr� r� )r r r r� rP r� )r r� r r r �load_long_binput� s z_Unpickler.load_long_binputc C s | j }| jd |t|�< d S )Nr� r� )r� rP rt )r r� r r r � load_memoize� s z_Unpickler.load_memoizec C s$ | j }|j� }|d }|j|� d S )Nr� r� )rP rT rQ )r rP r r� r r r � load_append� s z_Unpickler.load_appendc C sH | j � }| jd }t|t�r(|j|� n |j}x|D ] }||� q4W d S )Nr� r� )rU rP r� r� �extendrQ )r r� Zlist_objrQ r1 r r r � load_appends� s
z_Unpickler.load_appendsc C s* | j }|j� }|j� }|d }|||< d S )Nr� r� )rP rT )r rP r rS rF r r r � load_setitem� s
z_Unpickler.load_setitemc C sD | j � }| jd }x,tdt|�d�D ]}||d ||| < q$W d S )Nr� r r� r� )rU rP r� rt )r r� rF r� r r r �
load_setitems� s
z_Unpickler.load_setitemsc C sH | j � }| jd }t|t�r(|j|� n |j}x|D ] }||� q4W d S )Nr� r� )rU rP r� r. �update�add)r r� Zset_objr� r1 r r r �
load_additems� s
z_Unpickler.load_additemsc
C s� | j }|j� }|d }t|dd �}|d k r6||� d S d }t|t�rXt|�dkrX|\}}|r�|j}tj}x6|j � D ]*\}} t
|�t kr�| |||�< qr| ||< qrW |r�x |j � D ]\}} t ||| � q�W d S )Nr� � __setstate__r� r� )
rP rT r� r� r� rt �__dict__r� �internr� r� r� �setattr)
r rP r� Zinst�setstateZ slotstateZ inst_dictr� r, r- r r r �
load_build� s( z_Unpickler.load_buildc C s" | j j| j� g | _| jj| _d S )N)rO rQ rP )r r r r � load_mark� s z_Unpickler.load_markc C s | j j� }t|��d S )N)rP rT r )r r r r r � load_stop� s
z_Unpickler.load_stopN)�r r r r r rU rV r� rW r� r� ru r[ r� r\ r� r] r� r^ r� r_ r� ra �INTrb r� rc r� rd r� rf r rg r� rh r ri r rj r rk rm �STRINGrn � BINSTRINGro r rp r rq r rr r rs r rt �SHORT_BINSTRINGru r
rv r rw r rx r ry �TUPLE1rz �TUPLE2r| �TUPLE3r} r r~ r' r r/ r� r4 r� r r� r( r� r� �INSTr� �OBJr� r� r� r� r� r? r� r> r� r: r� r; r� r< r� r� r� r� r� r� r� r r� �DUPr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r! r� r$ r� r* r� r+ r� r0 r� r� r� r r� r� r r r r rG � s $
rG T)r� c C s t |||d�j| � d S )N)r� )r� r )r� r� r� r� r r r �_dump s r� c C s( t j� }t|||d�j| � |j� }|S )N)r� )rk rl r� r �getvalue)r� r� r� rx �resr r r �_dumps s r� rH rI )r� rJ rK c C s t | |||d�j� S )N)r� rJ rK )rG r )r� r� rJ rK r r r �_load
s r� c C s0 t | t�rtd��tj| �}t||||d�j� S )Nz%Can't load pickle from unicode string)r� rJ rK )r� r� r� rk rl rG r )�sr� rJ rK r� r r r �_loads s
r� ) r r
r r r r r r r c C s dd l } | j� S )Nr )�doctestZtestmod)r� r r r �_test* s r� r� z$display contents of the pickle files)Z description� pickle_file�br�*zthe pickle file)r� �nargs�helpz-tz--test�
store_truezrun self-test suite)�actionr� z-vz)run verbosely; only affects self-test run)N)N)��typesr �copyregr r r r � itertoolsr � functoolsr r� r Zstructr
r rb rk r r@ �__all__r � bytearrayZ bytes_typesZformat_versionZcompatible_formatsr� r� � Exceptionr r
r r Zorg.python.corer r8 r r� r� r r� r r� r� r� r r� r� r� r� r� r� r� r� r r r! r� r? r( r' r$ r� r� r� r� r r r� r� r� r� r* r r r+ r r� r� r� r� r: r; r< r� r� r� r� r� r� r r r r
r r r r/ r0 r4 r� r> r� ru r� �dirrg r| r� r� r� r� r� rG r� r� r� r� �_pickler r r r r r r� r �argparse�ArgumentParser�parser� add_argumentZFileType�
parse_argsr� Ztestr� Z
print_help�pprintrx r� r r r r �<module> s,
$*
-0