o i@sdZddlmZddlZddlmZddlmZmZm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZdd lmZerFdd lmZ d d!ddZ d ddddd"ddZdS)#z orc compat ) annotationsN) ModuleType) TYPE_CHECKINGAnyLiteral)FilePath ReadBuffer WriteBuffer)import_optional_dependency)is_categorical_dtypeis_interval_dtypeis_period_dtypeis_unsigned_integer_dtype) get_handle) DataFramepathFilePath | ReadBuffer[bytes]columnslist[str] | NonereturnrcKs^td}t|ddd}||j}|jdd|i|WdS1s(wYdS)a Load an ORC object from the file path, returning a DataFrame. .. versionadded:: 1.0.0 Parameters ---------- path : 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.orc``. columns : list, default None If not None, only these columns will be read from the file. **kwargs Any additional kwargs are passed to pyarrow. Returns ------- DataFrame Notes ----- Before using this function you should read the :ref:`user guide about ORC ` and :ref:`install optional dependencies `. pyarrow.orcrbFis_textrN)r rORCFilehandleread to_pandas)rrkwargsorchandlesorc_filerrH/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/io/orc.pyread_orc s   $r$pyarrow)engineindex engine_kwargsdf$FilePath | WriteBuffer[bytes] | Noner&Literal['pyarrow']r' bool | Noner(dict[str, Any] | None bytes | Nonec CsB|dur |jjddu}|duri}|jD]}t|s't|s't|s't|r+tdq|dkr4tdt |dd}t d}|du}|rHt }|dusNJt |d d d 4}t |ts]Jz|j|jj||d |jfi|Wnty} ztd| d} ~ wwWdn1swY|rt |t j sJ|SdS) a Write a DataFrame to the ORC format. .. versionadded:: 1.5.0 Parameters ---------- df : DataFrame The dataframe to be written to ORC. Raises NotImplementedError if dtype of one or more columns is category, unsigned integers, intervals, periods or sparse. path : str, file-like object or None, default None If a string, it will be used as Root Directory path when writing a partitioned dataset. By file-like object, we refer to objects with a write() method, such as a file handle (e.g. via builtin open function). If path is None, a bytes object is returned. engine : str, default 'pyarrow' ORC library to use. Pyarrow must be >= 7.0.0. index : bool, optional If ``True``, include the dataframe's index(es) in the file output. If ``False``, they will not be written to the file. If ``None``, similar to ``infer`` the dataframe's index(es) will be saved. However, instead of being saved as values, the RangeIndex will be stored as a range in the metadata so it doesn't require much space and is faster. Other indexes will be included as columns in the file output. engine_kwargs : dict[str, Any] or None, default None Additional keyword arguments passed to :func:`pyarrow.orc.write_table`. Returns ------- bytes if no path argument is provided else None Raises ------ NotImplementedError Dtype of one or more columns is category, unsigned integers, interval, period or sparse. ValueError engine is not pyarrow. Notes ----- * Before using this function you should read the :ref:`user guide about ORC ` and :ref:`install optional dependencies `. * This function requires `pyarrow `_ library. * For supported dtypes please refer to `supported ORC features in Arrow `__. * Currently timezones in datetime columns are not preserved when a dataframe is converted into ORC files. Nrz6The dtype of one or more columns is not supported yet.r%zengine must be 'pyarrow'z7.0.0) min_versionrwbFr)preserve_index)r'namesdtypesr r r rNotImplementedError ValueErrorr ioBytesIOr isinstancer write_tableTable from_pandasr TypeErrorgetvalue) r)rr&r'r(dtyper was_noner!errr#to_orcGs`>      rA)N)rrrrrr) r)rrr*r&r+r'r,r(r-rr.)__doc__ __future__rr6typesrtypingrrrpandas._typingrrr pandas.compat._optionalr pandas.core.dtypes.commonr r r rpandas.io.commonrpandasrr$rArrrr#s&      )