a Of`@sddlmZddlZddlZddlZddlZddlmZmZm Z ddl Z ddl Z ddl m Z mZddlmZddlZddlmZerddlmZmZddlmZmZdd d d d d dZdddddZdddddZddddddZdS)) annotationsN) TYPE_CHECKINGAnyCallable) get_bz2_file get_lzma_file)import_optional_dependency) ensure_clean)FilePathReadPickleBuffer) DataFrameSeriesrz"FilePath | ReadPickleBuffer | NonezDataFrame | Series)objpathreturncCs^|}|durdtd}t|&}t||t|WdS1sP0YdS)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_pathrL/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/_testing/_io.pyround_trip_pickle's   rz str | None)rcCsVtj}|durd}t|(}||||||}Wdn1sH0Y|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  *rcCsdddl}|dj}|dur d}t|(}||||||}Wdn1sV0Y|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 )rrrrZ LocalPathrrrrround_trip_localpath\s   *r teststrNone)destrc Cs|f}d}d}|dkr.tj}d}||f}d}n|dkrntj}d}tj|d}t|} t||_|| f}d}nR|d kr~t j }nB|d krt }n2|d krt d j }n|d krt}ntd||||d} t| ||Wdn1s0YdS)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)zipfileZipFiletarfileTarFileTarInfoioBytesIOlensizer-GzipFilerropenr ValueErrorgetattr) compressionrdatar$argsr0methodZcompress_methodfilebytesfrrrwrite_to_compressedys6    rE)N)N)N)r!) __future__rr-r6rr3typingrrrrr1Z pandas.compatrrZpandas.compat._optionalrZpandasrZpandas._testing.contextsr Zpandas._typingr r r r rrr rErrrrs&