3
���h� � 9 @ s� d Z ddlZddlZddlZddlZddlmZmZ ddl Z ddl
m Z ddl m
Z
erhddl mZmZ dd� Zd ZG d
d � d �ZG d d
� d
�Zd]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�i ZG dJdK� dKej�Zejej�e_eejejd < dLdM� Zeejejd < dNdO� Zyeejej d < W n e e fk
�r� Y nX d�ee e dQ�dRdS�Z!dTdUdVdW�e"e e e dX�dYdZ�Z#ej$d[d\� �Z%dS )�z/
Support pre-0.12 series pickle compatibility.
� N)�
TYPE_CHECKING�Optional)�
BaseOffset)�Index)� DataFrame�Seriesc C s� | j }|j� }|d }t|�r8t|d �tkr8|d j}y||� |d< d S tk
r� } zvd}|t|�kr�y|d }tj|�|d< d S tk
r� Y q�X n,|r�t |d t
�r�|d }|j|� |d< d S � W Y d d }~X nX d S )N� r z:_reconstruct: First argument must be a sub-type of ndarray���r r r ) �stack�pop�len�type�__name__� TypeError�str�object�__new__�
issubclassr )�selfr
�args�func�n�err�msg�cls� r �=/tmp/pip-build-5_djhm0z/pandas/pandas/compat/pickle_compat.py� load_reduce s,
r z�
Loading a saved '{cls}' as a {new} with sparse values.
'{cls}' is now removed. You should re-save this dataset in its new format.
c @ s e Zd Zdd�dd�ZdS )�_LoadSparseSeriesr )�returnc C s0 ddl m} tjtjddd�tdd� |td�S ) Nr )r � SparseSeriesr )r �new� )�
stacklevel)Zdtype)�pandasr �warnings�warn� _sparse_msg�format�
FutureWarningr )r r r r r r C s z_LoadSparseSeries.__new__N)r �
__module__� __qualname__r r r r r r = s r c @ s e Zd Zdd�dd�ZdS )�_LoadSparseFramer )r c C s, ddl m} tjtjddd�tdd� |� S )Nr )r �SparseDataFramer )r r! r" )r# )r$ r r% r& r' r( r) )r r r r r r U s z_LoadSparseFrame.__new__N)r r* r+ r r r r r r, O s r, �pandas.core.sparse.array� SparseArray�pandas.core.arrays�pandas.core.base�
FrozenNDArray�numpy�ndarray�pandas.core.indexes.frozen�
FrozenList�pandas.core.series�
TimeSeriesr �pandas.sparse.series�SparseTimeSeries�pandas.core.sparse.seriesr �pandas._sparse�
BlockIndex�pandas._libs.sparse� pandas.tslib� Timestamp�pandas._libs.tslib�pandas._period�Period�pandas._libs.tslibs.period�pandas._libs.period�__nat_unpickle�pandas._libs.tslibs.nattype�pandas.sparse.array�pandas.core.arrays.sparse�pandas.compat.pickle_compat�pandas.sparse.framer- �pandas.core.sparse.frame�pandas.indexes.base�
_new_Index�pandas.core.indexes.baser �pandas.indexes.numeric�
Int64Index�pandas.core.indexes.numeric�pandas.indexes.range�
RangeIndex�pandas.core.indexes.range�pandas.indexes.multi�
MultiIndex�pandas.core.indexes.multi�pandas.tseries.index�_new_DatetimeIndex� pandas.core.indexes.datetimes�
DatetimeIndex�pandas.tseries.period� PeriodIndex�pandas.core.indexes.period�pandas.core.categorical� Categorical�pandas.tseries.tdi�TimedeltaIndex� pandas.core.indexes.timedeltas� Float64Indexc s e Zd Z� fdd�Z� ZS )� Unpicklerc s& ||f}t j||�\}}t� j||�S )N)�_class_locations_map�get�super�
find_class)r �module�name�key)� __class__r r rj � s zUnpickler.find_class)r r* r+ rj �
__classcell__r r )rn r rf � s rf c C sH | j j� }| j d }t|t�r*tj|�}n|j|f|�� }|| j d< d S )Nr r r )r
r r r r r )r r r �objr r r � load_newobj� s
rq c C sT | j j� }| j j� }| j j� }t|t�r4tj|�}n|j|f|�|�}| j|� d S )N)r
r r r r r �append)r �kwargsr r rp r r r �load_newobj_ex� s
rt F)�encoding�
is_verbosec C sV y6| j d� |dk r"t| |d�}nt| �}||_|j� S ttfk
rP � Y nX dS )z�
Load a pickle, with a provided encoding,
Parameters
----------
fh : a filelike object
encoding : an optional encoding
is_verbose : show exception output
r N)ru )�seekrf rv �load�
ValueErrorr )�fhru rv Zupr r r rx � s
rx T�ASCII�strict)� fix_importsru �errors)� bytes_objectr} ru r~ c C s t j| �}t||||d�j� S )z%
Analogous to pickle._loads.
)r} ru r~ )�io�BytesIOrf rx )r r} ru r~ �fdr r r �loads� s
r� c c s$ t j} ztt _dV W d| t _X dS )z8
Temporarily patch pickle to use our unpickler.
N)�pklr� )Z
orig_loadsr r r � patch_pickle s
r� )r. r/ )r0 r/ )r1 r2 )r3 r4 )r5 r2 )r3 r4 )r1 r6 )r5 r6 )r7 r8 )r7 r )r9 r: )r; r )r< r= )r> r= )r? r@ )rA r@ )rB rC )rD rC )rE rC )rD rC )r? rF )rG rF )rA rF )rG rF )rH r/ )rI r/ )r9 r )rJ r )rK r- )rL r, )rM rN )rO rN )rM r )rO r )rP rQ )rR rQ )rS rT )rU rT )rV rW )rX rW )rY rZ )r[ rZ )rY r\ )r[ r\ )r] r^ )r_ r^ )r` ra )r0 ra )rb rc )rd rc )rP re )rR re )r; r )rJ r )rL r- )rJ r, )NF)&�__doc__�
contextlib�copyr� �pickler� Ztypingr r r% Zpandas._libs.tslibsr r$ r r r r r' r r, rg �
_Unpicklerrf �dispatch�REDUCErq �NEWOBJrt � NEWOBJ_EX�AttributeError�KeyErrorr �boolrx �bytesr� �contextmanagerr� r r r r �<module> s� &
|