U
    dd                     @   s@   d Z zddlmZ W n ek
r,   dZY nX G dd dZdS )z Tablib - DataFrame Support.
    )	DataFrameNc                   @   s:   e Zd ZdZdZedd Zed
ddZedd	 ZdS )DataFrameFormatdf)r   c                 C   sH   t dkrdS t|t rdS zt |  W dS  tk
rB   Y dS X dS )z,Returns True if given stream is a DataFrame.NFT)r   
isinstanceread
ValueError)clsstream r
   6/tmp/pip-unpacked-wheel-_3ph8cfy/tablib/formats/_df.pydetect   s    
zDataFrameFormat.detectNc                 C   s$   t dkrtdt |j|jd}|S )z-Returns DataFrame representation of DataBook.NzWDataFrame Format requires `pandas` to be installed. Try `pip install "tablib[pandas]"`.)columns)r   NotImplementedErrordictheaders)r   dsetindexZ	dataframer
   r
   r   
export_set   s    zDataFrameFormat.export_setc                 C   s   |   |jdd|_dS )zReturns dataset from DataFrame.records)ZorientN)ZwipeZto_dictr   )r   r   Z	in_streamr
   r
   r   
import_set%   s    zDataFrameFormat.import_set)N)	__name__
__module____qualname__title
extensionsclassmethodr   r   r   r
   r
   r
   r   r   
   s   
	r   )__doc__Zpandasr   ImportErrorr   r
   r
   r
   r   <module>   s
   
