o d@sddlmZddlmZddlZddlmZddlZddlm Z m Z m Z ddl Z ddl mZmZmZddlmZddlmZdd lmZdd lmZed(ddZed)ddZe d*d+ddZed,d d!Zed,d"d#Zed-d,d$d%Zd&d'ZdS).) annotations)contextmanagerN)Path)IOAny Generator) BaseBufferCompressionOptionsFilePath)PYPY)ChainedAssignmentError) set_option) get_handlepathFilePath | BaseBuffer compressionr return 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)rhandle)rrrr]/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/_testing/contexts.pydecompress_files "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' rNrNonecsB|durztjd=WdStyYdSw|tjd<dS)NTZ)osenvironKeyErrortzset)rtimerrsetTZJs   zset_timezone..setTZr)rr)r$rr get)rr%orig_tzrr#r set_timezone2s r(Freturn_filelikeboolkwargsrGenerator[Any, None, None]c kstt}|dur d}tt|}||}|t|}|r1|ddt|fi|}z|VWt |ts?| | rI| dSdSt |tsT| | r]| 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(). Nmodezw+b) rtempfile gettempdirruuiduuid4touch setdefaultopen isinstancecloseis_fileunlink)filenamer)r+folderr handle_or_strrrr ensure_clean\s.      r=c csHttj}zdVWtjtj|dStjtj|w)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)dictrr clearupdate) saved_environrrr!ensure_safe_environment_variabless   rBc 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)namer+rF_BUILTIN_DIALECTSrrrwith_csv_dialectsrLc cshddlm}|dur|j}|j}|j}td|||_zdVW||_td|dS||_td|w)Nr) expressionszcompute.use_numexpr)pandas.core.computationrM _MIN_ELEMENTS USE_NUMEXPRr )use min_elementsexprolduseoldminrrr use_numexprs   rVcCs.tr ddlm}|Sddlm}|tddS)Nr) nullcontext)assert_produces_warningzYA value is trying to be set on a copy of a DataFrame or Series through chained assignment)match)r contextlibrWpandas._testingrXr )rWrXrrrraises_chained_assignment_errors  r\)rrrr rr)rrrr)NF)r)r*r+rrr,)rr)N) __future__rrZrrpathlibrr/typingrrrr1pandas._typingrr r pandas.compatr pandas.errorsr pandasr pandas.io.commonrrr(r=rBrLrVr\rrrrs4        ),  !