a
    é´«j©  ã                   @  sÂ  U d dl mZ d dlmZmZmZmZ d dlmZm	Z	m
Z
mZmZmZ d dlmZ d dlmZmZ er�d dlmZ ee Zede
egef d�Zi Zd	ed
< dddœdd„ZdGddddœdd„Zdddœdd„Zdddœdd„Zdddœdd „Zed!dd"œd#d$„ƒZ ed%d%d"œd&d$„ƒZ eddd"œd'd$„ƒZ eddd"œd(d$„ƒZ d)d*d"œd+d$„Z ed,d-d.œd/d0„ƒZ!edd1d.œd2d0„ƒZ!d3d4d.œd5d0„Z!d6d7d8d9œd:d;„Z"d6d<d=d>œd?d@„Z#dAdBdBdCœdDdE„Z$dFS )Hé    )Úannotations)ÚHashableÚIterableÚMutableMappingÚSequence)ÚTYPE_CHECKINGÚAnyÚCallableÚLiteralÚTypeVarÚoverload©Úimport_optional_dependency)Ú
is_integerÚis_list_like)ÚExcelWriterÚusecols_func)Úboundz"MutableMapping[str, ExcelWriter_t]Ú_writersÚExcelWriter_tÚNone)ÚklassÚreturnc                 C  s"   t | ƒstdƒ‚| j}| t|< dS )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_name© r   úT/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/io/excel/_util.pyÚregister_writer"   s    
r   ÚreaderÚstrzLiteral['reader', 'writer'])ÚextÚmoder   c                 C  s^   ddddddœ}dddddœ}|dv s*J ‚|dkrRt d	d
d�}|rJd	|d< ||  S ||  S d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.
    ZopenpyxlZpyxlsbZxlrdZodf)ÚxlsxÚxlsmÚxlsbZxlsÚods)r#   r$   r%   r&   )r   Úwriterr'   Ú
xlsxwriterÚwarn)Úerrorsr#   Nr   )r!   r"   Z_default_readersZ_default_writersr(   r   r   r   Úget_default_engine2   s$    ûür+   )r   r   c              
   C  sF   z
t |  W S  ty@ } ztd| › d�ƒ|‚W Y d }~n
d }~0 0 d S )NzNo Excel writer 'ú')r   ÚKeyErrorr   )r   Úerrr   r   r   Ú
get_writer[   s    
r/   Úint)Úxr   c                 C  sd   d}|   ¡  ¡ D ]J}t|ƒ}|tdƒk s4|tdƒkrBtd| › �ƒ‚|d | tdƒ d }q|d S )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.
    r   ÚAÚZzInvalid column name: é   é   )ÚupperÚstripÚordr   )r1   ÚindexÚcÚcpr   r   r   Ú
_excel2numb   s    r<   z	list[int])Úareasr   c                 C  s^   g }|   d¡D ]J}d|v rJ|  d¡}| tt|d ƒt|d ƒd ƒ¡ q| 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]
    ú,ú:r   r5   )ÚsplitÚextendÚranger<   Úappend)r=   ÚcolsÚrngZrngsr   r   r   Ú_range2cols‚   s    
&rF   zstr | list[int])Úusecolsr   c                 C  s   d S ©Nr   ©rG   r   r   r   Úmaybe_convert_usecols£   s    rJ   z	list[str]c                 C  s   d S rH   r   rI   r   r   r   rJ   ¨   s    c                 C  s   d S rH   r   rI   r   r   r   rJ   ­   s    c                 C  s   d S rH   r   rI   r   r   r   rJ   ²   s    z1str | list[int] | list[str] | usecols_func | Nonez+None | list[int] | list[str] | usecols_funcc                 C  s2   | du r| S t | ƒrtdƒ‚t| tƒr.t| ƒ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.)r   r   Ú
isinstancer    rF   rI   r   r   r   rJ   ·   s    ÿ
ztuple[int, int]zLiteral[True])Úfreeze_panesr   c                 C  s   d S rH   r   ©rL   r   r   r   Úvalidate_freeze_panesÖ   s    rN   zLiteral[False]c                 C  s   d S rH   r   rM   r   r   r   rN   Û   s    ztuple[int, int] | NoneÚboolc                 C  s6   | d ur2t | ƒdkr*tdd„ | D ƒƒr*dS tdƒ‚dS )Né   c                 s  s   | ]}t |tƒV  qd S rH   )rK   r0   )Ú.0Úitemr   r   r   Ú	<genexpr>â   s   z(validate_freeze_panes.<locals>.<genexpr>TzLfreeze_panes must be of form (row, column) where row and column are integersF)ÚlenÚallr   rM   r   r   r   rN   à   s    ÿÿzlist[Hashable]z
list[bool]z!tuple[list[Hashable], list[bool]])ÚrowÚcontrol_rowr   c                 C  sf   | d }t dt| ƒƒD ]F}|| s*| | }| | dksB| | du rL|| |< qd||< | | }q| |fS )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
    r   r5   Ú NF)rB   rT   )rV   rW   ÚlastÚir   r   r   Úfill_mi_headerñ   s    

r[   zint | Sequence[int]z&tuple[Hashable | None, list[Hashable]])rV   Ú	index_colr   c                 C  sp   t |ƒr t|tƒsJ ‚t|ƒ}nt|tƒr.J ‚|}| | }|dkrFdn|}|| d|… dg | |d d…  fS )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.
    rX   Nr5   )r   rK   r   Úmax)rV   r\   rZ   Úheader_namer   r   r   Úpop_header_name  s    
r_   zdict[str, Any] | NoneÚdict)Úengine_kwargsÚkwargsr   c                 C  s$   | du ri }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)ÚcopyÚupdate)ra   rb   Úresultr   r   r   Úcombine_kwargs6  s
    
rf   N)r   )%Ú
__future__r   Úcollections.abcr   r   r   r   Útypingr   r   r	   r
   r   r   Zpandas.compat._optionalr   Zpandas.core.dtypes.commonr   r   Zpandas.io.excel._baser   Útyper   Úobjectr   r   Ú__annotations__r   r+   r/   r<   rF   rJ   rN   r[   r_   rf   r   r   r   r   Ú<module>   s>    	) !#"