3
���h� � @ st d dl mZ d dlmZmZ i Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� ZdS )� )�import_optional_dependency)�
is_integer� is_list_likec C s" 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�
ValueErrorZengine�_writers)�klass� engine_name� r
�7/tmp/pip-build-5_djhm0z/pandas/pandas/io/excel/_util.py�register_writer s
r c C s0 ddddd�}t dddd�}|r(d|d < || S )
z�
Return the default writer for the given extension.
Parameters
----------
ext : str
The excel file extension for which to get the default engine.
Returns
-------
str
The default engine for the extension.
ZopenpyxlZxlwtZodf)�xlsxZxlsmZxlsZods�
xlsxwriterF�warn)Zraise_on_missingZ
on_versionr
)r )�extZ_default_writersr r
r
r �_get_default_writer s r c C sB yt | S tk
r< } ztd| � d��|�W Y d d }~X nX d S )NzNo Excel writer '�')r �KeyErrorr )r �errr
r
r �
get_writer4 s r c C sh d}xZ| j � j� D ]J}t|�}|td�k s6|td�krDtd| � ���|d | td� d }qW |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�ZzInvalid column name: � � )�upper�strip�ordr )�x�index�c�cpr
r
r �
_excel2num; s r! c C sb g }xX| j d�D ]J}d|krL|j d�}|jtt|d �t|d �d �� q|jt|�� qW |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 r )�split�extend�ranger! �append)Zareas�cols�rngr
r
r � _range2cols[ s
&r* c C s2 | dkr | S t | �r td��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 �
isinstance�strr* )Zusecolsr
r
r �_maybe_convert_usecols| s
r- c C s6 | d k r2t | �dkr*tdd� | D ��r*dS td��dS )N� c s s | ]}t |t�V qd S )N)r+ �int)�.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 )Z freeze_panesr
r
r �_validate_freeze_panes� s r5 c C s: x4t | �dkr4| d dks&| d d kr4| dd � } qW | S )Nr � r )r3 )�rowr
r
r �_trim_excel_header� s &r8 c C sj | d }xXt dt| ��D ]F}|| s,| | }| | dksD| | dkrN|| |< qd||< | | }qW | |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
r r r6 NF)r& r3 )r7 Z control_row�last�ir
r
r �_fill_mi_header� s
r; c C sR t |�s |nt|�}| | }|dkr(dn|}|| d|� dg | |d d� fS )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.
r6 Nr )r �max)r7 Z index_colr: � header_namer
r
r �_pop_header_name� s r> N)Zpandas.compat._optionalr Zpandas.core.dtypes.commonr r r r r r r! r* r- r5 r8 r; r> r
r
r
r �<module> s ! ! |