WhHddlmZddlmZddlZddlmZddlmZddl Z ddl m Z m Z m Z ddlZddlZddlmZddlmZedd ZeddZedd dZed!dZed"dZed"dZed#d"dZGddZdS)$) annotations)contextmanagerN)Path)rmtree)IOAnyIterator) set_option) get_handlereturnIterator[IO[bytes]]c#tKt|d|d5}|jVddddS#1swxYwYdS)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 ^/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pandas/_testing/contexts.pydecompress_filers" D$K G G G6ms -11tzstrIterator[None]c#Kddlddlfd}jd}|| dV||dS#||wxYw)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=dS#t$rYdSwxYw|jd<dS)NTZ)environKeyErrortzset)rostimes rsetTZzset_timezone..setTZDs^ : Jt$$$     "BJt  JJLLLLLs  r)rr rget)rr!orig_tzrr s @@r set_timezoner$+s,IIIKKKjnnT""G E"III  ggs A AFreturn_filelikeboolkwargsrc+Kttj}|d}tt j|z}||z }|t|}|r#|ddt|fi|} |Vt|ts| | r| dSdS#t|ts| | r| wwxYw)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_strs r ensure_cleanr9VsD((%'' ( (F4:<<  8+H H DJJLLL!$iiM-&%(((T,,V,, --- "    ! ! ! <<>>  KKMMMMM  --- "    ! ! ! <<>>  KKMMMM s C11AE Iterator[str]c#Ktjd} |V t|dS#t$rYdSwxYw# t|w#t$rYwwxYwxYw)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%R000N  > " " " " "    DD   > " " " "    D s;A/ ==A"AA" AA"AA"c#DKttj} dVtjtj|dS#tjtj|wxYw)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_variablesrFs$$M)   -(((((  -((((s A ?Bc+Kddl}hd}||vrtd|j|fi| dV||dS#||wxYw)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'rK_BUILTIN_DIALECTSs rwith_csv_dialectrQs(JJJ666    ;<<<C(((((%  t$$$$$t$$$$s AAc#Kddlm}||j}|j}|j}t d|||_ dV||_t d|dS#||_t d|wxYw)Nr) expressionszcompute.use_numexpr)pandas.core.computationrS _MIN_ELEMENTS USE_NUMEXPRr )use min_elementsexprolduseoldmins r use_numexprr\s;;;;;;)  F  F$c***%D2 #(&11111$(&1111s AA.c*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() r Nonec||_dSN)seed)selfrbs r__init__zRNGContext.__init__s  ctj|_tj|jdSra)nprandom get_state start_staterb)rcs r __enter__zRNGContext.__enter__s49..00 ty!!!!!recNtj|jdSra)rgrh set_staterj)rcexc_type exc_value tracebacks r__exit__zRNGContext.__exit__s! D,-----reN)r r_)__name__ __module__ __qualname____doc__rdrkrqrerr^r^sZ  """" ......rer^)r r )rrr r)NF)r%r&r'r)r r:)r rra) __future__r contextlibrrpathlibrshutilrr+typingrrr r-numpyrgpandasr pandas.io.commonr rr$r9r@rFrQr\r^rvrerrs""""""%%%%%%   ''''''(''''T'''''T$ ) ) ) )%%%%B22222"..........re