U md@sddlmZddlmZddlZddlmZddlZddlm Z m Z m Z ddl Z ddl mZmZmZddlmZddlmZdd lmZdd lmZed d d dddZedddddZed&ddddddZeddddZeddd d!Zed'ddd"d#Zd$d%ZdS)() annotations)contextmanagerN)Path)IOAny Generator) BaseBufferCompressionOptionsFilePath)PYPY)ChainedAssignmentError) set_option) get_handlezFilePath | BaseBufferr z Generator[IO[bytes], None, None])path compressionreturnc cs(t|d|dd}|jVW5QRXdS)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)rhandle)rrrrQ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/_testing/contexts.pydecompress_filesrstrzGenerator[(None, None, None)])tzrc#sHddlddfdd }tjd}||z dVW5||XdS)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' rNNonercsB|dkr,z tjd=Wq>tk r(Yq>Xn|tjd<dS)NTZ)osenvironKeyErrortzset)rtimerrsetTZJs  zset_timezone..setTZr)r!rrget)rr"Zorig_tzrr r set_timezone2s  r$FboolrzGenerator[Any, None, None])return_filelikekwargsrc kstt}|dkrd}tt|}||}|t|}|r\|ddt|f|}z |VW5t |tsz| | r| XdS)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'folderrZ handle_or_strrrr ensure_clean\s      r7rc cs2ttj}z dVW5tjtj|XdS)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)dictrrclearupdate)Z saved_environrrr!ensure_safe_environment_variabless    r;c ksLddl}dddh}||kr"td|j|f|z dVW5||XdS)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. rNexcelz excel-tabunixz Cannot override builtin dialect.)csv ValueErrorregister_dialectunregister_dialect)namer'r>Z_BUILTIN_DIALECTSrrrwith_csv_dialects  rCc csXddlm}|dkr|j}|j}|j}td|||_z dVW5||_td|XdS)Nr) expressionszcompute.use_numexpr)Zpandas.core.computationrDZ _MIN_ELEMENTSZ USE_NUMEXPRr )ZuseZ min_elementsexprZolduseZoldminrrr use_numexprs   rFcCs2trddlm}|Sddlm}|tddSdS)Nr) nullcontext)assert_produces_warningzYA value is trying to be set on a copy of a DataFrame or Series through chained assignment)match)r contextlibrGZpandas._testingrHr )rGrHrrrraises_chained_assignment_errors  rK)NF)N) __future__rrJrrpathlibrr*typingrrrr,Zpandas._typingrr r Z pandas.compatr Z pandas.errorsr Zpandasr Zpandas.io.commonrrr$r7r;rCrFrKrrrrs4       ),!