o ۭjQ@sddlmZddlmZddlZddlmZddlZddlm Z m Z m Z ddl Z ddl mZddlmZmZddlmZdd lmZdd lmZe rZdd lmZdd lmZmZmZed-ddZed.ddZe d/d0d d!Z ed1d#d$Z!ed2d3d%d&Z"d4d)d*Z#d5d+d,Z$dS)6) annotations)contextmanagerN)Path)IO TYPE_CHECKINGAny)using_copy_on_write)PYPYWARNING_CHECK_DISABLED)ChainedAssignmentError) set_option) get_handle) Generator) BaseBufferCompressionOptionsFilePathpathFilePath | BaseBuffer compressionrreturn Generator[IO[bytes], None, None]ccs@t|d|dd }|jVWddS1swYdS)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)ris_textN)r handle)rrrrU/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/_testing/contexts.pydecompress_file$s "rtzstrGenerator[None, None, None]c#sLddldfdd }tjd}||z dVW||dS||w) 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' rNrNonecsPtdr&|durztjd=WdStyYdSw|tjd<dSdS)NtzsetTZ)hasattrosenvironKeyErrorr!)rtimerrsetTZSs    zset_timezone..setTZr")rr )r(r$r%get)rr)orig_tzrr'r set_timezone;s r,Freturn_filelikeboolkwargsrGenerator[Any, None, None]c kstt}|dur d}tt|}||}|t|}|dd}|rE|dd|dur;d|dvr;d}t |fd|i|}z|VWt |tsS| | r]| dSdSt |tsh| | rq| ww)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/folderr handle_or_strr2rrr ensure_cleanfs4       rDnamec ksXddl}hd}||vrtd|j|fi|z dVW||dS||w)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)rEr/rI_BUILTIN_DIALECTSrrrwith_csv_dialectsrNc cshddlm}|dur|j}|j}|j}td|||_zdVW||_td|dS||_td|w)Nr) expressionszcompute.use_numexpr)pandas.core.computationrO _MIN_ELEMENTS USE_NUMEXPRr )use min_elementsexprolduseoldminrrr use_numexprs   rXTrcCsddlm}|sddlm}|Ststr |s ddlm}|Sts$tr/|r/||d|dStr7t}d}nt }d}|rC|g|R}||d|g|RdS)Nrassert_produces_warning nullcontext|)matchzYA value is trying to be set on a copy of a DataFrame or Series through chained assignmentr ) pandas._testingrZ contextlibr\r r joinrr FutureWarning)warnextra_warnings extra_matchrZr\warningr^rrrraises_chained_assignment_errors0      rgcKsBddlm}|r trddlm}|S|sd}|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 rrYr[zSetting a value on a viewr^)r_rZr r`r\rb)rcr^r/rZr\rrrassert_cow_warnings  rh)rrrrrr)rrrr)NF)r-r.r/rrr0)rErrr)N)rr)Trr)TN)% __future__rr`rr$pathlibrr5typingrrrr7pandas._configr pandas.compatr r pandas.errorsr pandasr pandas.io.commonr collections.abcrpandas._typingrrrrr,rDrNrXrgrhrrrrs8         */ ! $