B
    x\	                 @   s*   d Z ddlmZ ddlmZ dddZdS )	z#
Read SAS sas7bdat or xport files.
    )compat)_stringify_pathNFc             C   s   |dkrXd}t | } t| tjs(t||  }|dr@d}n|drPd}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 )a  
    Read SAS files stored as either XPORT or SAS7BDAT format files.

    Parameters
    ----------
    filepath_or_buffer : string or file-like object
        Path to the SAS file.
    format : string {'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 : string, 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.
    iterator : bool, defaults to False
        If True, returns an iterator for reading the file incrementally.

    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)indexencoding	chunksize)SAS7BDATReaderzunknown SAS format)r   
isinstancer   Zstring_types
ValueErrorlowerendswithZpandas.io.sas.sas_xportr   Zpandas.io.sas.sas7bdatr   readclose)Zfilepath_or_bufferformatr   r   r   iteratorZbuffer_error_msgfnamer   readerr   data r   6lib/python3.7/site-packages/pandas/io/sas/sasreader.pyread_sas	   s6    



r   )NNNNF)__doc__Zpandasr   Zpandas.io.commonr   r   r   r   r   r   <module>   s    