o Eh@sZUddlmZddlmZmZmZmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZmZerHddlmZeeZe de egefdZiZd ed <dKddZdLdMddZdNddZdOddZdPd!d"Ze dQd%d&Z e dRd(d&Z e dSd)d&Z e dTd*d&Z dUd-d&Z e dVd1d2Z!e dWd4d2Z!dXd7d2Z!dYd=d>Z"dZdBdCZ#d[dHdIZ$dJS)\) annotations)HashableIterableMutableMappingSequence) TYPE_CHECKINGAnyCallableLiteralTypeVaroverloadimport_optional_dependency) is_integer is_list_like) ExcelWriter usecols_func)Zboundz"MutableMapping[str, ExcelWriter_t]_writersklass ExcelWriter_treturnNonecCs"t|std|j}|t|<dS)z Add engine to the excel writer registry.io.excel. You must use this method to integrate with ``to_excel``. Parameters ---------- klass : ExcelWriter z&Can only register callables as enginesN)callable ValueErrorZ_enginer)r engine_namer5lib/python3.10/site-packages/pandas/io/excel/_util.pyregister_writer"s  rreaderextstrmodeLiteral['reader', 'writer']cCsZdddddd}ddddd}|dvsJ|dkr)td d d }|r%d |d <||S||S) a Return the default reader/writer for the given extension. Parameters ---------- ext : str The excel file extension for which to get the default engine. mode : str {'reader', 'writer'} Whether to get the default engine for reading or writing. Either 'reader' or 'writer' Returns ------- str The default engine for the extension. ZopenpyxlZpyxlsbZxlrdZodf)xlsxxlsmxlsbZxlsods)r#r$r%r&)rwriterr' xlsxwriterwarn)errorsr#r )rr!Z_default_readersZ_default_writersr(rrrget_default_engine2s$  r+rc Cs6zt|WSty}z td|d|d}~ww)NzNo Excel writer '')rKeyErrorr)rerrrrr get_writer[s  r/xintcCsdd}|D]%}t|}|tdks|tdkr!td||d|tdd}q|dS)ai Convert Excel column name like 'AB' to 0-based column index. Parameters ---------- x : str The Excel column name to convert to a 0-based column index. Returns ------- num : int The column index corresponding to the name. Raises ------ ValueError Part of the Excel column name was invalid. rAZzInvalid column name: )upperstripordr)r0indexcZcprrr _excel2numbsr;areas list[int]cCs^g}|dD]%}d|vr%|d}|tt|dt|ddq|t|q|S)a Convert comma separated list of column names and ranges to indices. Parameters ---------- areas : str A string containing a sequence of column ranges (or areas). Returns ------- cols : list A list of 0-based column indices. Examples -------- >>> _range2cols('A:E') [0, 1, 2, 3, 4] >>> _range2cols('A,C,Z:AB') [0, 2, 25, 26, 27] ,:rr5)splitextendranger;append)r<ZcolsrngZrngsrrr _range2colss &rEusecolsstr | list[int]cCdSNrrFrrrmaybe_convert_usecolsrK list[str]cCrHrIrrJrrrrKrLcCrHrIrrJrrrrKrLcCrHrIrrJrrrrKrL1str | list[int] | list[str] | usecols_func | None+None | list[int] | list[str] | usecols_funccCs2|dur|St|rtdt|trt|S|S)a Convert `usecols` into a compatible format for parsing in `parsers.py`. Parameters ---------- usecols : object The use-columns object to potentially convert. Returns ------- converted : object The compatible format of `usecols`. Nz}Passing an integer for `usecols` is no longer supported. Please pass in a list of int from 0 to `usecols` inclusive instead.)rr isinstancer rErJrrrrKs  freeze_panestuple[int, int] Literal[True]cCrHrIrrQrrrvalidate_freeze_panesrLrULiteral[False]cCrHrIrrTrrrrUrLtuple[int, int] | NoneboolcCs6|durt|dkrtdd|DrdStddS)Ncss|]}t|tVqdSrI)rPr1).0itemrrr s  z(validate_freeze_panes..TzLfreeze_panes must be of form (row, column) where row and column are integersF)lenallrrTrrrrUsrowlist[Hashable] control_row list[bool]!tuple[list[Hashable], list[bool]]cCsf|d}tdt|D]#}||s||}||dks!||dur&|||<q d||<||}q ||fS)a Forward fill blank entries in row but only inside the same parent index. Used for creating headers in Multiindex. Parameters ---------- row : list List of items in a single row. control_row : list of bool Helps to determine if particular column is in same parent index as the previous value. Used to stop propagation of empty cells between different indexes. Returns ------- Returns changed row and control_row rr5NF)rBr])r_raZlastirrrfill_mi_headers  rf index_colint | Sequence[int]&tuple[Hashable | None, list[Hashable]]cCspt|rt|ts Jt|}n t|trJ|}||}|dkr#dn|}||d|dg||ddfS)a Pop the header name for MultiIndex parsing. Parameters ---------- row : list The data row to parse for the header name. index_col : int, list The index columns for our data. Assumed to be non-null. Returns ------- header_name : str The extracted header name. trimmed_row : list The original data row with the header name removed. rdNr5)rrPrmax)r_rgreZ header_namerrrpop_header_names &rk engine_kwargsdict[str, Any] | NonekwargsdictcCs$|duri}n|}|||S)a Used to combine two sources of kwargs for the backend engine. Use of kwargs is deprecated, this function is solely for use in 1.3 and should be removed in 1.4/2.0. Also _base.ExcelWriter.__new__ ensures either engine_kwargs or kwargs must be None or empty respectively. Parameters ---------- engine_kwargs: dict kwargs to be passed through to the engine. kwargs: dict kwargs to be psased through to the engine (deprecated) Returns ------- engine_kwargs combined with kwargs N)copyupdate)rlrnresultrrrcombine_kwargs6s  rsN)rrrr)r)rr r!r"rr )rr rr)r0r rr1)r<r rr=)rFrGrr=)rFrMrrM)rFrrr)rFrrr)rFrNrrO)rQrRrrS)rQrrrV)rQrWrrX)r_r`rarbrrc)r_r`rgrhrri)rlrmrnrorro)%Z __future__rZcollections.abcrrrrtypingrrr r r r Zpandas.compat._optionalrZpandas.core.dtypes.commonrrZpandas.io.excel._basertyperobjectrr__annotations__rr+r/r;rErKrUrfrkrsrrrrs@       )  !          #"