i ddlmZddlmZddlZddlZddlmZmZddl m Z ddl m Z ddl mZerddlmZdd lmZmZmZe dd Zedd Zedd Zdd Zy)) annotations)contextmanagerN)IO TYPE_CHECKING)CHAINED_WARNING_DISABLED)ChainedAssignmentError) get_handle) Generator) BaseBufferCompressionOptionsFilePathc#pKt|d|d5}|jdddy#1swYyxYww)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 d/mnt/e/genesis-system/scripts/.venv_scraper/lib/python3.12/site-packages/pandas/_testing/contexts.pydecompress_filers6& D$K G6mms6* 636c#Kddldfd }tjjd}|| d||y#||wxYww)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' rNctdrM| tjd=y|tjd<tj dk7rj yyy#t$rYywxYw)NtzsetTZwin32)hasattrosenvironKeyErrorsysplatformr)tztimes rsetTZzset_timezone..setTZHsf 4 !z 4($& 4 <<7*JJL+ " sA A('A(r)returnNone)r"rrget)r!r#orig_tzr"s @r set_timezoner(1sB* !jjnnT"G "I  ggs2AA A A  Ac+Kddl}hd}||vr td|j|fi| d|j|y#|j|wxYww)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)namekwargsr-_BUILTIN_DIALECTSs rwith_csv_dialectr4]s_(6   ;<<C((%  t$t$s+AAAAAcddlm}tr|s ddlm}|S|||St }d}|r|g|}|||g|S)Nr)assert_produces_warning) nullcontext)matchzRA value is being set on a copy of a DataFrame or Series through chained assignment)pandas._testingr6r contextlibr7r)extra_warnings extra_matchr6r7warningr8s rraises_chained_assignment_errorr>sg7 .= *!  ) )  00G& ';'  )rzFilePath | BaseBufferrr r$zGenerator[IO[bytes]])r!strr$Generator[None])r1r@r$rA)rB) __future__rr:rrrtypingrr pandas.compatr pandas.errorsrpandas.io.commonr collections.abcr pandas._typingr r r rr(r4r>rBr?rrJs"%  30') .@,((V%%B r?