a Of@s2ddlmZddlmZddlZddlmZddlZddlm Z m Z m Z ddl Z ddl mZddlmZddlmZdd lmZdd lmZe rdd lmZdd lmZmZmZed dddddZedddddZed+ddddddZedddd d!Z ed,dd"d#d$Z!d-d'd(Z"d.d)d*Z#dS)/) annotations)contextmanagerN)Path)IO TYPE_CHECKINGAny)using_copy_on_write)PYPY)ChainedAssignmentError) set_option) get_handle) Generator) BaseBufferCompressionOptionsFilePathzFilePath | BaseBufferrz Generator[IO[bytes], None, None])path compressionreturnccs<t|d|dd}|jVWdn1s.0YdS)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)rZis_textN)r handle)rrrrQ/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/_testing/contexts.pydecompress_file!srstrzGenerator[None, None, None])tzrc#sPddlddfdd }tjd}||zdVW||n ||0dS)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' rNNonercs@|dur*z tjd=Wq<ty&Yq<0n|tjd<dS)NTZ)osenvironKeyErrortzset)rtimerrsetTZPs   zset_timezone..setTZr)r#rrget)rr$Zorig_tzrr"r set_timezone8s r&FboolrzGenerator[Any, None, None])return_filelikekwargsrc kstt}|durd}tt|}||}|t|}|dd}|r|dd|durtd|dvrtd}t |fd|i|}z,|VWt |ts| | r| n$t |ts| | r| 0dS)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(). Nencodingmodezw+bbzutf-8)rtempfile gettempdirruuiduuid4touchpop setdefaultopen isinstancecloseis_fileunlink)filenamer(r)folderrZ handle_or_strr+rrr ensure_cleanbs.     r<)namerc ksXddl}hd}||vr td|j|fi|zdVW||n ||0dS)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>unixexcelz excel-tabz Cannot override builtin dialect.)csv ValueErrorregister_dialectunregister_dialect)r=r)r@Z_BUILTIN_DIALECTSrrrwith_csv_dialectsrDrc cshddlm}|dur|j}|j}|j}td|||_zdVW||_td|n||_td|0dS)Nr) expressionszcompute.use_numexpr)Zpandas.core.computationrEZ _MIN_ELEMENTSZ USE_NUMEXPRr )ZuseZ min_elementsexprZolduseZoldminrrr use_numexprs   rGTrcCsddlm}|s"ddlm}|Str<|sddlm}|s"ddlm}|S|s*d}|tfd|i|S)aq Assert that a warning is raised in the CoW warning mode. Parameters ---------- warn : bool, default True By default, check that a warning is raised. Can be turned off by passing False. match : str The warning message to match against, if different from the default. kwargs Passed through to assert_produces_warning rrHrJzSetting a value on a viewrM)rNrIrOrKrQ)rRrMr)rIrKrrrassert_cow_warnings  rU)NF)N)Trr)TN)$ __future__rrOrrpathlibrr.typingrrrr0Zpandas._configrZ pandas.compatr Z pandas.errorsr Zpandasr Zpandas.io.commonr collections.abcr Zpandas._typingrrrrr&r<rDrGrTrUrrrrs6         )/! $