U sVc@sddlmZddlmZddlZddlmZddlmZddl Z ddl m Z m Z m Z ddlZddlZddlmZddlmZed d d d Zed ddddZed#dddddZedd ddZedd ddZedd ddZed$dd dd ZGd!d"d"ZdS)%) annotations)contextmanagerN)Path)rmtree)IOAnyIterator) set_option) get_handlezIterator[IO[bytes]]returnc cs(t|d|dd}|jVW5QRXdS)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)pathrrr>> 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|dkr,z jd=Wq>tk r(Yq>Xn|jd<dS)NTZ)environKeyErrortzset)rostimerrsetTZDs  zset_timezone..setTZr)rrrget)rrZorig_tzrrr set_timezone+s  rFboolr)return_filelikekwargsc kstt}|dkrd}tt|}||}|t|}|r\|ddt|f|}z |VW5t |tsz| | r| XdS)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 gettempdirruuidZuuid4touch setdefaultopen isinstancecloseis_fileunlink)filenamer!r"folderrZ handle_or_strrrr ensure_cleanVs      r1z Iterator[str]ccs@tjdd}z |VW5z t|Wntk r8YnXXdS)z{ Get a temporary directory path and agrees to remove on close. Yields ------ Temporary directory path r#)suffixN)r%mkdtemprOSError)Zdirectory_namerrrensure_clean_dirs   r5c cs2ttj}z dVW5tjtj|XdS)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    r9c ksLddl}dddh}||kr"td|j|f|z dVW5||XdS)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. rNexcelz excel-tabunixz Cannot override builtin dialect.)csv ValueErrorregister_dialectunregister_dialect)namer"r<Z_BUILTIN_DIALECTSrrrwith_csv_dialects  rAc csXddlm}|dkr|j}|j}|j}td|||_z dVW5||_td|XdS)Nr) expressionszcompute.use_numexpr)Zpandas.core.computationrBZ _MIN_ELEMENTSZ USE_NUMEXPRr )ZuseZ min_elementsexprZolduseZoldminrrr use_numexprs   rDc@s:eZdZdZddddZddddZdddd 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() Noner cCs ||_dSN)seed)selfrHrrr__init__szRNGContext.__init__cCstj|_tj|jdSrG)nprandomZ get_state start_staterH)rIrrr __enter__s zRNGContext.__enter__cCstj|jdSrG)rKrLZ set_staterM)rIexc_type exc_value tracebackrrr__exit__szRNGContext.__exit__N)__name__ __module__ __qualname____doc__rJrNrRrrrrrEsrE)NF)N) __future__r contextlibrrpathlibrshutilrr%typingrrrr'ZnumpyrKZpandasr Zpandas.io.commonr rrr1r5r9rArDrErrrrs2      **!