U
    ?vf$                     @  s   d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	m
Z
mZmZmZ ddlmZ ddlmZ ddlmZ dd	lZdd
lmZmZ ddlmZ ddlmZ eed dddddddddZeed dd	dd	ejfddddddddZd	S )z feather-format compat     )annotations)HashableSequence)lib)DtypeBackendFilePath
ReadBufferStorageOptionsWriteBuffer)import_optional_dependency)doc)check_dtype_backendN)	DataFrame
RangeIndex)_shared_docs)
get_handlestorage_options)r   r   zFilePath | WriteBuffer[bytes]r	   None)dfpathr   returnc              	   K  s   t d ddlm} t| ts&tdddh}| jjdksTt| j}td| d	| j	t
tt| svtd
| jjdk	rtd| jj|krtdt|d|dd}|j| |jf| W 5 Q R X dS )aK  
    Write a DataFrame to the binary Feather format.

    Parameters
    ----------
    df : DataFrame
    path : str, path object, or file-like object
    {storage_options}

        .. versionadded:: 1.2.0

    **kwargs :
        Additional keywords passed to `pyarrow.feather.write_feather`.

        .. versionadded:: 1.1.0
    pyarrowr   featherz'feather only support IO with DataFramesstringunicodeZint64z%feather does not support serializing zG for the index; you can .reset_index() to make the index into column(s)zfeather does not support serializing a non-default index for the index; you can .reset_index() to make the index into column(s)Nz=feather does not serialize index meta-data on a default indexz%feather must have string column nameswbFr   Zis_text)r   r   r   
isinstancer   
ValueErrorindexZdtypetypeequalsr   Z
from_rangerangelennamecolumnsZinferred_typer   Zwrite_featherhandle)r   r   r   kwargsr   Zvalid_typestyphandles r+   </tmp/pip-unpacked-wheel-vdrwu74i/pandas/io/feather_format.py
to_feather   s8    


   r-   TzFilePath | ReadBuffer[bytes]zSequence[Hashable] | NoneboolzDtypeBackend | lib.NoDefault)r   r&   use_threadsr   dtype_backendc           	   
   C  s   t d ddlm} t| t| d|dd}|tjkrZ|j|j|t	|dW  5 Q R  S |j
|j|t	|d}|dkrdd	lm} |j| jd
W  5 Q R  S |dkr|jtjd
W  5 Q R  S W 5 Q R X dS )a  
    Load a feather-format object from the file path.

    Parameters
    ----------
    path : str, path object, or file-like object
        String, path object (implementing ``os.PathLike[str]``), or file-like
        object implementing a binary ``read()`` function. The string could be a URL.
        Valid URL schemes include http, ftp, s3, and file. For file URLs, a host is
        expected. A local file could be: ``file://localhost/path/to/table.feather``.
    columns : sequence, default None
        If not provided, all columns are read.
    use_threads : bool, default True
        Whether to parallelize reading using multiple threads.
    {storage_options}

        .. versionadded:: 1.2.0

    dtype_backend : {{"numpy_nullable", "pyarrow"}}, defaults to NumPy backed DataFrames
        Which dtype_backend to use, e.g. whether a DataFrame should have NumPy
        arrays, nullable dtypes are used for all dtypes that have a nullable
        implementation when "numpy_nullable" is set, pyarrow is used for all
        dtypes if "pyarrow" is set.

        The dtype_backends are still experimential.

        .. versionadded:: 2.0

    Returns
    -------
    type of object stored in file
    r   r   r   rbFr   )r&   r/   Znumpy_nullable)_arrow_dtype_mapping)Ztypes_mapperN)r   r   r   r   r   r   
no_defaultread_featherr'   r.   Z
read_tableZpandas.io._utilr2   Z	to_pandasgetpdZ
ArrowDtype)	r   r&   r/   r   r0   r   r*   Zpa_tabler2   r+   r+   r,   r4   c   s4    (   
    r4   )N) __doc__
__future__r   typingr   r   Zpandas._libsr   Zpandas._typingr   r   r   r	   r
   Zpandas.compat._optionalr   Zpandas.util._decoratorsr   Zpandas.util._validatorsr   Zpandasr6   Zpandas.core.apir   r   Zpandas.core.shared_docsr   Zpandas.io.commonr   r-   r3   r4   r+   r+   r+   r,   <module>   s(    C