U
    f/e"                     @  s  d Z ddlmZ ddlZddlZddlZddlZddlm	Z	 ddl
Z
ddlZddlmZ ddlmZ ddlmZ ddlmZmZmZ dd	lmZ e	rdd
lmZmZ dd ZdZG dd dZG dd dZdddddddddddddddddd d!d"d#d$d%d&d'd(dd)d*ZG d+d, d,ej Z!ee!j"e!_"ee!j"ej#d < d-d. Z$e$e!j"ej%d < d/d0 Z&ze&e!j"ej'd < W n e(e)fk
r   Y nX dBd2d3d4d5d6Z*d7d8d9d:d;d3d<d<d=d>d?Z+ej,d@dA Z-dS )Cz/
Support pre-0.12 series pickle compatibility.
    )annotationsN)TYPE_CHECKING)NDArrayBacked)
BaseOffset)Index)DatetimeArrayPeriodArrayTimedeltaArray)BlockManager)	DataFrameSeriesc                 C  s6  | j }| }|d }t|r8t|d tkr8|d j}z|| |d< W d S  tk
r0 } zd}|t|krz"|d }t||d< W W Y d S  tk
r   Y nX nv|rt	|d trt
|d tr|d }|j| |d< W Y @d S |rt
|d tr|d }tj| |d< W Y 
d S  W 5 d }~X Y nX d S )Nr   z:_reconstruct: First argument must be a sub-type of ndarray)stackpoplentype__name__	TypeErrorstrobject__new__
isinstance
issubclassr   r   r   )selfr   argsfuncnerrmsgcls r    ?/tmp/pip-unpacked-wheel-tiezk1ph/pandas/compat/pickle_compat.pyload_reduce!   s4    
 

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 Zd ZddddZdS )_LoadSparseSeriesr   returnc                 C  s0   ddl m} tjtjdddtdd |tdS )	Nr   )r   SparseSeriesr   r   new   
stackleveldtype)pandasr   warningswarn_sparse_msgformatFutureWarningr   )r   r   r    r    r!   r   T   s    z_LoadSparseSeries.__new__Nr   
__module____qualname__r   r    r    r    r!   r#   N   s   r#   c                   @  s   e Zd ZddddZdS )_LoadSparseFramer   r$   c                 C  s,   ddl m} tjtjdddtdd | S )Nr   )r   SparseDataFramer   r'   r)   r*   )r.   r   r/   r0   r1   r2   r3   )r   r   r    r    r!   r   f   s    z_LoadSparseFrame.__new__Nr4   r    r    r    r!   r7   `   s   r7   )pandas.core.arraysSparseArray)numpyZndarray)pandas.core.indexes.frozen
FrozenList)pandas.core.seriesr   zpandas.core.sparse.seriesr&   )zpandas._libs.sparse
BlockIndex)pandas._libs.tslib	Timestamp)zpandas._libs.tslibs.periodPeriod)zpandas._libs.tslibs.nattype__nat_unpickle)zpandas.core.arrays.sparser:   )pandas.compat.pickle_compatr#   )pandas.core.sparse.framer7   )pandas.core.indexes.base
_new_Index)rG   r   )pandas.core.indexes.numeric
Int64Index)zpandas.core.indexes.range
RangeIndex)zpandas.core.indexes.multi
MultiIndex)pandas.core.indexes.datetimes_new_DatetimeIndex)rM   DatetimeIndex)zpandas.core.indexes.periodPeriodIndex)r9   Categorical)zpandas.core.indexes.timedeltasTimedeltaIndex)rI   Float64Index)rE   r7   ))zpandas.core.sparse.arrayr:   )pandas.core.baseFrozenNDArray)r<   rU   )rT   r=   )r>   Z
TimeSeries)pandas.sparse.seriesZSparseTimeSeries)zpandas._sparser@   )pandas.tslibrB   )zpandas._periodrC   )zpandas._libs.periodrC   )rW   rD   )rA   rD   )zpandas.sparse.arrayr:   )rV   r&   )zpandas.sparse.framer8   )pandas.indexes.baserH   )rX   r   )pandas.indexes.numericrJ   )zpandas.indexes.rangerK   )zpandas.indexes.multirL   )pandas.tseries.indexrN   )rZ   rO   )zpandas.tseries.periodrP   )zpandas.core.categoricalrQ   )zpandas.tseries.tdirR   )rY   rS   r?   )rF   r8   c                      s   e Zd Z fddZ  ZS )	Unpicklerc                   s&   ||f}t ||\}}t ||S N)_class_locations_mapgetsuper
find_class)r   modulenamekey	__class__r    r!   r`      s    zUnpickler.find_class)r   r5   r6   r`   __classcell__r    r    rd   r!   r[      s   r[   c                 C  s   | j  }| j d }t|tr*t|}nt|trX|sXtjg dd}||||j	}n\t|t
r|stjg dd}||||j	}n.|tkr|s||dg d}n|j|f| }|| j d< d S )Nr   zM8[ns]r,   zm8[ns]r    F)r   r   r   r   r   r   r   nparrayr-   r	   r
   )r   r   r   objZarrr    r    r!   load_newobj   s    


rj   c                 C  sT   | j  }| j  }| j  }t|tr4t|}n|j|f||}| | d S r\   )r   r   r   r   r   r   append)r   kwargsr   r   ri   r    r    r!   load_newobj_ex   s    



rm   Fz
str | Nonebool)encoding
is_verbosec              	   C  sX   z8|  d |dk	r"t| |d}nt| }||_| W S  ttfk
rR    Y nX dS )z
    Load a pickle, with a provided encoding,

    Parameters
    ----------
    fh : a filelike object
    encoding : an optional encoding
    is_verbose : show exception output
    r   N)ro   )seekr[   rp   load
ValueErrorr   )fhro   rp   Zupr    r    r!   rr      s    


rr   TASCIIstrictfix_importsro   errorsbytesr   )bytes_objectrx   ro   ry   c                C  s   t | }t||||d S )z%
    Analogous to pickle._loads.
    rw   )ioBytesIOr[   rr   )r{   rx   ro   ry   fdr    r    r!   loads  s    

   r   c               
   c  s0   t j} ztt dt dV  W 5 tt d|  X dS )z8
    Temporarily patch pickle to use our unpickler.
    r   N)pklr   setattr)Z
orig_loadsr    r    r!   patch_pickle'  s
    
r   )NF).__doc__
__future__r   
contextlibcopyr|   pickler   typingr   r/   r;   rg   Zpandas._libs.arraysr   Zpandas._libs.tslibsr   r.   r   Zpandas.core.arraysr   r   r	   Zpandas.core.internalsr
   r   r   r"   r1   r#   r7   r]   
_Unpicklerr[   dispatchREDUCErj   NEWOBJrm   	NEWOBJ_EXAttributeErrorKeyErrorrr   r   contextmanagerr   r    r    r    r!   <module>   s   &V