JezK
Edit File: pickle.cpython-39.opt-1.pyc
a �i�� � @ sH d Z ddlmZ ddlmZ ddlmZmZmZ ddlm Z ddl m Z ddl Z ddl m Z dd lmZmZ ddlZddlZddlZddlZg d �Zz dd lmZ e�d � d ZW n ey� dZY n0 ee fZ dZ g d�Z dZ dZ!G dd� de"�Z#G dd� de#�Z$G dd� de#�Z%G dd� de"�Z&zddl'm(Z( W n e�yN dZ(Y n0 d Z)d Z*d Z+d Z,d Z-d!Z.d"Z/d#Z0d$Z1d%Z2d&Z3d'Z4d(Z5d)Z6d*Z7d+Z8d,Z9d-Z:d.Z;d/Z<d0Z=d1Z>d2Z?d3Z@d4ZAd5ZBd6ZCd7ZDd8ZEd9ZFd:ZGd;ZHd<ZId=ZJd>ZKd?ZLd@ZMdAZNdBZOdCZPdDZQdEZRdFZSdGZTdHZUdIZVdJZWdKZXdLZYdMZZdNZ[dOZ\dPZ]dQZ^dRZ_eOeYeZe[gZ`dSZadTZbdUZcdVZddWZedXZfdYZgdZZhd[Zid\Zjd]Zkd^Zld_Zmd`ZndaZoe�pdbdc� eq� D �� G ddde� de�ZrG dfdg� dg�Zsdhdi� Ztdjdk� Zudldm� Zvdndo� ZwG dpdq� dq�ZxG drds� ds�Zyd�d ddt�dudv�Zzd�d ddt�dwdx�Z{d dydzdd{�d|d}�Z|d dydzdd{�d~d�Z}z0dd lm#Z#m$Z$m%Z%m~Z~mZm�Z�m�Z�m�Z�m�Z� W n2 e�y� exey Z~Zeze{e|e}f\Z�Z�Z�Z�Y n0 d�d�� Z�e�d�k�rDddl�Z�e�j�d�d��Z�e�j�d�e���d��d�d�d�� e�j�d�d�d�d�d�� e�j�d�d�d�d�� e���� Z�e�j��r e�� n:e�j��s e���� n(ddl�Z�e�j�D ]Z�e�e��Z�e���e�� �q*dS )�a� Create portable serialized representations of Python objects. See module copyreg for a mechanism for registering custom picklers. See module pickletools source for extensive comments. Classes: Pickler Unpickler Functions: dump(object, file) dumps(object) -> string load(file) -> object loads(bytes) -> object Misc variables: __version__ format_version compatible_formats � )� FunctionType)�dispatch_table)�_extension_registry�_inverted_registry�_extension_cache)�islice)�partialN)�maxsize)�pack�unpack) � PickleError� PicklingError�UnpicklingError�Pickler� Unpickler�dump�dumps�load�loads)� PickleBufferr TF�4.0)z1.0z1.1z1.2z1.3z2.0z3.0r z5.0� � c @ s e Zd ZdZdS )r z6A common base class for the other pickling exceptions.N��__name__� __module__� __qualname__�__doc__� r r � /usr/lib64/python3.9/pickle.pyr I s r c @ s e Zd ZdZdS )r z]This exception is raised when an unpicklable object is passed to the dump() method. Nr r r r r r M s r c @ s e Zd ZdZdS )r a This exception is raised when there is a problem unpickling an object, such as a security violation. Note that other exceptions may also be raised during unpickling, including (but not necessarily limited to) AttributeError, EOFError, ImportError, and IndexError. Nr r r r r r T s r c @ s e Zd Zdd� ZdS )�_Stopc C s || _ d S �N)�value��selfr" r r r �__init__b s z_Stop.__init__N)r r r r% r r r r r a 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 �d |�r|�qS )z[A-Z][A-Z0-9_]+$)�re�match)�.0�xr r r � <listcomp>� � ro c @ sF e Zd ZdZdZdd� Zdd� Zdd� Zdd d �Zd d � Z dd� Z dS )�_Framerr i c C s || _ d | _d S r! )� file_write� current_frame)r$ rr r r r r% � s z_Framer.__init__c C s t �� | _d S r! )�io�BytesIOrs �r$ r r r � start_framing� s z_Framer.start_framingc C s* | j r&| j �� dkr&| jdd� d | _ d S )Nr T��force)rs �tell� commit_framerv r r r � end_framing� s z_Framer.end_framingFc C sf | j rb| j }|�� | jks |rb|�� }| j}t|�| jkrP|ttdt|�� � ||� t � � | _ d S )N�<Q) rs rz �_FRAME_SIZE_TARGET� getbufferrr �len�_FRAME_SIZE_MIN�FRAMEr rt ru )r$ ry �f�data�writer r r r{ � s z_Framer.commit_framec C s | j r| j �|�S | �|�S d S r! )rs r� rr �r$ r� r r r r� � s z _Framer.writec C s, | j }| jr| jdd� ||� ||� d S )NTrx )rr rs r{ )r$ �header�payloadr� r r r �write_large_bytes� s z_Framer.write_large_bytesN)F) r r r r� r~ r% rw r| r{ r� r� r r r r rq � s rq c @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd d � ZdS ) � _UnframerNc C s || _ || _d | _d S r! )� file_read� file_readliners )r$ r� r� � file_tellr r r r% s z_Unframer.__init__c C s� | j rb| j �|�}|dkrJt|�dkrJd | _ t|�}| �|�|d d �< |S |t|�k r^td��|S t|�}| �|�|d d �< |S d S �Nr �$pickle exhausted before end of frame)rs �readintor� r� r )r$ �buf�nr r r r� s �z_Unframer.readintoc C sT | j rF| j �|�}|s.|dkr.d | _ | �|�S t|�|k rBtd��|S | �|�S d S r� )rs �readr� r� r �r$ r� r� r r r r� s �z_Unframer.readc C sF | j r:| j �� }|s"d | _ | �� S |d dkr6td��|S | �� S d S )N���� r� )rs �readliner� r r� r r r r� , s �z_Unframer.readlinec C s2 | j r | j �� dkr td��t�| �|��| _ d S )Nrp z4beginning of a new frame before end of current frame)rs r� r rt ru r� �r$ � frame_sizer r r � load_frame9 s �z_Unframer.load_frame)N)r r r r% r� r� r� r� r r r r r� s r� c C sh |� d�D ]T}|dkr&td�|| ���z| }t| |�} W q ty\ td�|| ��d �Y q 0 q | |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�name�subpath�parentr r r � _getattributeB s � �� r� c C s� t | dd�}|dur|S tj�� �� D ]V\}}|dks&|dks&|du rHq&z t||�d | u rf|W S W q& tyz Y q&0 q&dS )z$Find the module an object belong to.r N�__main__� __mp_main__r )r� �sys�modules�copy�itemsr� r� )r� r� � module_name�moduler r r � whichmoduleO s �� r� c C sh | dkr dS | � � d? d }| j|ddd�}| dk rd|dkrd|d d krd|d d @ dkrd|d d� }|S ) a� Encode a long to a two's complement little-endian binary string. Note that 0 is a special case, returning an empty string, to save a byte in the LONG1 pickling context. >>> encode_long(0) b'' >>> encode_long(255) b'\xff\x00' >>> encode_long(32767) b'\xff\x7f' >>> encode_long(-256) b'\x00\xff' >>> encode_long(-32768) b'\x00\x80' >>> encode_long(-128) b'\x80' >>> encode_long(127) b'\x7f' >>> r rp � � �littleT�� byteorder�signedr� � ���� N)� bit_length�to_bytes)rn �nbytes�resultr r r � encode_longb s r� c C s t j| ddd�S )a\ Decode a long from a two's complement little-endian binary string. >>> decode_long(b'') 0 >>> decode_long(b"\xff\x00") 255 >>> decode_long(b"\xff\x7f") 32767 >>> decode_long(b"\x00\xff") -256 >>> decode_long(b"\x00\x80") -32768 >>> decode_long(b"\x80") -128 >>> decode_long(b"\x7f") 127 r� Tr� )�int� from_bytes)r� r r r � decode_long� s r� c @ s� e Zd Zd;ddd�dd�Zdd� Zdd � Zd d � Zd d � Zdd� Zd<dd�Z dd� Z dd� Z d=dd�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< er�d$d%� Zee e < d&d'� Z e e e < d(d)� Z e e e < d*d+� Z!e!e e"< d,Z#d-d.� Z$d/d0� Z%e%e e&< e'du�r@e%e e'< d1d2� Z(d3d4� Z)e)e e*< d5d6� Z+e+e e,< d>d7d8�Z-d9d:� Z.e-e e/< e.e e< dS )?�_PicklerNT�� fix_imports�buffer_callbackc C s� |du r t }|dk rt}n"d| kr.tks<n tdt ��|durT|dk rTtd��|| _z |j| _W n ty� td��Y n0 t| j�| _ | j j| _| j j | _ i | _ t |�| _|dk| _d| _|o�|dk | _dS ) a! This takes a binary file for writing a pickle data stream. The optional *protocol* argument tells the pickler to use the given protocol; supported protocols are 0, 1, 2, 3, 4 and 5. The default protocol is 4. It was introduced in Python 3.4, and is incompatible with previous versions. Specifying a negative protocol version selects the highest protocol version supported. The higher the protocol used, the more recent the version of Python needed to read the pickle produced. The *file* argument must have a write() method that accepts a single bytes argument. It can thus be a file object opened for binary writing, an io.BytesIO instance, or any other custom object that meets this interface. If *fix_imports* is True and *protocol* is less than 3, pickle will try to map the new Python 3 names to the old module names used in Python 2, so that the pickle data stream is readable with Python 2. If *buffer_callback* is None (the default), buffer views are serialized into *file* as part of the pickle stream. If *buffer_callback* is not None, then it can be called any number of times with a buffer view. If the callback returns a false value (such as None), the given buffer is out-of-band; otherwise the buffer is serialized in-band, i.e. inside the pickle stream. It is an error if *buffer_callback* is not None and *protocol* is None or smaller than 5. Nr z pickle protocol must be <= %dr z#buffer_callback needs protocol >= 5z"file must have a 'write' attributer� r� )�DEFAULT_PROTOCOL�HIGHEST_PROTOCOL� ValueError�_buffer_callbackr� � _file_writer� � TypeErrorrq �framerr� �_write_large_bytes�memor� �proto�bin�fastr� )r$ �file�protocolr� r� r r r r% � s* # z_Pickler.__init__c C s | j �� dS )a Clears the pickler's "memo". The memo is the data structure that remembers which objects the pickler has already seen, so that shared or recursive objects are pickled by reference and not by value. This method is useful when re-using picklers. N)r� �clearrv r r r � clear_memo� s z_Pickler.clear_memoc C sr t | d�s td| jjf ��| jdkr<| �ttd| j� � | jdkrP| j� � | � |� | �t � | j� � dS )z7Write a pickled representation of obj to the open file.r� z2Pickler.__init__() was not called by %s.__init__()� �<Br N) �hasattrr � __class__r r� r� �PROTOr r� rw �save�STOPr| �r$ r� r r r r � s � z _Pickler.dumpc C s: | j r dS t| j�}| �| �|�� ||f| jt|�< dS )z Store an object in the memo.N)r� r� 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|��d� d S d S )Nr � r� �<I�ascii� ) 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|��d� d S )Nr� r� r� r� r� )r� �BINGETr � LONG_BINGET�GETr� r� �r$ �ir r r �get s z _Pickler.getc C s� | j �� | �|�}|d ur.|r.| �|� d S | j�t|��}|d ur^| �| �|d �� d S t}t | dd �}|d ur~||�}|tu �r@t |�}| j �|�}|d ur�|| |� d S t | dt ��|�}|d ur�||�}njt |t �r�| �|� d S t |dd �}|d u�r|| j�}n0t |dd �}|d u�r.|� }ntd|j|f ��t|t��r\| �||� d S t|t��sttd| ��t|�} d| k�r�d k�s�n td | ��| j|d |i� d S ) Nr �reducer_overrider � __reduce_ex__� __reduce__zCan't pickle %r object: %rz %s must return string or tupler� � z2Tuple returned by %s must have two to six elementsr� )r� r{ � persistent_id� save_persr� r� r� r� �NotImplementedr� �type�dispatchr � issubclass� save_globalr� r r � isinstance�str�tupler� � save_reduce) r$ r� �save_persistent_id�pidrn �rv�reduce�tr� �lr r r r� sZ � �z _Pickler.savec C s d S r! r r� r r r r� ] s z_Pickler.persistent_idc C s` | j r | j|dd� | �t� n<z | �tt|��d� d � W n tyZ td��Y n0 d S )NF)r� r� r� �2persistent IDs in protocol 0 must be ASCII strings) r� r� r� � BINPERSID�PERSIDr� r� �UnicodeEncodeErrorr �r$ r� r r r r� a s �z_Pickler.save_persc C sD t |t�std��t|�s"td��| j}| j} t|dd�} | jdkr�| dkr�|\} }} t| d�sntd� | ���|d ur�| |j ur�td � | ���| jd kr�|| � ||� || � | t � n2t | j | g|�R i | ��}||� |d � | t� n�| jdk�rd| d k�rd|d } t| d��s td��|d u�r>| |j u�r>td��|dd � }|| � ||� | t� n||� ||� | t� |d u�r�t|�| jv �r�| t| �| jt|� d � � n | �|� |d u�r�| �|� |d u�r�| �|� |d u�r@|d u �r||� | t� n0||� ||� ||� | t� | t� | 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 r � __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�TUPLE2) r$ �func�args�state� listitems� dictitems� state_setterr� r� r� � func_name�cls�kwargsr r r r� m sz �� �� " z_Pickler.save_reducec C s | � t� d S r! )r� �NONEr� r r r � save_none� s z_Pickler.save_nonec C s4 | j dkr | �|rtnt� n| �|r*tnt� d S )Nr� )r� r� �NEWTRUE�NEWFALSE�TRUE�FALSEr� r r r � save_bool� s z_Pickler.save_boolc C s. | j r~|dkrN|dkr.| �ttd|� � d S |dkrN| �ttd|� � d S d| krbdkr~n n| �ttd|� � d S | jd kr�t|�}t|�}|d k r�| �t td|� | � n| �t td|� | � d S d| kr�dk�rn n | �t t |�� d � d � n | �tt |�� d � d � d S )Nr r� r� � �<Hi �i����<ir� r� r� r� s L )r� r� �BININT1r �BININT2�BININTr� r� r� �LONG1�LONG4�INTr� r� �LONG�r$ r� �encodedr� r r r � save_long� s* z_Pickler.save_longc C s< | j r | �ttd|� � n | �tt|��d� d � d S )N�>dr� r� )r� r� �BINFLOATr �FLOATr� r� r� r r r � save_float s z_Pickler.save_floatc C s� | j dk r@|s | jtd|d� n | jtjt|d�df|d� d S t|�}|dkrj| �tt d|� | � nf|dkr�| j dkr�| � t t d |� |� n<|| j j kr�| � tt d |� |� n| �tt d |� | � | �|� d S ) Nr� r �r� �latin1r� r� � �� r r} r� )r� r� �bytes�codecsr� r� r� r� �SHORT_BINBYTESr r� � BINBYTES8r� r~ �BINBYTESr� �r$ r� r� r r r � save_bytes s � z_Pickler.save_bytesc C s� | j dk r:|s | jtd|d� n| jtt|�f|d� d S t|�}|| jjkrf| �tt d|� |� n| � tt d|� | � d S )Nr r r7 r} ) r� r� � bytearrayr: r� r� r~ r� � BYTEARRAY8r r� r? r r r �save_bytearray) s z_Pickler.save_bytearrayc C s� | j dk rtd��|�� �~}|js*td��d}| jd urFt| �|��}|rp|jr`| �|�� � q�| � |�� � n| � t � |jr�| � t � W d � n1 s�0 Y d S )Nr z0PickleBuffer can only pickled with protocol >= 5zHPickleBuffer can not be pickled when pointing to a non-contiguous bufferT) r� r �raw� contiguousr� �bool�readonlyr@ �tobytesrC r� � NEXT_BUFFER�READONLY_BUFFER)r$ r� �m�in_bandr r r �save_picklebuffer8 s z_Pickler.save_picklebufferc C s | j r�|�dd�}t|�}|dkrF| jdkrF| �ttd|� | � nf|dkrp| jdkrp| �ttd|� |� n<|| j j kr�| �t td|� |� n| �t td|� | � nT|� d d �}|� d d �}|� d d�}|� dd�}|� dd�}| �t |�d� d � | �|� d S )N�utf-8� surrogatepassr� r r� r9 r} r� �\z\u005c� z\u0000� z\u000a� z\u000d�z\u001a�raw-unicode-escaper� )r� r� r� r� r� �SHORT_BINUNICODEr r� � BINUNICODE8r� r~ � BINUNICODE�replace�UNICODEr� r0 r r r �save_strR s& �z_Pickler.save_strc C s: |s(| j r| �t� n| �tt � d S t|�}| j}| j}|dkr�| jdkr�|D ] }||� qRt |�|v r�| � |t |� d �}| �t | | � n| �t | � | � |� d S | j}|t� |D ] }||� q�t |�|v �r$| � |t |� d �}| j �r |t| � n|t |d | � d S |t� | � |� d S )Nr� r� r r� )r� r� � EMPTY_TUPLE�MARK�TUPLEr� r� r� r� r� r� r �_tuplesize2coder� �POP_MARK)r$ r� r� r� r� �elementr� r� r r r � save_tuplei s: z_Pickler.save_tuplec C s8 | j r| �t� n| �tt � | �|� | �|� d S r! )r� r� � EMPTY_LISTr] �LISTr� r r� r r r � save_list� s z_Pickler.save_listi� c C s� | j }| j}| js0|D ]}||� |t� qd S t|�}tt|| j��}t|�}|dkr||t � |D ] }||� qd|t � n|r�||d � |t� || jk r8d S q8d S �Nr� r ) r� r� r� �APPEND�iter�listr � _BATCHSIZEr� r] �APPENDS)r$ r� r� r� rn �it�tmpr� r r r r � s( z_Pickler._batch_appendsc C s< | j r| �t� n| �tt � | �|� | �|�� � d S r! )r� r� � EMPTY_DICTr] �DICTr� r r� r� r r r � save_dict� s z_Pickler.save_dictc C s� | j }| j}| js<|D ] \}}||� ||� |t� qd S t|�}tt|| j��}t|�}|dkr�|t � |D ]\}}||� ||� qp|t � n(|r�|d \}}||� ||� |t� || jk rDd S qDd S rf ) r� r� r� �SETITEMrh ri r rj r� r] �SETITEMS) r$ r� r� r� �k�vrl rm r� r r r r � s0 z_Pickler._batch_setitemsc C s� | j }| j}| jdk r0| jtt|�f|d� d S |t� | �|� t|�}tt || j ��}t |�}|dkr�|t � |D ] }||� qv|t � || j k rJd S qJd S �Nr r7 r )r� r� r� r� �setri � EMPTY_SETr� rh r rj r� r] �ADDITEMS)r$ r� r� r� rl �batchr� �itemr r r �save_set� s" z_Pickler.save_setc C s� | j }| j}| jdk r0| jtt|�f|d� d S |t� |D ] }||� q<t|�| jv r||t | � | jt|� d � � d S |t � | � |� d S ru ) r� r� r� r� � frozensetri r] r� r� r` r� � FROZENSETr� )r$ r� r� r� rz r r r �save_frozenset s z_Pickler.save_frozensetc C sT | j }| j}|d u r t|dd �}|d u r.|j}t||�}z(t|dd� tj| }t||�\}}W n, t t t fy� t d|||f �d �Y n 0 ||ur�t d|||f ��| j dk�rt�||f�} | �r| dkr�|ttd| � � n0| d k�r|ttd | � � n|ttd | � � d S |�d �d } ||u �r4| }| j d k�r^| �|� | �|� |t� n�||u�rz| �t|| f� n�| j dk�r�|tt|d� d t|d� d � n�| j�r�tj } tj } ||f| v �r�| ||f \}}n|| v �r�| | }z(|tt|d� d t|d� d � W n* t �yD t d||| j f �d �Y n0 | � |� 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� r� r& r' r( r� r r� rN r� 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_pickle�REVERSE_NAME_MAPPING�REVERSE_IMPORT_MAPPINGr r� ) r$ r� r� r� r� r� r� �obj2r� �code�lastname�r_name_mapping�r_import_mappingr r r r� s� ����� �� �� ���z_Pickler.save_globalc C s` |t d �u r | jt d|d�S |t t�u r:| jt tf|d�S |t d�u rV| jt d|d�S | �|�S )Nr! r7 .).)r� r� r� r� r� r r r � save_typea s z_Pickler.save_type)N)T)NNNNN)N)0r r r r% r� r r� r� r� r� r� r� r� r� r r� r% rF r2 r� r6 �floatr@ r: rC rA �_HAVE_PICKLE_BUFFERrM r r[ r� rb r� re ri rj r rp �dictr& r r{ rv r~ r| r� r� r r r r r r� � sf � 9 F � u 1 B r� c @ s� e Zd Zddddd�dd�Zdd � Zd d � Zd d � Zi Zdd� 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/d0� Z(d1d2� Z)e)ee*d < d3d4� Z+e+ee,d < d5d6� Z-e-ee.d < d7d8� Z/e/ee0d < d9d:� Z1e1ee2d < d;d<� Z3e3ee4d < d=d>� Z5e5ee6d < d?d@� Z7e7ee8d < dAdB� Z9e9ee:d < dCdD� Z;e;ee<d < dEdF� Z=e=ee>d < dGdH� Z?e?ee@d < dIdJ� ZAeAeeBd < dKdL� ZCeCeeDd < dMdN� ZEeEeeFd < dOdP� ZGeGeeHd < dQdR� ZIeIeeJd < dSdT� ZKeKeeLd < dUdV� ZMeMeeNd < dWdX� ZOeOeePd < dYdZ� ZQeQeeRd < d[d\� ZSeSeeTd < d]d^� ZUeUeeVd < d_d`� ZWeWeeXd < dadb� ZYdcdd� ZZeZee[d < dedf� Z\e\ee]d < dgdh� Z^e^ee_d < didj� Z`e`eead < dkdl� Zbebeecd < dmdn� Zdedeeed < dodp� Zfefeegd < dqdr� Zheheeid < dsdt� Zjejeekd < dudv� Zldwdx� Zmdydz� Zneneeod < d{d|� Zpepeeqd < d}d~� Zrereesd < dd�� 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�d�� Z�e�ee�d < d�d�� Z�e�ee�d < d�d�� Z�e�ee�d < dS )�� _UnpicklerT�ASCII�strictN�r� �encoding�errors�buffersc C sH |durt |�nd| _|j| _|j| _i | _|| _|| _d| _ || _ dS )a� This takes a binary file for reading a pickle data stream. The protocol version of the pickle is detected automatically, so no proto argument is needed. The argument *file* must have two methods, a read() method that takes an integer argument, and a readline() method that requires no arguments. Both methods should return bytes. Thus *file* can be a binary file object opened for reading, an io.BytesIO object, or any other custom object that meets this interface. The file-like object must have two methods, a read() method that takes an integer argument, and a readline() method that requires no arguments. Both methods should return bytes. Thus file-like object can be a binary file object opened for reading, a BytesIO object, or any other custom object that meets this interface. If *buffers* is not None, it should be an iterable of buffer-enabled objects that is consumed each time the pickle stream references an out-of-band buffer view. Such buffers have been given in order to the *buffer_callback* of a Pickler object. If *buffers* is None (the default), then the buffers are taken from the pickle stream, assuming they are serialized there. It is an error for *buffers* to be None if the pickle stream was produced with a non-None *buffer_callback*. Other optional arguments are *fix_imports*, *encoding* and *errors*, which are used to control compatibility support for pickle stream generated by Python 2. If *fix_imports* is True, pickle will try to map the old Python 2 names to the new names used in Python 3. The *encoding* and *errors* tell pickle how to decode 8-bit string instances pickled by Python 2; these default to 'ASCII' and 'strict', respectively. *encoding* can be 'bytes' to read these 8-bit string instances as bytes objects. Nr ) rh �_buffersr� �_file_readliner� � _file_readr� r� r� r� r� )r$ r� r� r� r� r� r r r r% r s 'z_Unpickler.__init__c C s� t | d�s td| jjf ��t| j| j�| _| jj| _| jj | _ | jj | _ g | _ g | _ | j j | _ d| _| j}| j}z&|d�}|s�t�||d | � qtW n* ty� } z|jW Y d}~S d}~0 0 dS )z�Read a pickled object representation from the open file. Return the reconstituted object hierarchy specified in the file. r� z4Unpickler.__init__() was not called by %s.__init__()r r� N)r� r r� r r� r� r� � _unframerr� r� r� � metastack�stack�appendr� r� �EOFErrorr r" )r$ r� r� �key�stopinstr r r r � s* � z_Unpickler.loadc C s | j }| j�� | _ | j j| _|S r! )r� r� �popr� �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 �persistent_load� s z_Unpickler.persistent_loadc C s: | � d�d }d| kr"tks0n td| ��|| _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| �d��\}|tjkr(td| ��| j�|� d S )Nr} � z frame size > sys.maxsize: %d)r r� r� r r� r� r� r� r r r r� � s z_Unpickler.load_framec C sJ z| � � d d� �d�}W n ty4 td��Y n0 | �| �|�� d S )Nr� r� r )r� �decode�UnicodeDecodeErrorr r� r� r r r r � load_persid� s � z_Unpickler.load_persidc C s | j �� }| �| �|�� d S r! )r� r� r� r� r r r r �load_binpersid� s z_Unpickler.load_binpersidc C s | � d � d S r! �r� rv r r r � load_none� s z_Unpickler.load_nonec C s | � d� d S )NFr� rv r r r � load_false� s z_Unpickler.load_falsec C s | � d� d S )NTr� rv r r r � load_true� s z_Unpickler.load_truec C sL | � � }|tdd � kr d}n |tdd � kr4d}n t|d�}| �|� d S )Nr� FTr )r� r$ r# r� r� )r$ r� �valr r r �load_int� s z_Unpickler.load_intc C s | � td| �d��d � d S )Nr( r r �r� r r� rv r r r � load_binint� s z_Unpickler.load_binintc C s | � | �d�d � d S rf )r� r� rv r r r � load_binint1 s z_Unpickler.load_binint1c C s | � td| �d��d � d S )Nr' r� r r� rv r r r � load_binint2 s z_Unpickler.load_binint2c C s@ | � � d d� }|r,|d dkr,|d d� }| �t|d�� d S )Nr� �L r )r� r� r� )r$ r� r r r � load_long s z_Unpickler.load_longc C s* | � d�d }| � |�}| �t|�� d S rf )r� r� r� r� r r r � load_long1 s z_Unpickler.load_long1c C s> t d| �d��\}|dk r"td��| �|�}| �t|�� d S )Nr( r r z#LONG pickle has negative byte count)r r� r r� r� r� r r r � load_long4 s z_Unpickler.load_long4c C s | � t| �� d d� �� d S �Nr� )r� r� r� rv r r r � load_float! s z_Unpickler.load_floatc C s | � td| �d��d � d S )Nr3 r� r r� rv r r r � load_binfloat% s z_Unpickler.load_binfloatc C s" | j dkr|S |�| j | j�S d S )Nr: )r� r� r� r# r r r �_decode_string) s z_Unpickler._decode_stringc C sl | � � d d� }t|�dkrF|d |d krF|d dv rF|dd� }ntd��| �| �t�|�d �� d S )Nr� r� r s "'r� z)the STRING opcode argument must be quoted)r� r� r r� r� r; � escape_decoder� r r r � load_string2 s (z_Unpickler.load_stringc C s@ t d| �d��\}|dk r"td��| �|�}| �| �|�� d S )Nr( r r z(BINSTRING pickle has negative byte count)r r� r r� r� �r$ r� r� r r r �load_binstring< s z_Unpickler.load_binstringc C s: t d| �d��\}|tkr&tdt ��| �| �|�� d S )Nr� r z2BINBYTES exceeds system's maximum size of %d bytes�r r� r r r� �r$ r� r r r � load_binbytesE s �z_Unpickler.load_binbytesc C s | � t| �� d d� d�� d S )Nr� rU )r� r� r� rv r r r � load_unicodeM s z_Unpickler.load_unicodec C sB t d| �d��\}|tkr&tdt ��| �t| �|�dd�� d S )Nr� r z4BINUNICODE exceeds system's maximum size of %d bytesrN rO �r r� r r r� r� r� r r r �load_binunicodeQ s �z_Unpickler.load_binunicodec C sB t d| �d��\}|tkr&tdt ��| �t| �|�dd�� d S )Nr} r� z5BINUNICODE8 exceeds system's maximum size of %d bytesrN rO r� r� r r r �load_binunicode8Y s �z_Unpickler.load_binunicode8c C s: t d| �d��\}|tkr&tdt ��| �| �|�� d S )Nr} r� z3BINBYTES8 exceeds system's maximum size of %d bytesr� r� r r r �load_binbytes8a s �z_Unpickler.load_binbytes8c C sF t d| �d��\}|tkr&tdt ��t|�}| �|� | �|� d S )Nr} r� z4BYTEARRAY8 exceeds system's maximum size of %d bytes)r r� r r rA r� r� )r$ r� �br r r �load_bytearray8i s � z_Unpickler.load_bytearray8c C sJ | j d u rtd��zt| j �}W n ty: td��Y n0 | �|� d S )NzLpickle stream refers to out-of-band data but no *buffers* argument was givenz not enough out-of-band buffers)r� r �next� StopIterationr� )r$ r� r r r �load_next_buffers s z_Unpickler.load_next_bufferc C sJ | j d }t|��$}|js(|�� | j d< W d � n1 s<0 Y d S r� )r� � memoryviewrG � toreadonly)r$ r� rK r r r �load_readonly_buffer~ s z _Unpickler.load_readonly_bufferc C s, | � d�d }| � |�}| �| �|�� d S rf )r� r� r� r� r r r �load_short_binstring� s z _Unpickler.load_short_binstringc C s" | � d�d }| �| � |�� d S rf )r� r� r� r r r �load_short_binbytes� s z _Unpickler.load_short_binbytesc C s* | � d�d }| �t| � |�dd�� d S )Nr� r rN rO )r� r� r� r� r r r �load_short_binunicode� s z _Unpickler.load_short_binunicodec C s | � � }| �t|�� d S r! )r� r� r� r� r r r � load_tuple� s z_Unpickler.load_tuplec C s | � d� d S )Nr r� rv r r r �load_empty_tuple� s z_Unpickler.load_empty_tuplec C s | j d f| j d< d S r� �r� rv r r r � load_tuple1� s z_Unpickler.load_tuple1c C s$ | j d | j d fg| j dd �< d S )Nr� r� r� rv r r r � load_tuple2� s z_Unpickler.load_tuple2c C s, | j d | j d | j d fg| j dd �< d S )N���r� r� r� rv r r r � load_tuple3� s z_Unpickler.load_tuple3c C s | � g � d S r! r� rv r r r �load_empty_list� s z_Unpickler.load_empty_listc C s | � i � d S r! r� rv r r r �load_empty_dictionary� s z _Unpickler.load_empty_dictionaryc C s | � t� � d S r! )r� rv rv r r r �load_empty_set� s z_Unpickler.load_empty_setc C s | � � }| �t|�� d S r! )r� r� r| r� r r r �load_frozenset� s z_Unpickler.load_frozensetc C s | � � }| �|� d S r! )r� r� r� r r r � load_list� s z_Unpickler.load_listc s4 | � � � � fdd�tdt� �d�D �}| �|� d S )Nc s i | ]}� | � |d �qS )r� r )rm r� �r� r r � <dictcomp>� s �z(_Unpickler.load_dict.<locals>.<dictcomp>r r� )r� �ranger� r� )r$ �dr r� r � load_dict� s �z_Unpickler.load_dictc C s� |st |t�rt|d�rlz ||� }W qv tyh } z,td|jt|�f t�� d ��W Y d }~qvd }~0 0 n |�|�}| � |� d S )N�__getinitargs__zin constructor for %s: %sr� ) r� r� r� r� r r� r� �exc_infor r� )r$ �klassr r" �errr r r � _instantiate� s � � � z_Unpickler._instantiatec C sL | � � d d� �d�}| � � d d� �d�}| �||�}| �|| �� � d S )Nr� r� )r� r� � find_classr� r� �r$ r� r� r� r r r � load_inst� s z_Unpickler.load_instc C s" | � � }|�d�}| �||� d S �Nr )r� r� r� )r$ r r r r r �load_obj� s z_Unpickler.load_objc C s4 | j �� }| j �� }|j|g|�R � }| �|� d S r! �r� r� r r� )r$ r r r� r r r � load_newobj� s z_Unpickler.load_newobjc C sD | j �� }| j �� }| j �� }|j|g|�R i |��}| �|� d S r! r� )r$ r r r r� r r r �load_newobj_ex� s z_Unpickler.load_newobj_exc C sF | � � d d� �d�}| � � d d� �d�}| �||�}| �|� d S )Nr� rN )r� r� r� r� r� r r r � load_global� s z_Unpickler.load_globalc C sJ | j �� }| j �� }t|�tus,t|�tur4td��| �| �||�� d S )NzSTACK_GLOBAL requires str)r� r� r� r� r r� r� )r$ r� r� r r r �load_stack_global� s z _Unpickler.load_stack_globalc C s | � d�d }| �|� d S rf )r� � get_extension�r$ r� r r r � load_ext1 s z_Unpickler.load_ext1c C s t d| �d��\}| �|� d S )Nr' r� �r r� r� r� r r r � load_ext2 s z_Unpickler.load_ext2c C s t d| �d��\}| �|� d S )Nr( r r� r� r r r � load_ext4 s z_Unpickler.load_ext4c C sp g }t �||�}||ur&| �|� d S t�|�}|sP|dkrDtd��td| ��| j|� }|t |< | �|� d S )Nr zEXT specifies code <= 0z unregistered extension code %d)r r� r� r r r� r� )r$ r� �nilr� r� r r r r� s z_Unpickler.get_extensionc C s� t �d||� | jdk rT| jrT||ftjv r@tj||f \}}n|tjv rTtj| }t|dd� | jdkr~tt j | |�d S t t j | |�S d S )Nzpickle.find_classr� r r r ) r� �auditr� r� r� � NAME_MAPPING�IMPORT_MAPPINGr� r� r� r� )r$ r� r� r r r r� # s z_Unpickler.find_classc C s& | j }|�� }|d }||� |d< d S r� �r� r� )r$ r� r r r r r � load_reduce1 s z_Unpickler.load_reducec C s | j r| j d= n| �� d S r� )r� r� rv r r r �load_pop8 s z_Unpickler.load_popc C s | � � d S r! )r� rv r r r � load_pop_mark? s z_Unpickler.load_pop_markc C s | � | jd � d S r� )r� r� rv r r r �load_dupC s z_Unpickler.load_dupc C sT t | �� d d� �}z| �| j| � W n& tyN d|� �}t|�d �Y n0 d S )Nr� � Memo value not found at index )r� r� r� r� r� r )r$ r� �msgr r r �load_getG s z_Unpickler.load_getc C s` | � d�d }z| �| j| � W n8 tyZ } z d|� �}t|�d �W Y d }~n d }~0 0 d S )Nr� r r )r� r� r� r� r �r$ r� �excr r r r � load_bingetP s z_Unpickler.load_bingetc C sd t d| �d��\}z| �| j| � W n8 ty^ } z d|� �}t|�d �W Y d }~n d }~0 0 d S )Nr� r r )r r� r� r� r� r r r r r �load_long_bingetY s z_Unpickler.load_long_bingetc C s8 t | �� d d� �}|dk r$td��| jd | j|< d S )Nr� r znegative PUT argument)r� r� r� r� r� r� r r r �load_putb s z_Unpickler.load_putc C s2 | � d�d }|dk r td��| jd | j|< d S )Nr� r znegative BINPUT argumentr� )r� r� r� r� r� r r r � load_binputi s z_Unpickler.load_binputc C s6 t d| �d��\}|tkr"td��| jd | j|< d S )Nr� r z negative LONG_BINPUT argumentr� )r r� r r� r� r� r� r r r �load_long_binputp s z_Unpickler.load_long_binputc C s | j }| jd |t|�< d S r� )r� r� r� )r$ r� r r r � load_memoizew s z_Unpickler.load_memoizec C s$ | j }|�� }|d }|�|� d S r� )r� r� r� )r$ r� r" ri r r r � load_append| s z_Unpickler.load_appendc C sX | � � }| jd }z |j}W n ty. Y n0 ||� d S |j}|D ] }||� qFd S r� )r� r� �extendr� r� )r$ r� �list_objr r� rz r r r � load_appends� s z_Unpickler.load_appendsc C s* | j }|�� }|�� }|d }|||< d S r� r )r$ r� r" r� r� r r r � load_setitem� s z_Unpickler.load_setitemc C s@ | � � }| jd }tdt|�d�D ]}||d ||| < q"d S )Nr� r r� r� )r� r� r� r� )r$ r� r� r� r r r � load_setitems� s z_Unpickler.load_setitemsc C sD | � � }| jd }t|t�r(|�|� n|j}|D ] }||� q2d S r� )r� r� r� rv �update�add)r$ r� �set_objr rz r r r � load_additems� s z_Unpickler.load_additemsc C s� | j }|�� }|d }t|dd �}|d ur6||� d S d }t|t�rXt|�dkrX|\}}|r�|j}tj}|� � D ]*\}} t |�t u r�| |||�< qp| ||< qp|r�|� � D ]\}} t ||| � q�d S )Nr� � __setstate__r� ) r� r� r� r� r� r� �__dict__r� �internr� r� r� �setattr) r$ r� r �inst�setstate� slotstate� inst_dictr rs rt r r r � load_build� s( z_Unpickler.load_buildc C s" | j �| j� g | _| jj| _d S r! )r� r� r� rv r r r � load_mark� s z_Unpickler.load_markc C s | j �� }t|��d S r! )r� r� r r# r r r � load_stop� s z_Unpickler.load_stop)�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� r5 r� r4 r� r� �STRINGr� � BINSTRINGr� r> r� rZ r� rX r� rW r� r= r� rB r� rI r� rJ r� �SHORT_BINSTRINGr� r< r� rV r� r^ r� r\ r� �TUPLE1r� r r� �TUPLE3r� rc r� rn r� rw r� r} r� rd r� ro 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 rg r rk r rq r rr r rx r& r r' r] r( r� r r r r r� p s& � 0 r� r� c C s t ||||d��| � d S �Nr� )r� r )r� r� r� r� r� r r r �_dump� s ��r2 c C s* t �� }t||||d��| � |�� }|S r1 )rt ru r� r �getvalue)r� r� r� r� r� �resr r r �_dumps� s ��r5 r� r� r� c C s t | ||||d��� S )N�r� r� r� r� )r� r )r� r� r� r� r� r r r �_load� s �r7 c C s2 t | t�rtd��t�| �}t|||||d��� S )Nz%Can't load pickle from unicode stringr6 )r� r� r� rt ru r� r )�sr� r� r� r� r� r r r �_loads� s �r9 c C s dd l } | �� S r� )�doctest�testmod)r: r r r �_test s r< r� z$display contents of the pickle files)� description� pickle_file�br�*zthe pickle file)r� �nargs�helpz-tz--test� store_truezrun self-test suite)�actionrB z-vz)run verbosely; only affects self-test run)N)N)�r �typesr �copyregr r r r � itertoolsr � functoolsr r� r �structr r rk rt r; r� �__all__�_pickler r� r� r� r: rA � bytes_types�format_version�compatible_formatsr� r� � Exceptionr r r r Zorg.python.corer& r] r� r r` r0 r5 r. r+ r) r/ r* r r r r r) r* r+ rZ rX rg r r� ro rn rk r� r� r. r� rd rc r/ r� r� r� rq r^ r\ rr r4 r# r$ r� r r� r� r� r, r r- r! r" r, r- r_ r> r< rV rW r= rw rx r} r r� r� r� rB rI rJ r �dirrq r� r� r� r� r� r� r� r2 r5 r7 r9 r r r r r r r< r �argparse�ArgumentParser�parser� add_argument�FileType� parse_argsr �testr> � print_helpZpprintr� r� r r r r �<module> sB ?; ^ i� � 0 � ���