B
    Y0d              
   @  s   d Z ddlmZ ddlmZmZ ddlmZmZm	Z	 ddl
mZ ddlmZ er\ddlmZ G dd	 d	ed
Ze	dddddddd	dddZe	dddddddddddZdddddddddddZdS )z#
Read SAS sas7bdat or xport files.
    )annotations)ABCMetaabstractmethod)TYPE_CHECKINGHashableoverload)FilePathOrBuffer)stringify_path)	DataFramec               @  s:   e Zd ZdZedddZedd Zdd Zd	d
 ZdS )
ReaderBasez>
    Protocol for XportReader and SAS7BDATReader classes.
    Nc             C  s   d S )N )selfZnrowsr   r   I/var/www/html/venv/lib/python3.7/site-packages/pandas/io/sas/sasreader.pyread   s    zReaderBase.readc             C  s   d S )Nr   )r   r   r   r   close"   s    zReaderBase.closec             C  s   | S )Nr   )r   r   r   r   	__enter__&   s    zReaderBase.__enter__c             C  s   |    d S )N)r   )r   exc_type	exc_value	tracebackr   r   r   __exit__)   s    zReaderBase.__exit__)N)	__name__
__module____qualname____doc__r   r   r   r   r   r   r   r   r   r      s   r   )	metaclass.r   z
str | NonezHashable | Noneintbool)filepath_or_bufferformatindexencoding	chunksizeiteratorreturnc             C  s   d S )Nr   )r   r   r   r    r!   r"   r   r   r   read_sas-   s    	r$   NonezDataFrame | ReaderBasec             C  s   d S )Nr   )r   r   r   r    r!   r"   r   r   r   r$   9   s    	NFz
int | Nonec          	   C  s   |dkrVd}t | } t| ts&t||  }|dr>d}n|drNd}ntd| dkrdd	lm} || |||d
}	n2| dkrddlm	}
 |
| |||d
}	ntd|s|r|	S |	
 |	
 S Q R X dS )a)  
    Read SAS files stored as either XPORT or SAS7BDAT format files.

    Parameters
    ----------
    filepath_or_buffer : str, path object or file-like object
        Any valid string path is acceptable. 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.sas``.

        If you want to pass in a path object, pandas accepts any
        ``os.PathLike``.

        By file-like object, we refer to objects with a ``read()`` method,
        such as a file handle (e.g. via builtin ``open`` function)
        or ``StringIO``.
    format : str {'xport', 'sas7bdat'} or None
        If None, file format is inferred from file extension. If 'xport' or
        'sas7bdat', uses the corresponding format.
    index : identifier of index column, defaults to None
        Identifier of column that should be used as index of the DataFrame.
    encoding : str, default is None
        Encoding for text data.  If None, text data are stored as raw bytes.
    chunksize : int
        Read file `chunksize` lines at a time, returns iterator.

        .. versionchanged:: 1.2

            ``TextFileReader`` is a context manager.
    iterator : bool, defaults to False
        If True, returns an iterator for reading the file incrementally.

        .. versionchanged:: 1.2

            ``TextFileReader`` is a context manager.

    Returns
    -------
    DataFrame if iterator=False and chunksize=None, else SAS7BDATReader
    or XportReader
    NzVIf this is a buffer object rather than a string name, you must specify a format stringz.xptZxportz	.sas7bdatZsas7bdatz"unable to infer format of SAS filer   )XportReader)r   r    r!   )SAS7BDATReaderzunknown SAS format)r	   
isinstancestr
ValueErrorlowerendswithZpandas.io.sas.sas_xportr&   Zpandas.io.sas.sas7bdatr'   r   )r   r   r   r    r!   r"   Zbuffer_error_msgfnamer&   readerr'   r   r   r   r$   E   s<    2




).....).....)NNNNF)r   
__future__r   abcr   r   typingr   r   r   Zpandas._typingr   Zpandas.io.commonr	   Zpandasr
   r   r$   r   r   r   r   <module>   s2               