o թZh`@sddlmZddlZddlZddlZddlZddlmZmZm Z ddl Z ddl Z ddl m Z mZddlmZddlZddlmZerRddlmZmZddlmZmZ ddddZddddZddddZdd ddZdS)!) annotationsN) TYPE_CHECKINGAnyCallable) get_bz2_file get_lzma_file)import_optional_dependency) ensure_clean)FilePathReadPickleBuffer) DataFrameSeriesobjrpath"FilePath | ReadPickleBuffer | NonereturnDataFrame | SeriescCs^|}|durdtd}t|}t||t|WdS1s(wYdS)a Pickle an object and then read it again. Parameters ---------- obj : any object The object to pickle and then re-read. path : str, path object or file-like object, default None The path where the pickled object is written and then read. Returns ------- pandas object The original object that was pickled and then re-read. N__z __.pickle)uuiduuid4r pdZ to_pickleZ read_pickle)rr_pathZ temp_pathrJ/var/www/html/lang_env/lib/python3.10/site-packages/pandas/_testing/_io.pyround_trip_pickle's  $r str | NonecCsXtj}|dur d}t|}||||||}Wd|S1s%wY|S)a Write an object to file specified by a pathlib.Path and read it back Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. NZ ___pathlib___)pathlibPathr )writerreaderrrrrrrround_trip_pathlibAs   r cCsfddl}|dj}|durd}t|}||||||}Wd|S1s,wY|S)a Write an object to file specified by a py.path LocalPath and read it back. Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. rNzpy.pathZ___localpath___)pytestZ importorskiplocalr )rrrr!Z LocalPathrrrrround_trip_localpath\s    r#testdeststrNonec Cs|f}d}d}|dkrtj}d}||f}d}nI|dkr7tj}d}tj|d}t|} t||_|| f}d}n)|d kr?t j }n!|d krGt }n|d krQt d j }n|d krYt}ntd||||d} t| ||WddS1sywYdS)a Write data to a compressed file. Parameters ---------- compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd'} The compression type to use. path : str The file path to write the data. data : str The data to write. dest : str, default "test" The destination file (for ZIP only) Raises ------ ValueError : An invalid compression value was passed in. wbwritezipwwritestrtar)nameaddfilegzipbz2ZzstdZ zstandardxzzUnrecognized compression type: )modeN)zipfileZipFiletarfileTarFileTarInfoioBytesIOlensizer0GzipFilerropenr ValueErrorgetattr) compressionrdatar%argsr3methodZcompress_methodfilebytesfrrrwrite_to_compressedys8    "rH)N)rrrrrr)rr)r$)r%r&rr') __future__rr0r9rr6typingrrrrr4Z pandas.compatrrZpandas.compat._optionalrZpandasrZpandas._testing.contextsr Zpandas._typingr r r r rr r#rHrrrrs(