U b @sdZddlmZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZer`ddlmZGdd d ed Ze dd d dd ddd dddZe dd d dd ddddddZdd d dd ddddddZdS)z# Read SAS sas7bdat or xport files. ) annotations)ABCMetaabstractmethod) TYPE_CHECKINGHashableoverload)FilePath ReadBuffer)stringify_path) DataFramec@s:eZdZdZed ddZeddZddZd d ZdS) ReaderBasez> Protocol for XportReader and SAS7BDATReader classes. NcCsdSN)selfZnrowsrr;/tmp/pip-unpacked-wheel-ck39h295/pandas/io/sas/sasreader.pyread!szReaderBase.readcCsdSr rrrrrclose%szReaderBase.closecCs|Sr rrrrr __enter__)szReaderBase.__enter__cCs |dSr )r)rexc_type exc_value tracebackrrr__exit__,szReaderBase.__exit__)N) __name__ __module__ __qualname____doc__rrrrrrrrrr s  r ) metaclass.zFilePath | ReadBuffer[bytes]z str | NonezHashable | Noneintbool)filepath_or_bufferformatindexencoding chunksizeiteratorreturncCsdSr rr r!r"r#r$r%rrrread_sas0s r(NonezDataFrame | ReaderBasecCsdSr rr'rrrr(<s NFz int | Nonec Cs|dkrVd}t|}t|ts&t||}|dr>d}n|drNd}ntd|dkrdd lm}|||||d } n2|dkrdd lm } | ||||d } ntd |s|r| S| | W5QRSQRXdS) 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.sas``. 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.xptZxportz .sas7bdatZsas7bdatz"unable to infer format of SAS filer) XportReader)r"r#r$)SAS7BDATReaderzunknown SAS format) r isinstancestr ValueErrorlowerendswithZpandas.io.sas.sas_xportr*Zpandas.io.sas.sas7bdatr+r) r r!r"r#r$r%Zbuffer_error_msgfnamer*readerr+rrrr(HsB,       ).....).....)NNNNF)r __future__rabcrrtypingrrrZpandas._typingrr Zpandas.io.commonr Zpandasr r r(rrrrs8