o d @sdZddlmZddlmZmZddlZddlmZddl Z ddl Z ddl Z ddl m Z mZmZmZmZmZddlZddlZddlmZddlZddlZddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z)m*Z*m+Z+m,Z,ddl-Z-ddl.Z.dd l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7dd l8m9Z9dd l:m;Z;dd lddl?m@Z@ddlAmBZBddlCmDZDmEZEmFZFmGZGddlHmIZIddlJmKZKeLe,e*e+ZMeMNdeOdZPe#de0dZQe jRGdddZSe jRGdddeeZTddd ZUe%dd#d$ZVe%dd%d$ZVdd'd$ZVdd*d+ZWe%ddd/d0ZXe% ,ddd1d0ZX 2ddd4d0ZXd5d6ZYdd8d9ZZe@eKd:eKd;d!d< =  > dddDdEZ[ddGdHZ\dIdIdIdIdJdKdLdMdNdO Z]eLe]^Z_ddQdRZ`e@eKd;d!dSddUdVZaddXdYZbe%d,d,d,d,d,dZddadbZce%d,d,d,d,d,d,dcddfdbZce%d,d,d,d,d,d,dcddhdbZce@eKd;d[dSddd2didddcddjdbZcGdkdldleeZdGdmdndnedZeGdodpdpedZfGdqdrdrZgGdsdtdtZhddxdyZiddzd{Zjdd|d}Zke jldddZm ddddZndddZodS)zCommon IO api utilities) annotations)ABCabstractmethodN) defaultdict)BufferedIOBaseBytesIO RawIOBaseStringIO TextIOBase TextIOWrapper)Path) IOAnyAnyStr DefaultDictGenericHashableLiteralMappingSequenceTypeVarcastoverload)urljoinurlparse uses_netloc uses_params uses_relative) BaseBufferCompressionDictCompressionOptionsFilePath ReadBuffer ReadCsvBufferStorageOptions WriteBuffer) get_lzma_file)import_optional_dependency)BZ2File)doc)find_stack_level)is_bool is_file_like is_integer is_list_like) MultiIndex) _shared_docsz^[A-Za-z][A-Za-z0-9+\-+.]*:// BaseBufferT)boundc@s>eZdZUdZded<ded<ded<ded<d Zd ed <d S) IOArgsz? Return value of io/common.py:_get_filepath_or_buffer. str | BaseBufferfilepath_or_bufferstrencodingmoder compressionFbool should_closeN)__name__ __module__ __qualname____doc____annotations__r<rBrBU/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/io/common.pyr4Ss r4c@s`eZdZUdZded<ded<ejedZded<d Z d ed <dddZ dddZ dddZ dS) IOHandlesau Return value of io/common.py:get_handle Can be used as a context manager. This is used to easily close created buffers and to handle corner cases when TextIOWrapper is inserted. handle: The file handle to be used. created_handles: All file handles that are created by get_handle is_wrapped: Whether a TextIOWrapper needs to be detached. z IO[AnyStr]handlerr:)default_factoryzlist[IO[bytes] | IO[str]]created_handlesFr; is_wrappedreturnNonecCs\|jrt|jts J|j|j|j|j|jD]}|qg|_d|_dS)z Close all created buffers. Note: If a TextIOWrapper was inserted, it is flushed and detached to avoid closing the potentially user-created buffer. FN) rH isinstancerEr flushdetachrGremoveclose)selfrErBrBrCrOus     zIOHandles.closeIOHandles[AnyStr]cCs|SNrBrPrBrBrC __enter__szIOHandles.__enter__argsrcGs |dSrR)rO)rPrUrBrBrC__exit__ zIOHandles.__exit__NrIrJ)rIrQ)rUrrIrJ) r=r>r?r@rA dataclassesfieldlistrGrHrOrTrVrBrBrBrCrD`s    rDurlobjectrIr;cCst|tsdSt|jtvS)z Check to see if a URL has a valid protocol. Parameters ---------- url : str or unicode Returns ------- isurl : bool If `url` has a valid protocol return True otherwise False. F)rKr7 parse_urlscheme _VALID_URLSr\rBrBrCis_urls rbr6r7cCdSrRrBr6rBrBrC _expand_userrecCrcrRrBrdrBrBrCrerfstr | BaseBufferTcCst|tr tj|S|S)a] Return the argument with an initial component of ~ or ~user replaced by that user's home directory. Parameters ---------- filepath_or_buffer : object to be converted if possible Returns ------- expanded_filepath_or_buffer : an expanded filepath or the input if not expandable )rKr7ospath expanduserrdrBrBrCres  headerrJcCs|durdSt|rtt|}|dkrtddSt|ddr>tt|}ttt|s/tdtdd|Drsz&validate_header_arg..z8cannot specify multi-index header with negative integerszPassing a bool to header is invalid. Use header=None for no header or header=int or list-like of ints to specify the row(s) making up the column names) r-rint ValueErrorr.rallmapanyr+ TypeError)rkrBrBrCvalidate_header_args*   rv.r!convert_file_likecCrcrRrBr6rwrBrBrCstringify_pathrfrycCrcrRrBrxrBrBrCrysFFilePath | BaseBufferTcCs2|s t|r tt|St|tjr|}t|S)a Attempt to convert a path-like object to a string. Parameters ---------- filepath_or_buffer : object to be converted Returns ------- str_filepath_or_buffer : maybe a string version of the object Notes ----- Objects supporting the fspath protocol (python 3.6+) are coerced according to its __fspath__ method. Any other object is passed through unchanged, which includes bytes, strings, buffers, or anything else that's not even path-like. )r,rr2rKrhPathLike __fspath__rerxrBrBrCrys   cOsddl}|jj|i|S)z` Lazy-import wrapper for stdlib urlopen, as that imports a big chunk of the stdlib. rN)urllib.requestrequesturlopen)rUkwargsurllibrBrBrCrsrFilePath | BaseBuffercCs$t|tott|o|d S)zR Returns true if the given URL looks like something fsspec can handle )zhttp://zhttps://)rKr7r;_RFC_3986_PATTERNmatch startswithrarBrBrC is_fsspec_urls   rstorage_optionscompression_options)rrutf-8rr8r:r r9r$c Cst|}t|\}}t||}|r%t|dr%d|vr%tjdttdd}t||d}d|vrE|dvrE|d vrEtj|d |t td|}d |vrSd|vrS|d7}t |t rt |r|p_i}d dl }|jj||d }t|} | jdd} | dkrddi}t| } Wdn1swYt| ||d|dSt|r,t |t sJ|dr|dd}|dr|dd}td} g} ztdd dlm}m}||tg} Wn tyYnwz| j|fd|i|pi}Wn.t | y"|durddi}nt|}d|d<| j|fd|i|pi}Ynwt|||d|dS|r3t!dt |t t"t#j#frHtt$|||d|dSt|ds_t|ds_dt%|}t!|t|||d|dS)a% If the filepath_or_buffer is a url, translate and return the buffer. Otherwise passthrough. Parameters ---------- filepath_or_buffer : a url, filepath (str, py.path.local or pathlib.Path), or buffer {compression_options} .. versionchanged:: 1.4.0 Zstandard support. encoding : the encoding to use to decode bytes, default is 'utf-8' mode : str, optional {storage_options} .. versionadded:: 1.2.0 ..versionchange:: 1.2.0 Returns the dataclass IOArgs. writebzDcompression has no effect when passing a non-binary object as input.) stacklevelN)methodw)bz2xz)zutf-16zutf-32z( will not write the byte order mark for tr)headerszContent-EncodinggziprT)r6r8r:r<r9zs3a://zs3://zs3n://fsspecbotocore) ClientErrorNoCredentialsErrorr9anonz?storage_options passed with file object or non-fsspec file pathFreadz)Invalid file path or buffer object type: )&ryget_compression_methodinfer_compressionhasattrwarningswarnRuntimeWarningr*dictUnicodeWarningrKr7rbr}r~Requestrrgetrrr4rrreplacer'botocore.exceptionsrrPermissionError ImportErroropentuplerqbytesmmapretype)r6r8r:r9rcompression_method fsspec_moderreq_inforeqcontent_encodingreaderrerr_types_to_retry_with_anonrrfile_objmsgrBrBrC_get_filepath_or_buffers"                rricCsddlm}td||S)z converts an absolute native path to a FILE URL. Parameters ---------- path : a path in native format Returns ------- a valid FILE URL r) pathname2urlzfile:)r}rr)rirrBrBrCfile_path_to_urls rtarrrziprzstd) .tar.tar.gz.tar.bz2.tar.xz.gz.bz2.zip.xzz.zst"tuple[str | None, CompressionDict]c CsZt|tr%t|}z |d}W||fSty$}ztd|d}~wwi}|}||fS)a Simplifies a compression argument to a compression method string and a mapping containing additional arguments. Parameters ---------- compression : str or mapping If string, specifies the compression method. If mapping, value at key 'method' specifies compression method. Returns ------- tuple of ({compression method}, Optional[str] {compression arguments}, Dict[str, Any]) Raises ------ ValueError on mapping missing 'method' key rz.If mapping, compression must have key 'method'N)rKrrpopKeyErrorrq)r:compression_argsrerrrBrBrCrs   r)r str | NonecCs|durdS|dkr-t|dd}t|tsdStD]\}}||r*|SqdS|tvr3|Sddgtt}d|d|}t |)a/ Get the compression method for filepath_or_buffer. If compression='infer', the inferred compression method is returned. Otherwise, the input compression method is returned unchanged, unless it's invalid, in which case an error is raised. Parameters ---------- filepath_or_buffer : str or file handle File path or object. {compression_options} .. versionchanged:: 1.4.0 Zstandard support. Returns ------- string or None Raises ------ ValueError on invalid compression specified. NinferT)rwUnrecognized compression type: z Valid compression types are ) ryrKr7extension_to_compressionitemslowerendswith_supported_compressionssortedrq)r6r: extensionvalidrrBrBrCrs&  r Path | strcCs&t|j}|std|ddS)z Check if parent directory of a file exists, raise OSError if it does not Parameters ---------- path: Path or str Path to check parent directory of z1Cannot save file into a non-existent directory: ''N)r parentis_dirOSError)rirrBrBrCcheck_parent_directoryMs r)r8r: memory_maperrorsr path_or_bufris_textLiteral[False]rIOHandles[bytes]cCrcrRrBrr9r8r:rrrrrBrBrC get_handle[ r)r8r:rrrr Literal[True]IOHandles[str]cCrcrRrBrrBrBrCrjr!IOHandles[str] | IOHandles[bytes]cCrcrRrBrrBrBrCryrTcCs|pd}|pd}t||rd|vr|d7}t|t|tr$t|t|||||d}|j} t| |\} }} t| t} t |j } | d}d|vrS| rSt t| |r|dkrc|j dd |_ n|dkrsd|j vrs|j d7_ |d krt| trtjd*| |j d | } n tjd*| |j d | } n|d krt| fd|j i| } n|dkrt| |j fi| } | jj dkr| | | j} t| dkr| j| } n| std|td| n|dkrN| d|j t| trtd*d| i| } n td*d| i| } t| tsJd| jj vrM| | | j}t|dkr<| j|d}|dus9J|} nY|sFtd|td|nG|dkr[t| |j } n:|dkrtd}d|j vrud|jd*i| i}n d|j d*i| i}|j| fd|j i|} n d|}t|t| trJ| | n't| tr|j!rd|j vrt| |j |j!|d d} nt| |j } | | d }|s|j d!krt| t"rt#| |j!d"} n?|r"|s|st| |j r"t$| d#rt$| d$rt$| d%s t%| } t&| |j!|d d} | | t|jtp |j' }d|j vr9t$| d&s9t(d't)|jd(| *|j'rPt|jtrJJ| |jt+| | ||j d)S)+a Get file handle for given path/buffer and mode. Parameters ---------- path_or_buf : str or file handle File path or object. mode : str Mode to open path_or_buf with. encoding : str or None Encoding to use. {compression_options} .. versionchanged:: 1.0.0 May now be a dict with key 'method' as compression mode and other keys as compression options if compression mode is 'zip'. .. versionchanged:: 1.1.0 Passing compression options as keys in dict is now supported for compression modes 'gzip', 'bz2', 'zstd' and 'zip'. .. versionchanged:: 1.4.0 Zstandard support. memory_map : bool, default False See parsers._parser_params for more information. Only used by read_csv. is_text : bool, default True Whether the type of the content passed to the file/buffer is string or bytes. This is not the same as `"b" not in mode`. If a string content is passed to a binary file/buffer, a wrapper is inserted. errors : str, default 'strict' Specifies how encoding and decoding errors are to be handled. See the errors argument for :func:`open` for a full list of options. storage_options: StorageOptions = None Passed to _get_filepath_or_buffer .. versionchanged:: 1.2.0 Returns the dataclass IOHandles rstrictr)r8r:r9rrrrrr1r)filenamer9)fileobjr9rr9rzZero files found in ZIP file z9Multiple files found in ZIP file. Only one file per ZIP: rnamerrNz Zero files found in TAR archive zDMultiple files found in TAR archive. Only one file per TAR archive: r zstandarddctxcctxr)r8rnewlineFrb)r8readablewritableseekablerz1Expected file path name or file-like object, got z type)rErGrHr:rB),_is_binary_modecodecslookuprKr7 lookup_errorrr6_maybe_memory_maprr:rrr9rrGzipFile_BZ2File _BytesZipFilebufferappendnamelistlenrrq setdefault _BytesTarFilegetnames extractfiler&r'ZstdDecompressorZstdCompressorr8r _BytesIOWrapperr _IOWrapperr r<rurreverserD)rr9r8r:rrrrioargsrEhandlesis_pathr zip_namesfilesfiler open_argsrrHrBrBrCrsL6                             cs0eZdZdZedddZdfdd ZZS) _BufferedWriterz Some objects do not support multiple .write() calls (TarFile and ZipFile). This wrapper writes to the underlying buffer on close. rIrJcCrcrRrBrSrBrBrCwrite_to_bufferrfz_BufferedWriter.write_to_buffercsd|jrdS|r&|d|j |Wdn1s wYn|jtdS)Nr)closedgetvalueseekrrrOsuperrS __class__rBrCrOs   z_BufferedWriter.closerX)r=r>r?r@rrrO __classcell__rBrBrrCrs  rcsFeZdZ    ddfd d ZdddZdddZdddZZS)rNrrrr9Literal['r', 'a', 'w', 'x']r-ReadBuffer[bytes] | WriteBuffer[bytes] | None archive_namerIrJc s<t||_||_tjjd||||d||_dS)N)rr9rrB) r__init__rrtarfileTarFiler extend_moder)rPrr9rrrrrBrCrs  z_BytesTarFile.__init__r7cCsP|dd}|dkr |S|jdur&t|jj}|dvr&|d|dd}|S)Nrr1r)rrr:r)rrr suffix)rPr9r!rBrBrCrs   z_BytesTarFile.extend_modecCsP|jdurdSt|j}|jdkr|djS|jdvr%|ddjS|jS)z If an explicit archive_name is not given, we still want the file inside the zip file not to be named something.tar, because that causes confusion (GH39465). Nrr1)rrr)rr r! with_suffixrPrrBrBrCinfer_filenames     z_BytesTarFile.infer_filenamecCs>|jp|pd}tj|d}t||_|j||dS)Nr)r) rr$rTarInforrsizeraddfile)rPrtarinforBrBrCrs z_BytesTarFile.write_to_buffer)NrNN) rrr9rrrrrrIrJ)r9r7rIr7rIrrX)r=r>r?rrr$rrrBrBrrCrs  rcs6eZdZ ddfd d Zdd d ZdddZZS)rNr1FilePath | ReadBuffer[bytes] | WriteBuffer[bytes]r9r7rrrIrJc sDt|dd}||_|dtjtj||fi||_dS)Nrr1r:) rrrrrzipfile ZIP_DEFLATEDZipFiler)rPrr9rrrrBrCrs  z_BytesZipFile.__init__cCs@t|jjtjtfrt|jj}|jdkr|dj S|j SdS)z If an explicit archive_name is not given, we still want the file inside the zip file not to be named something.zip, because that causes confusion (GH39465). rr1N) rKrrrhr{r7r r!r"rr#rBrBrCr$s    z_BytesZipFile.infer_filenamecCs(|jp|pd}|j||dS)Nr)rr$rwritestrr)rPrrBrBrCrsz_BytesZipFile.write_to_bufferrR)rr*r9r7rrrIrJr)rX)r=r>r?rr$rrrBrBrrCrs   rc@s>eZdZdddZdd d Zdd d ZdddZdddZdS)rrrrIrJcCs ||_dSrR)r)rPrrBrBrCrs z_IOWrapper.__init__rr7cC t|j|SrRgetattrr)rPrrBrBrC __getattr__rWz_IOWrapper.__getattr__r;cCt|jdr |jSdS)NrT)rrrrSrBrBrCr"  z_IOWrapper.readablecCr3)NrT)rrrrSrBrBrCr'r4z_IOWrapper.seekablecCr3)NrT)rrrrSrBrBrCr,r4z_IOWrapper.writableN)rrrIrJ)rr7)rIr;)r=r>r?rr2rrrrBrBrBrCrs     rc@s.eZdZdddd Zdd d ZddddZdS)rrrStringIO | TextIOBaser8r7rIrJcCs||_||_d|_dS)N)rr8overflow)rPrr8rBrBrCr5s z_BytesIOWrapper.__init__attrcCr/rRr0)rPr8rBrBrCr2>rWz_BytesIOWrapper.__getattr__n int | NonercCsp|jdusJ|j||j}|j|}|dus$|dks$|t|kr)d|_|S|d|}||d|_|S)Nrr6)rrencoder8r7r)rPr: bytestringcombined_bytestring to_returnrBrBrCrAs  z_BytesIOWrapper.readN)r)rr5r8r7rIrJ)r8r7)r9)r:r;rIr)r=r>r?rr2rrBrBrBrCr2s  rrEr5/tuple[str | BaseBuffer, bool, list[BaseBuffer]]c Csg}|t|dp t|tM}|s|||fStt|}t|tr)t|d}||zttj| dtj d}Wt |D]}| q.)rKr;rr/rr)rRrTrBrWrCis_potential_multi_indexs rXnamesSequence[Hashable]is_potential_multiindexcCst|}tt}t|D]E\}}||}|dkrG|d||<|r8t|ts'J|dd|dd|f}n|d|}||}|dks|||<|d||<q |S)a: Rename column names if duplicates exist. Currently the renaming is done by appending a period and an autonumeric, but a custom pattern may be supported in the future. Examples -------- >>> dedup_names(["x", "y", "x", "x"], is_potential_multiindex=False) ['x', 'y', 'x.1', 'x.2'] rrNr9.)r[rrp enumeraterKr)rYr[countsrncol cur_countrBrBrC dedup_namess " ra)r\r]rIr;)r6r7rIr7)r6r2rIr2)r6rgrIrg)rkr]rIrJ).)r6r!rwr;rIr7)r6r2rwr;rIr2)F)r6rzrwr;rIrg)r\rrIr;)rNrN) r6rr8r7r:r r9r7rr$rIr4)rir7rIr7)r:r rIr)r6rr:rrIr)rirrIrJ)rrr9r7r8rr:r rr;rrrrrr$rIr)rrr9r7r8rr:r rr;rrrrrr$rIr)rrr9r7r8rr:r rr;rr;rrrr$rIr)rEr5rr;rIr@)r6rrIr;)rErr9r7rIr;)rIrNrR)rRrSrTrUrIr;)rYrZr[r;rIrZ)pr@ __future__rabcrrr collectionsrrY functoolsriorrrr r r rrhpathlibr rertypingr rrrrrrrrrrr urllib.parserrr^rrrrr+pandas._typingrrr r!r"r#r$r% pandas.compatr&pandas.compat._optionalr'pandas.compat.compressorsr(rpandas.util._decoratorsr)pandas.util._exceptionsr*pandas.core.dtypes.commonr+r,r-r.pandas.core.indexes.apir/pandas.core.shared_docsr0setr`discardcompilerr2 dataclassr4rDrbrervryrrrrrvaluesrrrrrrrrrrrrGr lru_cacherLrXrarBrBrBrCs    8(           ,     "    3 #  67$  "