a oEb@sddlmZddlmZddlZddlmZddlZddlm Z ddl Z ddl Z ddl m Z mZddlZddlmZddlmZed d Zed d d dZeddddddZeddZeddZeddZed ddZGdddZdS)!) annotations)contextmanagerN)Path)rmtree)IOAny) set_option) get_handleccs<t|d|dd}|jVWdn1s.0YdS)a& Open a compressed file and return a file object. Parameters ---------- path : str The path where the file is read from. compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd', None} Name of the decompression to use Returns ------- file object rbF) compressionZis_textN)r handle)pathr r rX/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/_testing/contexts.pydecompress_filesrstrtzc#sTddlddlfdd}jd}||zdVW||n ||0dS)a Context manager for temporarily setting a timezone. Parameters ---------- tz : str A string representing a valid timezone. Examples -------- >>> from datetime import datetime >>> from dateutil.tz import tzlocal >>> tzlocal().tzname(datetime(2021, 1, 1)) # doctest: +SKIP 'IST' >>> with set_timezone('US/Eastern'): ... tzlocal().tzname(datetime(2021, 1, 1)) ... 'EST' rNcs@|dur*z jd=Wq<ty&Yq<0n|jd<dS)NTZ)environKeyErrortzsetrostimerrsetTZDs   zset_timezone..setTZr)rrrget)rrZorig_tzrrr set_timezone+s rFboolr)return_filelikekwargsc kstt}|durd}dtjtjtjdd|}||}| t |}|rp| ddt |fi|}z,|VWt |t s||r|n$t |t s||r|0dS)az Gets a temporary path and agrees to remove on close. This implementation does not use tempfile.mkstemp to avoid having a file handle. If the code using the returned path wants to delete the file itself, windows requires that no program has a file handle to it. Parameters ---------- filename : str (optional) suffix of the created file. return_filelike : bool (default False) if True, returns a file-like which is *always* cleaned. Necessary for savefig and other functions which want to append extensions. **kwargs Additional keywords are passed to open(). N)kmodezw+b)rtempfile gettempdirjoinrandomchoicesstring ascii_lettersdigitstouchr setdefaultopen isinstancecloseis_fileunlink)filenamerr folderr Z handle_or_strrrr ensure_cleanVs*    r6ccs^tjdd}z*|VWz t|WqZty4YqZ0n"z t|WntyVYn00dS)z{ Get a temporary directory path and agrees to remove on close. Yields ------ Temporary directory path r!)suffixN)r%mkdtemprOSError)Zdirectory_namerrrensure_clean_dirs     r:c csHttj}z dVWtjtj|ntjtj|0dS)z Get a context manager to safely set environment variables All changes will be undone on close, hence environment variables set within this contextmanager will neither persist nor change global state. N)dictrrclearupdate)Z saved_environrrr!ensure_safe_environment_variabless   r>c ksXddl}hd}||vr td|j|fi|zdVW||n ||0dS)au Context manager to temporarily register a CSV dialect for parsing CSV. Parameters ---------- name : str The name of the dialect. kwargs : mapping The parameters for the dialect. Raises ------ ValueError : the name of the dialect conflicts with a builtin one. See Also -------- csv : Python's CSV library. rN>excelz excel-tabunixz Cannot override builtin dialect.)csv ValueErrorregister_dialectunregister_dialect)namer rAZ_BUILTIN_DIALECTSrrrwith_csv_dialectsrFc cshddlm}|dur|j}|j}|j}td|||_zdVW||_td|n||_td|0dS)Nr) expressionszcompute.use_numexpr)Zpandas.core.computationrGZ _MIN_ELEMENTSZ USE_NUMEXPRr)ZuseZ min_elementsexprZolduseZoldminrrr use_numexprs   rIc@s(eZdZdZddZddZddZdS) RNGContexta, Context manager to set the numpy random number generator speed. Returns to the original value upon exiting the context manager. Parameters ---------- seed : int Seed for numpy.random.seed Examples -------- with RNGContext(42): np.random.randn() cCs ||_dSN)seed)selfrLrrr__init__szRNGContext.__init__cCstj|_tj|jdSrK)npr(Z get_state start_staterL)rMrrr __enter__s zRNGContext.__enter__cCstj|jdSrK)rOr(Z set_staterP)rMexc_type exc_value tracebackrrr__exit__szRNGContext.__exit__N)__name__ __module__ __qualname____doc__rNrQrUrrrrrJsrJ)NF)N) __future__r contextlibrrpathlibrr(shutilrr*r%typingrrnumpyrOZpandasrZpandas.io.commonr rrr6r:r>rFrIrJrrrrs4       *,   !