3
���hd � @ s d Z ddlmZ ddlZddlZddlmZmZmZm Z m
Z
ddl Z ddl Z
ddlmZ ddlmZ ddlmZmZ ddlmZ dd lmZmZmZmZ ddl j j Z dd
l m!Z! dd l"m#Z#m$Z$ dd l%m&Z& dd
l'm(Z( G dd� d�Z)G dd� d�Z*G dd� d�Z+dS )zC
Utilities for conversion to writer-agnostic Excel representation.
� )�reduceN)�Callable�Dict�Optional�Sequence�Union)�Label)�missing)�is_float� is_scalar)�ABCIndex)� DataFrame�Index�
MultiIndex� PeriodIndex)�stringify_path)� CSSResolver�
CSSWarning)�get_level_lengths)� pprint_thingc @ s& e Zd Zd ZeZd eed�d d
�ZdS )
� ExcelCell�row�col�val�style�
mergestart�mergeendN)r r c C s( || _ || _|| _|| _|| _|| _d S )N)r r r r r r )�selfr r r r r r � r �9/tmp/pip-build-5_djhm0z/pandas/pandas/io/formats/excel.py�__init__ s zExcelCell.__init__)r r r r r r )NNN)�__name__�
__module__� __qualname__Z
__fields__� __slots__�intr r r r r r s r c @ s� e Zd ZdZd:ee d�dd�Ze� Zee ee eef f d�dd�Z
e eef e ee eef f d �d
d �Z d d d
dddd�Z e eee
eef f d�dd�Ze e ee eef f d �dd�Zee d�dd�Ze eef d�dd�Zddddddd d d d d d d d �
Zd ddd �Ze eee
eeef f d�d d �Zd!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d1d2d3d4�Zee d5�d6d7�Ze e eee f d �d8d9�ZdS );�CSSToExcelConverteraK
A callable for converting CSS declarations to ExcelWriter styles
Supports parts of CSS 2.2, with minimal CSS 3.0 support (e.g. text-shadow),
focusing on font styling, backgrounds, borders and alignment.
Operates by first computing CSS styles in a fairly generic
way (see :meth:`compute_css`) then determining Excel style
properties from CSS properties (see :meth:`build_xlstyle`).
Parameters
----------
inherited : str, optional
CSS declarations understood to be the containing scope for the
CSS processed by :meth:`__call__`.
N)� inheritedc C s |d k r| j |�}|| _d S )N)� compute_cssr' )r r' r r r r B s
z CSSToExcelConverter.__init__)�declarations_str�returnc C s | j || j�}| j|�S )ax
Convert CSS declarations to ExcelWriter style.
Parameters
----------
declarations_str : str
List of CSS declarations.
e.g. "font-weight: bold; background: blue"
Returns
-------
xlstyle : dict
A style as interpreted by ExcelWriter when found in
ExcelCell.style.
)r( r' �
build_xlstyle)r r) Z
propertiesr r r �__call__J s z CSSToExcelConverter.__call__)�propsr* c sV | j |�| j|�| j|�| j|�| j|�d�}tttf d d�� fdd� � � |� |S )N)� alignmentZborder�fill�fontZ
number_format)�dr* c sJ xDt | j� �D ]4\}}|dkr&| |= qt|t�r� |� |s| |= qW dS )z4Remove key where value is None, through nested dictsN)�list�items�
isinstance�dict)r1 �k�v)� remove_noner r r8 i s
z6CSSToExcelConverter.build_xlstyle.<locals>.remove_none)�build_alignment� build_border�
build_fill�
build_font�build_number_formatr �str)r r- �outr )r8 r r+ ^ s
z!CSSToExcelConverter.build_xlstyle�top�center�bottom)r@ ztext-topZmiddleZbaselinerB z text-bottom)r* c C s: |j d�| jj |j d��|j d�d kr*d n
|d dkd�S ) Nz
text-alignzvertical-alignz white-space�nowrap�pre�pre-line)�
horizontal�vertical� wrap_text)rC rD rE )�get� VERTICAL_MAP)r r- r r r r9 � s z#CSSToExcelConverter.build_alignmentc s � �fdd�dD �S )Nc sP i | ]H}�j � jd |� d��� jd |� d����j� jd |� d���d�|�qS )zborder-z-stylez-widthz-color)r �color)�
_border_stylerI �color_to_excel)�.0Zside)r- r r r �
<dictcomp>� s z4CSSToExcelConverter.build_border.<locals>.<dictcomp>r@ �rightrB �left)r@ rP rB rQ r )r r- r )r- r r r: � s z CSSToExcelConverter.build_border)r c C s� |d kr|d krd S |dks$|dkr(d S |d kr4d}t |d d� �}|dk rPd S |dk r^d}n|dk rld }nd
}|dkr|d}|dkr�dS |dkr�|S |dkr�|dkr�dS dS |dkr�|dkr�dS dS d S )N�none�hiddenZ2pt� g�h㈵��>g�������?�thingffffff@ZmediumZthick�groove�ridge�inset�outset�solid�doubleZdotted�hairZmediumDashDotDotZdashedZ mediumDashed���)NrV rW rX rY )r\ rU )r\ rU )�float)r r �widthZ
width_namer r r rL � s8 z!CSSToExcelConverter._border_style)r- c C s& |j d�}|dkr"| j|�dd�S d S )Nzbackground-color� transparentrR rZ )ZfgColorZ patternType)Nr` rR )rI rM )r r- Z
fill_colorr r r r; � s
z CSSToExcelConverter.build_fillTF)
�boldZbolderZ600Z700Z800Z900�normalZlighterZ100Z200Z300Z400�500)rb �italicZobliquec
C s� |j d�}|d k r0|jd�s t�t|d d � �}tjd|j dd��}g }xn|D ]f}|d d� dkrx|dd � jd d�}n.|d d� d
kr�|dd � jd d
�}n|j� }|rN|j|� qNW d }xJ|D ]B}|d kr�d}P q�|d
kr�d}P q�|dkr�d}P q�|dkr�d}P q�W |j d�}|d k �r&|j � }nf }|�r8|d nd ||| j
j |j d��| j j |j d��d|k�rldnd d|k�pzd | j |j d��d|k�r�t
tjd|d ��nd d � S ) Nz font-size�ptrT z�(?x)
(
"(?:[^"]|\\")+"
|
'(?:[^']|\\')+'
|
[^'",]+
)(?=,|\s*$)
z font-family� � �"z\"�'z\'Zserifz
sans-serifZcursive� Zfantasy� ztext-decorationr z font-weightz
font-style� underlineZsinglez line-throughrK z text-shadowz ^[^#(]*[1-9]) �name�family�sizera rd rl �strikerK Zshadowr] ���rq )rI �endswith�AssertionErrorr^ �re�findall�replace�strip�append�split�BOLD_MAP�
ITALIC_MAPrM �bool�search)r r- ro Zfont_names_tmpZ
font_namesrm rn Z
decorationr r r r< � sX
z CSSToExcelConverter.build_fontZ800000ZA52A2AZFF0000ZFFC0CBZFFA500ZFFFF00Z808000Z008000Z800080ZFF00FFZ00FF00Z008080Z00FFFFZ0000FFZ000080Z000000Z808080ZC0C0C0ZFFFFFF)ZmaroonZbrownZredZpinkZorangeZyellowZoliveZgreenZpurpleZfuchsiaZlimeZtealZaquaZblueZnavyZblackZgrayZgreyZsilverZwhite)r c C s� |d kr d S |j d�r2t|�dkr2|dd � j� S |j d�rpt|�dkrp|d d |d d |d d j� S y
| j| S tk
r� tjdt|�� �t� Y nX d S )N�#� rg rj rT � zUnhandled color format: ) �
startswith�len�upper� NAMED_COLORS�KeyError�warnings�warn�reprr )r r r r r rM A s (
z"CSSToExcelConverter.color_to_excelc C s d|j d�iS )NZ format_codez
number-format)rI )r r- r r r r= M s z'CSSToExcelConverter.build_number_format)N)r! r" r# �__doc__r r> r r r( r r, r+ rJ r r| r9 r: rL r; rz r{ r% r< r� rM r= r r r r r& + sh (" 1 $J r& c @ s� e Zd ZdZd&Zd'Zd(eee eee e
ee e f e ee
e ee f e eee d
� d d �Z
ed
d� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zed �d d �Zd d!� Zd)d$d%�ZdS )*�ExcelFormattera^
Class for formatting a DataFrame to a list of ExcelCells,
Parameters
----------
df : DataFrame or Styler
na_rep: na representation
float_format : string, default None
Format string for floating point numbers
cols : sequence, optional
Columns to write
header : boolean or list of string, default True
Write out column names. If a list of string is given it is
assumed to be aliases for the column names
index : boolean, default True
output row names (index)
index_label : string or sequence, default None
Column label for index column(s) if desired. If None is given, and
`header` and `index` are True, then the index names are used. A
sequence should be given if the DataFrame uses MultiIndex.
merge_cells : boolean, default False
Format MultiIndex and Hierarchical Rows as merged cells.
inf_rep : string, default `'inf'`
representation for np.inf values (which aren't representable in Excel)
A `'-'` sign will be added in front of -inf.
style_converter : callable, optional
This translates Styler styles (CSS) into ExcelWriter styles.
Defaults to ``CSSToExcelConverter()``.
It should have signature css_declarations string -> excel style.
This is only called for body cells.
rT � � rf NTF�inf) �na_rep� float_format�cols�header�index� index_label� merge_cells�inf_rep�style_converterc C s� d| _ || _t|t�s8|| _|j}|
d kr0t� }
|
| _nd | _|| _|d k r�t t
|�|j @ �sft d��t t
|�|j @ �t |�kr�t d��|j
|d�| _| jj | _ || _|| _|| _|| _|| _| | _d S )Nr z,passes columns are not ALL present dataframez.Not all names specified in 'columns' are found)�columns)�
rowcounterr� r4 r
�styler�datar& r� �dfr� r r� r� Zreindexr� r� r� r� r� r� ) r r� r� r� r� r� r� r� r� r� r� r r r r u s.
zExcelFormatter.__init__c C s ddiddddd�ddd�d�S ) Nra TrU )r@ rP rB rQ rA r@ )rF rG )r0 Zbordersr. r )r r r r � header_style� s zExcelFormatter.header_stylec C s� t |�rtj|�r| j}nJt|�rdtj|�r4| j}n0tj|�rLd| j� �}n| jd k rdt | j| �}t
|dd �d k r|t d��|S )N�-�tzinfoz{Excel does not support datetimes with timezones. Please ensure that datetimes are timezone unaware before writing to Excel.) r r Zisnar� r
Zisposinf_scalarr� Zisneginf_scalarr� r^ �getattr�
ValueError)r r r r r �
_format_value� s
z ExcelFormatter._format_valuec c s� | j jdkr| jstd��t| jtttj t
f�}|p8| js>d S | j }|j | j ddd�}t
|�}d}d}| jr�t| jjt�r�t| jjd �d }| j �rhx0tt|��D ] }|j| }t|||| j�V q�W x�tt||j|j��D ]�\}\}} }
| j|
�} xj|D ]b} || dk�r>t||| d | | | j||| || �V n t||| d | | | j�V � q�W q�W nFxDtt|� �D ]4\} } djtt | ��}
t||| d |
| j�V �qvW || _ d S )Nrg z]Writing to Excel with MultiIndex columns and no index ('index'=False) is not yet implemented.F)�sparsify�adjoin�namesr �.) r� Znlevelsr� �NotImplementedErrorr4 r� �tupler2 �np�ndarrayr �formatr� r r� r r� �ranger� r r� � enumerate�zip�levels�codes�take�join�mapr r� )r � has_aliasesr� �
level_strs�
level_lengths� coloffset�lnumrm �spansr� � level_codes�values�ir7 r r r �_format_header_mi� sJ
( z ExcelFormatter._format_header_mic c s� t | jtttjtf�}|s | jr�d}| jrLd}t | jjt �rLt
| jjd �}| j }|r�t
| j�t
| j �kr�t dt
| j �� dt
| j�� d���n| j}x,t
|�D ] \}}t| j|| || j�V q�W d S )Nr rg zWriting z cols but got z aliases)r4 r� r� r2 r� r� r r� r� r r� r� r� r� r r� r� )r r� r� Zcolnames�colindexZcolnamer r r �_format_header_regular� s
"z%ExcelFormatter._format_header_regularc s� t � jt�r� j� }n� j� }f }� jjjr�dd� � jjjD �dgt� j� }t dd� t
dd� |��r�� fdd�t |�D �}� j d 7 _ t
j||�S )
Nc S s g | ]}|d k r|nd�qS )Nrf r )rN �xr r r �
<listcomp> s z1ExcelFormatter._format_header.<locals>.<listcomp>rf c S s | o|S )Nr )r� �yr r r �<lambda> s z/ExcelFormatter._format_header.<locals>.<lambda>c S s | dkS )Nrf r )r� r r r r� s c 3 s$ | ] \}}t � j||� j�V qd S )N)r r� r� )rN r� r )r r r � <genexpr> s z0ExcelFormatter._format_header.<locals>.<genexpr>rg )r4 r� r r� r� r� r� r� r� r r� r� r� � itertools�chain)r �genZgen2r r )r r �_format_header s
z ExcelFormatter._format_headerc C s" t | jjt�r| j� S | j� S d S )N)r4 r� r� r �_format_hierarchical_rows�_format_regular_rows)r r r r � _format_body# s zExcelFormatter._format_bodyc c s@ t | jtttjtf�}|s | jr.| jd7 _| j�r | j r^t | j tttjt
f�r^| j d }n(| j rxt | j t �rx| j }n| j jj
d }t | jt�r�| jd7 _|r�| jdk r�t| jd d|| j�V | j j}t | j jt�r�| j jj� }x,t|�D ] \}}t| j| d|| j�V q�W d}nd}x| j|�D ] }|V �q,W d S )Nrg r F)r4 r� r� r2 r� r� r r� r� r� r r> r� r� r� r r r� r Z to_timestampr� �_generate_body)r r� r� Z index_values�idxZidxvalr� �cellr r r r� ) s.
z#ExcelFormatter._format_regular_rowsc
c s t | jtttjtf�}|s | jr.| jd7 _d}| j�r�| j jj
}| j rft | j tttjt f�rf| j }t | j
t�r�| jr�| jd7 _tj|� r�| jdk r�x,t|�D ] \}}t| jd ||| j�V q�W | j�r�| j jjdddd�}t|�}x�t|| j jj| j jj�D ]�\}} }
| j|
| jdd�} xl|D ]d} || dk�rnt| j| || | | j| j| || d |�V n t| j| || | | j�V �q(W |d7 }�qW nPxNt| j j� D ]>}
x.t|
�D ]"\}}t| j| ||| j�V �q�W |d7 }�q�W x| j |�D ] }|V �q�W d S )Nrg r FT)r� r� r� )Z
allow_fillZ
fill_value) r4 r� r� r2 r� r� r r� r� r� r� r� r r� r r� �comZ any_not_noner� r r� r� r r� r� r� r� Z _can_hold_nar� )r r� ZgcolidxZ index_labelsZcidxrm r� r� r� r� r� r� r� Z indexcolvalsr� Z indexcolvalr� r r r r� Q s^
z(ExcelFormatter._format_hierarchical_rows)r� c c s� | j d krd }n| j j� j}|s$d }d }xztt| j��D ]h}| jjd d �|f }xNt|�D ]B\}}|d k r�| j dj
|||f ��}t | j | || ||�V qZW q8W d S )N�;)
r� Z_compute�ctxr� r� r� r� Zilocr� r� r� r r� )r r� ZstylesZxlstyleZcolidxZseriesr� r r r r r� � s
z ExcelFormatter._generate_bodyc c s6 x0t j| j� | j� �D ]}| j|j�|_|V qW d S )N)r� r� r� r� r� r )r r� r r r �get_formatted_cells� s z"ExcelFormatter.get_formatted_cells�Sheet1r c C s� ddl m} | jj\}} || jks,| | jkrPtd|� d| � d| j� d| j� ���t||�r`d}
n|t|�|d�}d}
| j � } |j
| ||||d � |
r�|j � d
S ) a�
writer : string or ExcelWriter object
File path or existing ExcelWriter
sheet_name : string, default 'Sheet1'
Name of sheet which will contain DataFrame
startrow :
upper left cell row to dump data frame
startcol :
upper left cell column to dump data frame
freeze_panes : tuple of integer (length 2), default None
Specifies the one-based bottommost row and rightmost column that
is to be frozen
engine : string, default None
write engine to use if writer is a path - you can also set this
via the options ``io.excel.xlsx.writer``, ``io.excel.xls.writer``,
and ``io.excel.xlsm.writer``.
r )� ExcelWriterz-This sheet is too large! Your sheet size is: z, z Max sheet size is: F)�engineT)�startrow�startcol� freeze_panesN) Zpandas.io.excelr� r� �shape�max_rows�max_colsr� r4 r r� Z write_cells�save) r �writerZ
sheet_namer� r� r� r� r� Znum_rowsZnum_colsZ need_saveZformatted_cellsr r r �write� s$ "
zExcelFormatter.writei i @ ) rf NNTTNFr� N)r� r r NN)r! r" r# r� r� r� r> r r r r r| r r �propertyr� r� r� r� r� r� r� r� r% r� r� r� r r r r r� Q s6 L!
7(M r� ),r� � functoolsr r� rt Ztypingr r r r r r� Znumpyr� Zpandas._typingr Zpandas.core.dtypesr Zpandas.core.dtypes.commonr
r Zpandas.core.dtypes.genericr Zpandasr
r r r Zpandas.core.common�core�commonr� Zpandas.io.commonr Zpandas.io.formats.cssr r Zpandas.io.formats.formatr Zpandas.io.formats.printingr r r& r� r r r r �<module> s* ( |