o i@sddlmZddlmZddlZddlmZddlmZddl Z ddl m Z m Z m Z ddlZddlZddlmZddlmZed$d d Zed%ddZed&d'ddZed(ddZed)ddZed)ddZed*d)d d!ZGd"d#d#ZdS)+) annotations)contextmanagerN)Path)rmtree)IOAnyIterator) set_option) get_handlereturnIterator[IO[bytes]]ccs@t|d|dd }|jVWddS1swYdS)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) compressionis_textN)r handle)pathrrrS/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/_testing/contexts.pydecompress_files "rtzstrIterator[None]c#sTddlddlfdd}jd}||z dVW||dS||w)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' rNcsB|durzjd=WdStyYdSw|jd<dS)NTZ)environKeyErrortzset)rostimerrsetTZDs   zset_timezone..setTZr)rrrget)rrorig_tzrrr set_timezone+s r"Freturn_filelikeboolkwargsrc kstt}|dur d}tt|}||}|t|}|r1|ddt|fi|}z|VWt |ts?| | rI| dSdSt |tsT| | r]| ww)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(). Nmodezw+b) rtempfile gettempdirruuiduuid4touch setdefaultopen isinstancecloseis_fileunlink)filenamer#r%folderr handle_or_strrrr ensure_cleanVs.      r6 Iterator[str]ccs\tjdd}z|VWzt|WdStyYdSwzt|Wwty-Yww)z{ Get a temporary directory path and agrees to remove on close. Yields ------ Temporary directory path r&)suffixN)r(mkdtemprOSError)directory_namerrrensure_clean_dirs    r<c csHttj}zdVWtjtj|dStjtj|w)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) saved_environrrr!ensure_safe_environment_variabless   rAc ksXddl}hd}||vrtd|j|fi|z dVW||dS||w)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> excel-tabunixexcelz Cannot override builtin dialect.)csv ValueErrorregister_dialectunregister_dialect)namer%rE_BUILTIN_DIALECTSrrrwith_csv_dialectsrKc cshddlm}|dur|j}|j}|j}td|||_zdVW||_td|dS||_td|w)Nr) expressionszcompute.use_numexpr)pandas.core.computationrL _MIN_ELEMENTS USE_NUMEXPRr )use min_elementsexprolduseoldminrrr use_numexprs   rUc@s.eZdZdZd ddZd ddZd dd Zd S) 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() r NonecCs ||_dSN)seed)selfrYrrr__init__s zRNGContext.__init__cCstj|_tj|jdSrX)nprandom get_state start_staterY)rZrrr __enter__s zRNGContext.__enter__cCstj|jdSrX)r\r] set_stater_)rZexc_type exc_value tracebackrrr__exit__szRNGContext.__exit__N)r rW)__name__ __module__ __qualname____doc__r[r`rerrrrrVs   rV)r r )rrr r)NF)r#r$r%r)r r7)r rrX) __future__r contextlibrrpathlibrshutilrr(typingrrrr*numpyr\pandasr pandas.io.commonr rr"r6r<rArKrUrVrrrrs4       **   !