`VgddlmZddlmZddlZddlmZddlmZddl Z ddl m Z m Z m Z ddlZddlZddlmZddlmZedd Zedd Zeddd Zedd Zedd ZeddZedddZGddZy)) annotations)contextmanagerN)Path)rmtree)IOAnyIterator) set_option) get_handlec#pKt|d|d5}|jdddy#1swYyxYww)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)pathrrs Y/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/pandas/_testing/contexts.pydecompress_filers6" D$K G6mms6* 636c#Kddlddlfd}jjd}|| d||y#||wxYww)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' rNc| jd=y|jd<jy#t$rYywxYw)NTZ)environKeyErrortzset)tzostimes rsetTZzset_timezone..setTZDsF : JJt$ "BJJt  JJL   s 3 ??r)rrrget)rrorig_tzrrs @@r set_timezoner +sE,jjnnT"G "I  ggs2AA A AAc+&Kttj}|d}tt j |z}||z }|j t|}|r|jddt|fi|} |t|ts|j|jr|jyy#t|ts|j|jr|jwwxYww)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 gettempdirstruuiduuid4touch setdefaultopen isinstancecloseis_fileunlink)filenamereturn_filelikekwargsfolderr handle_or_strs r ensure_cleanr5Vs((%%' (F4::< 8+H H DJJL!$iM&%(T,V, --    ! <<> KKM --    ! <<> KKM s BDC AD ADDc#Ktjd} | t|y#t$rYywxYw# t|w#t$rYwwxYwxYww)z{ Get a temporary directory path and agrees to remove on close. Yields ------ Temporary directory path r")suffixN)r$mkdtemprOSError)directory_names rensure_clean_dirr;s`%%R0N  > "     > "   sRA: +A 7A7AA AA AAAAAc#FKttj} dtjjtjj |y#tjjtjj |wxYww)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_environs r!ensure_safe_environment_variablesrAse$M)   -(  -(sB!A>B!?BB!c+Kddl}hd}||vr td|j|fi| d|j|y#|j|wxYww)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)namer2rF_BUILTIN_DIALECTSs rwith_csv_dialectrLs_(6   ;<<C((%  t$t$s+AAAAAc#Kddlm}| |j}|j}|j}t d|||_ d||_t d|y#||_t d|wxYww)Nr) expressionszcompute.use_numexpr)pandas.core.computationrN _MIN_ELEMENTS USE_NUMEXPRr )use min_elementsexprolduseoldmins r use_numexprrWst;))   F   F$c*%D2 #(&1$(&1sAA3AA3A00A3c(eZdZdZddZddZddZy) 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() c||_yN)seed)selfr\s r__init__zRNGContext.__init__s  ctjj|_tjj |jyr[)nprandom get_state start_stater\)r]s r __enter__zRNGContext.__enter__s+99..0 tyy!r_cVtjj|jyr[)rarb set_staterd)r]exc_type exc_value tracebacks r__exit__zRNGContext.__exit__s D,,-r_N)returnNone)__name__ __module__ __qualname____doc__r^rerkr_rrYrYs " .r_rY)rlzIterator[IO[bytes]])rr&rlIterator[None])NF)r1boolr2r)rlz Iterator[str])rlrsr[) __future__r contextlibrrpathlibrshutilrr$typingrrr r'numpyrapandasr pandas.io.commonr rr r5r;rArLrWrYrrr_rr}s"%   '(''T''T$ ) )%%B22"..r_