U f/e@sddlmZddlmZddlZddlmZddlZddlm Z ddl Z ddl Z ddl m Z mZddlZddlmZedd Zed d d d ZeddddddZeddZeddZeddZedddZGdddZdS) ) annotations)contextmanagerN)Path)rmtree)IOAny) get_handlec 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', None} Name of the decompression to use Returns ------- file object rbF) compressionZis_textN)rhandle)pathr r r >> from datetime import datetime >>> from dateutil.tz import tzlocal >>> tzlocal().tzname(datetime.now()) 'IST' >>> with set_timezone('US/Eastern'): ... tzlocal().tzname(datetime.now()) ... 'EDT' rNcsB|dkr,z jd=Wq>tk r(Yq>Xn|jd<dS)NTZ)environKeyErrortzsetrostimer rsetTZBs  zset_timezone..setTZr)rrrget)rrZorig_tzr rr set_timezone)s  rFboolr)return_filelikekwargsc kstt}|dkrd}dtjtjtjdd|}||}| t |}|rl| ddt |f|}z |VW5t |t s||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(). N)kmodezw+b)rtempfile gettempdirjoinrandomchoicesstring ascii_lettersdigitstouchr setdefaultopen isinstancecloseis_fileunlink)filenamerrfolderr Z handle_or_strr r r ensure_cleanTs"     r5ccs@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_namer r rensure_clean_dirs   r9c 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_environr r r!ensure_safe_environment_variabless    r=cksDddl}dddh}||kr"td|j|f|dV||dS)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)namerr@Z_BUILTIN_DIALECTSr r rwith_csv_dialects rEccsPddlm}|dkr|j}|j}|j}||||_dV||_||dS)Nr) expressions)Zpandas.core.computationrFZ _MIN_ELEMENTSZ USE_NUMEXPRZset_use_numexpr)ZuseZ min_elementsexprZolduseZoldminr r r use_numexprs  rHc@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)selfrKr r r__init__szRNGContext.__init__cCstj|_tj|jdSrJ)npr'Z get_state start_staterK)rLr r r __enter__s zRNGContext.__enter__cCstj|jdSrJ)rNr'Z set_staterO)rLexc_type exc_value tracebackr r r__exit__szRNGContext.__exit__N)__name__ __module__ __qualname____doc__rMrPrTr r r rrIsrI)NF)N) __future__r contextlibrrpathlibrr'shutilrr)r$typingrrZnumpyrNZpandas.io.commonrrrr5r9r=rErHrIr r r rs2      *,