d6ddlmZddlmZddlZddlmZddlZddlm Z m Z m Z ddl Z ddl mZmZmZddlmZddlmZdd lmZdd lmZed!dZed"dZe d#d$dZed%dZed%dZed&d%dZd ZdS)') annotations)contextmanagerN)Path)IOAny Generator) BaseBufferCompressionOptionsFilePath)PYPY)ChainedAssignmentError) set_option) get_handlepathFilePath | BaseBuffer compressionr return Generator[IO[bytes], None, None]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)ris_textN)rhandle)rrrs T/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/_testing/contexts.pydecompress_filers& D$K G G G6ms -11tzstrGenerator[None, None, None]c#Kddldfd }tjd}|| 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' rNrNonec|! tjd=dS#t$rYdSwxYw|tjd<dS)NTZ)osenvironKeyErrortzset)rtimes rsetTZzset_timezone..setTZJs^ : Jt$$$     "BJt  JJLLLLLs  ""r!)rr)r&r"r#get)rr'orig_tzr&s @r set_timezoner*2s,KKKjnnT""G E"III  ggs A AFreturn_filelikeboolkwargsrGenerator[Any, None, None]c+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_cleanr@\sD,(%'' ( (F4:<<  8+H H DJJLLL!$iiM-&%(((T,,V,, --- "    ! ! ! <<>>  KKMMMMM  --- "    ! ! ! <<>>  KKMMMM s C11AEc#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)dictr"r#clearupdate) 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`trddlm}|Sddlm}|t dS)Nr) nullcontext)assert_produces_warningzYA value is trying to be set on a copy of a DataFrame or Series through chained assignment)match)r contextlibr^pandas._testingr_r )r^r_s rraises_chained_assignment_errorrcs\  ******{}};;;;;;&& "-    )rrrr rr)rrrr)NF)r+r,r-rrr.)rr)N) __future__rrarr"pathlibrr2typingrrrr4pandas._typingr r r pandas.compatr pandas.errorsr pandasrpandas.io.commonrrr*r@rFrQr\rcrdrrns""""""%%%%%%    000000'''''',&&&&R+0)))))X ) ) ) )%%%%B22222"     rd