
    .i0                       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 ddlmZ ddlmZ e	r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d       G d ded   e             Zeddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 dd       Zeddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 dd       Z ed      ddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 dd       Zy)z#
Read SAS sas7bdat or xport files.
    )annotations)ABCabstractmethod)Iterator)TYPE_CHECKINGSelfoverload)
set_module)stringify_path)Hashable)TracebackType)CompressionOptionsFilePath
ReadBuffer)	DataFramezpandas.api.typingc                  V    e Zd ZdZeddd       Zed	d       Zd
dZ	 	 	 	 	 	 	 	 ddZy)	SASReaderz<
    Abstract class for XportReader and SAS7BDATReader.
    Nc                     y N )selfnrowss     U/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/io/sas/sasreader.pyreadzSASReader.read)   s    ;>    c                     y r   r   r   s    r   closezSASReader.close,   s     r   c                    | S r   r   r   s    r   	__enter__zSASReader.__enter__/   s    r   c                $    | j                          y r   )r   )r   exc_type	exc_value	tracebacks       r   __exit__zSASReader.__exit__2   s     	

r   r   )r   
int | Nonereturnr   )r'   None)r'   r   )r"   ztype[BaseException] | Noner#   zBaseException | Noner$   zTracebackType | Noner'   r(   )	__name__
__module____qualname____doc__r   r   r   r    r%   r   r   r   r   r   #   sU     > >   , ( (	
 
r   r   r   .)formatindexencoding	chunksizeiteratorcompressionc                    y r   r   filepath_or_bufferr-   r.   r/   r0   r1   r2   s          r   read_sasr6   ;   s     r   c                    y r   r   r4   s          r   r6   r6   H   s      r   pandasNFinferc                  |Td}t        |       } t        | t              st        |      | j	                         }d|v rd}nd|v rd}nt        d|      |j	                         dk(  rdd	lm}	  |	| ||||
      }
n2|j	                         dk(  rddlm}  || ||||
      }
nt        d      |s|r|
S |
5  |
j                         cddd       S # 1 sw Y   yxY w)a
  
    Read SAS files stored as either XPORT or SAS7BDAT format files.

    Parameters
    ----------
    filepath_or_buffer : 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.sas7bdat``.
    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.
    iterator : bool, defaults to False
        If True, returns an iterator for reading the file incrementally.
    compression : str or dict, default 'infer'
        For on-the-fly decompression of on-disk data. If 'infer' and
        'filepath_or_buffer' is path-like, then detect compression from the
        following extensions: '.gz', '.bz2', '.zip', '.xz', '.zst', '.tar',
        '.tar.gz', '.tar.xz' or '.tar.bz2' (otherwise no compression).
        Set to ``None`` for no decompression.
        Can also be a dict with key ``'method'`` set to one of {``'zip'``,
        ``'gzip'``, ``'bz2'``, ``'zstd'``, ``'xz'``, ``'tar'``} and other
        key-value pairs are forwarded to ``zipfile.ZipFile``,
        ``gzip.GzipFile``, ``bz2.BZ2File``, ``zstandard.ZstdCompressor``,
        ``lzma.LZMAFile`` or ``tarfile.TarFile``, respectively.
        As an example, the following could be passed for faster compression
        and to create a reproducible gzip archive:
        ``compression={'method': 'gzip', 'compresslevel': 1, 'mtime': 1}``.

    Returns
    -------
    DataFrame, SAS7BDATReader, or XportReader
        DataFrame if iterator=False and chunksize=None, else SAS7BDATReader
        or XportReader, file format is inferred from file extension.

    See Also
    --------
    read_csv : Read a comma-separated values (csv) file into a DataFrame.
    read_excel : Read an Excel file into a pandas DataFrame.
    read_spss : Read an SPSS file into a pandas DataFrame.
    read_orc : Load an ORC object into a pandas DataFrame.
    read_feather : Load a feather-format object into a pandas DataFrame.

    Examples
    --------
    >>> df = pd.read_sas("sas_data.sas7bdat")  # doctest: +SKIP
    NzVIf this is a buffer object rather than a string name, you must specify a format stringz.xptxportz	.sas7bdatsas7bdatz2unable to infer format of SAS file from filename: r   )XportReader)r.   r/   r0   r2   )SAS7BDATReaderzunknown SAS format)
r   
isinstancestr
ValueErrorlowerpandas.io.sas.sas_xportr=   pandas.io.sas.sas7bdatr>   r   )r5   r-   r.   r/   r0   r1   r2   buffer_error_msgfnamer=   readerr>   s               r   r6   r6   U   s	   D ~C 	 ,,>?,c2-.."((*U?FE!FDUIN 
 ||~ 7#
 
:	%9#
 -..9	 {{}  s   8CC)r5   FilePath | ReadBuffer[bytes]r-   
str | Noner.   Hashable | Noner/   rI   r0   intr1   boolr2   r   r'   r   )r5   rH   r-   rI   r.   rJ   r/   rI   r0   r(   r1   rL   r2   r   r'   DataFrame | SASReader)r5   rH   r-   rI   r.   rJ   r/   rI   r0   r&   r1   rL   r2   r   r'   rM   )r,   
__future__r   abcr   r   collections.abcr   typingr   r   r	   pandas.util._decoratorsr
   pandas.io.commonr   r   typesr   pandas._typingr   r   r   r8   r   r   r6   r   r   r   <module>rV      s   # %  / +(#  !  %s  !. 
  &)	4	 	 		
 	 	 	 $	 	 
	 
  &)	 4	  	  		 
 	  	  	  $	  	  
	  H ! &-o4o o 	o
 o o o $o o or   