3
���h�Y � @ s� d Z ddlmZ ddlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZmZ ddlmZ ddlmZmZmZmZ dd l m Z G d
d � d e�Z G d d
� d
e �Z dS )z,
Module for formatting output data in HTML.
� )�dedent)
�IO�Any�Dict�Iterable�List�Mapping�Optional�Tuple�Union�cast)�
get_option)�lib)�
MultiIndex�option_context)�is_url)�DataFrameFormatter�TableFormatter�buffer_put_lines�get_level_lengths)� pprint_thingc
@ s e Zd ZdZdZd>eeeee e e
edf f ee dd�dd�Z e
ed�d d
��Ze
ed�d d ��Ze
e d�d
d��Zed�dd�Ze
ed�dd��Ze
e d�dd��Zd?ee dd�dd�Zd@eee ee dd�dd �ZdAee ee dd �d d �ZdBeee ee dd!�d"d#�ZdCee e eee eee ef e dd$�d%d&�Z e e d�d'd(�Z e e dd)�d*d+�Z dDe dd,�d-d.�Z e dd,�d/d0�Z!e dd,�d1d2�Z"e dd,�d3d4�Z#ee e e f d�d5d6�Z$e dd,�d7d8�Z%e&e e e f e dd9�d:d;�Z'e&e e e f e dd9�d<d=�Z(dS )E�
HTMLFormatterac
Internal class for formatting output data in html.
This class is intended for shared functionality between
DataFrame.to_html() and DataFrame._repr_html_().
Any logic in common with other output formatting methods
should ideally be inherited from classes in format.py
and this class responsible for only producing html markup.
� N.)� formatter�classes�border�returnc C s� || _ || _| j j| _| j jj| _g | _| j j| _| j j| _| j j| _|d kr\t t
t d��}|| _ | j j
| _
| j j| _dd� | j jj� D �| _d S )Nzdisplay.html.borderc S s* i | ]"\}}t |t�r |� d �n||�qS )Zpx)�
isinstance�int)�.0�column�value� r" �8/tmp/pip-build-5_djhm0z/pandas/pandas/io/formats/html.py�
<dictcomp>9 s z*HTMLFormatter.__init__.<locals>.<dictcomp>)�fmtr �frame�tr_frame�columns�elements� bold_rows�escapeZshow_dimensionsr r r
r �table_id� render_links� col_space�items)�selfr r r r" r" r# �__init__$ s
zHTMLFormatter.__init__)r c C s | j jS )N)r% �show_row_idx_names)r0 r" r" r# r2 > s z HTMLFormatter.show_row_idx_namesc C s | j jS )N)r% �show_col_idx_names)r0 r" r" r# r3 B s z HTMLFormatter.show_col_idx_namesc C s | j jr| jjjS | jr dS dS )N� r )r% �indexr& �nlevelsr3 )r0 r" r" r# �
row_levelsF s
zHTMLFormatter.row_levelsc C s | j S )N)r( )r0 r" r" r# � _get_columns_formatted_valuesU s z+HTMLFormatter._get_columns_formatted_valuesc C s | j jS )N)r% � is_truncated)r0 r" r" r# r9 Y s zHTMLFormatter.is_truncatedc C s t | jjj�S )N)�lenr% r' r( )r0 r" r" r# �ncols] s zHTMLFormatter.ncolsr )�s�indentr c C s t |�}| jjd| | � d S )N� )r r) �append)r0 r< r= �rsr" r" r# �writea s zHTMLFormatter.writeF)r< �headerr= �tagsr c C sH | j j|d�}|r2|dk r2|p d}|d|� d�7 }| j|d||d� dS )a�
Method for writing a formatted <th> cell.
If col_space is set on the formatter then that is used for
the value of min-width.
Parameters
----------
s : object
The data to be written inside the cell.
header : bool, default False
Set to True if the <th> is for use inside <thead>. This will
cause min-width to be set if there is one.
indent : int, default 0
The indentation level of the cell.
tags : str, default None
Tags to include in the cell.
Returns
-------
A written <th> cell.
N� zstyle="min-width: z;"�th)�kindr= rC )r. �get� _write_cell)r0 r< rB r= rC r. r" r" r# �write_the s
zHTMLFormatter.write_th)r< r= rC r c C s | j |d||d� d S )N�td)rF r= rC )rH )r0 r< r= rC r" r" r# �write_td� s zHTMLFormatter.write_tdrJ )r< rF r= rC r c
C s� |d k r d|� d|� d�}n d|� d�}| j r<dddd�}ni }t||d�j� }| jr�t|�r�t|i d�j� }|d |� d
�7 }d } nd } | j|� |� | � d
|� d�|� d S )N�<r> �>z&z<z>)�&rL rM )Z escape_charsz <a href="z" target="_blank">z</a>rD z</)r+ r �stripr- r rA )
r0 r< rF r= rC Z start_tag�escr@ Z rs_unescapedZend_ar" r" r# rH � s zHTMLFormatter._write_cell)�liner= � indent_deltarB �alignrC �
nindex_levelsr c C s� |d kr i }|d kr"| j d|� n| j d|� d�|� ||7 }xVt|�D ]J\}} |j|d �}
|sn| jr�||k r�| j| |||
d� qH| j| ||
d� qHW ||8 }| j d|� d S )Nz<tr>z<tr style="text-align: z;">)r= rB rC )rC z</tr>)rA � enumeraterG r* rI rK ) r0 rQ r= rR rB rS rC rT �ir< Zval_tagr" r" r# �write_tr� s
zHTMLFormatter.write_trc C sH | j � | jrBtd�}| jdt| j�� d|� dt| jj�� d�� | jS )N�� z<p>z rows r> z columns</p>)� _write_tableZshould_show_dimensions�chrrA r: r& r( r) )r0 Zbyr" r" r# �render� s (zHTMLFormatter.render)�bufr c C s t || j� � d S )N)r r[ )r0 r\ r" r" r# � write_result� s zHTMLFormatter.write_result)r= r c C s� dg}t d�}|s |jd� | jd k rnt| jt�r>| jj� | _t| jttf�sbtdt | j�� ���|j
| j� | j d kr~d}nd| j � d�}| j d| j
� d d
j|�� d|� d �|� | jjs�| jr�| j|| j � | j|| j � | j d |� d S )
NZ dataframezdisplay.html.use_mathjaxZtex2jax_ignorez.classes must be a string, list, or tuple, not rD z id="�"z<table border="z " class="r> rM z</table>)r
r? r r �str�split�list�tuple� TypeError�type�extendr, rA r �joinr% rB r2 �
_write_headerrR � _write_body)r0 r= Z_classesZ use_mathjaxZ
id_sectionr" r" r# rY � s*
zHTMLFormatter._write_tablec C sj | j j}t| jt��r�d}| j jr*tj}nd}| jj|ddd�}t ||�}t
|�d }�x t t ||��D �]^\}\} }
|�r�| j j
} | j j�r�i } x�t| j� �D ]�\}
}|
| kr�|| |
d < nr|
| | k�r(|d | |
< ||k�r |
d | � d |
| d � }
n&|
d | � |
| d f |
| d � }
n|| |
< |
| | kr�d| | < |
d | � d |
| d � }
q�W | } t
|�d }||k�r�d| | < nji } x:t| j� �D ]*\}
}|
| k�r�|| |
d < n|| |
< �q�W d| | < | } |
d | � dg |
| d � }
dg| jd }| j j�s| j�rJ| j j�r@| jj| }|jt|�p8d�� n
|jd� i }t
|�}xZt |
�D ]N\}}|| k�r`| | dk�r�|j| | d�||< n�q`|d7 }|j|� �q`W | j||| j|dd � qhW n�dg| jd }| j j�s�| j�r| j j�r|j| jj�p
d� n
|jd� |j| j� � | j j }|�rP| j| j j
} |j | d� | j||| jd|d
� d S )
Nz colspan="{span:d}" halign="left"F)�sparsify�adjoin�namesr4 �...rD )�spanT)rC rB )rB rS )rl )rl ) r% �
truncate_hr r( r ri r �
no_default�formatr r: rU �zip�
tr_col_numra r/ r7 r5 r3 Zshow_index_namesrk r? r rW rR �namere r8 Zjustify�insert)r0 r= rn �template�sentinel�levels�
level_lengths� inner_lvl�lnum�records�valuesZins_colZrecs_new�tagrm �rowrs rC �jrV �vrS r" r" r# �_write_col_header� s�
(
z HTMLFormatter._write_col_headerc C sL | j j}dd� | jjjD �dg| j|r*dnd }| j||| jdd� d S )Nc S s g | ]}|d k r|nd�qS )NrD r" )r �xr" r" r# �
<listcomp>d s z3HTMLFormatter._write_row_header.<locals>.<listcomp>rD r4 r T)rB )r% rn r& r5 rk r; rW rR )r0 r= rn r~ r" r" r# �_write_row_headerb s z HTMLFormatter._write_row_headerc C sJ | j d|� | jjr$| j|| j � | jr:| j|| j � | j d|� d S )Nz<thead>z</thead>)rA r% rB r� rR r2 r� )r0 r= r" r" r# rg i s zHTMLFormatter._write_headerc
s2 t dd �� � fdd�t� j�D �}W d Q R X |S )Nzdisplay.max_colwidthc s i | ]}� j j|�|�qS r" )r% � _format_col)r rV )r0 r" r# r$ v s z7HTMLFormatter._get_formatted_values.<locals>.<dictcomp>)r �ranger; )r0 �
fmt_valuesr" )r0 r# �_get_formatted_valuest s "z#HTMLFormatter._get_formatted_valuesc C s` | j d|� | j� }| jjr>t| jjt�r>| j||| j � n| j ||| j � | j d|� d S )Nz<tbody>z</tbody>)
rA r� r% r5 r r& r �_write_hierarchical_rowsrR �_write_regular_rows)r0 r= r� r" r" r# rh y s zHTMLFormatter._write_body)r� r= r c s, | j j}| j j}t| j j�}| j jrX| j jd�}|d k rJ| j jjj|�}n| j jjj� }g }x�t |�D ]��|r��| j j
kr�dgt|� } | j | || j d | j
d� g }| j jr�|j|� � n| jr�|jd� |j� �fdd�t | j�D �� |�r | j j| j
}
|j|
d� | j ||| j d | j
d� qfW d S )N� __index__z...)rC rT rD c 3 s | ]}� | � V qd S )Nr" )r r )r� rV r" r# � <genexpr>� s z4HTMLFormatter._write_regular_rows.<locals>.<genexpr>)r% rn �
truncate_vr: r' r5 Z_get_formatter�maprp r� �
tr_row_numrW rR r7 r? r3 re r; rr rt ) r0 r� r= rn r� �nrowsr% Z index_valuesr~ � str_sep_rowZ
dot_col_ixr" )r� rV r# r� � s:
z!HTMLFormatter._write_regular_rowsc s� d}| j j}| j j}| j j}t|�}t|jt�s4t�|jj dddd�}t
t |� �}| j j �rt
j} |jj | ddd�}
t|
| �} t| �d } |�r| j j}
tt|
�}
d}�x4t| �D �]&\}}i }�xt
|j� �D � ]�\}}||
kr�|||d < n||| |
k�rd|d ||< |�s>t
||
d �}d|d
< |j|
t|�� d}n$t
||
�}d|| | < t|�||
< n|||< || |
kr�d||
< |dk�r�|j|
tdgt| � �� q�|r�t
||
�}d|| | < t|�||
< q�W || |< q�W d| | |
< x&tt� ��D ]}� | j|
d� �q�W |d7 }x�t|�D ]މg }i }d}d}xht | |� �D ]V\}}�|k�r~|� dk�r�|j |� d�||< n |d7 }�qH|d7 }|j|� �qHW |j� �fd d
�t| j�D �� |�r�|j| j| | j j d� | j ||| j |t|
�| d � �q$W n�g }x�tt|��D ]|�rX�| j jk�rXdgt|� }| j ||| j d | jd � t
t |jj dddd�� �}g }|j|� � |j� �fd d
�t| j�D �� |�r�|j| j| j j d� | j ||| j d |jj d � �qW d S )Nz rowspan="{span}" valign="top"F)ri rj rk r4 z...Tr )rm c 3 s | ]}� | � V qd S )Nr" )r r )r� rV r" r# r� s z9HTMLFormatter._write_hierarchical_rows.<locals>.<genexpr>)rC rT c 3 s | ]}� | � V qd S )Nr" )r r )r� rV r" r# r� ' s ���) r% rn r� r' r: r r5 r �AssertionErrorrp ra rq ri r ro r r� r r rU r/ rt rb r� r? re r; r7 rr rW rR r6 ) r0 r� r= ru rn r� r& r� Z
idx_valuesrv rw rx ry Zins_rowZinsertedrz r{ Zrec_newr} rm Zdot_rowZix_colr~ rC Z
sparse_offsetr r� r� r" )r� rV r# r� � s�
z&HTMLFormatter._write_hierarchical_rows)NN)r )Fr N)r N)rJ r N)r r FNNr )r ))�__name__�
__module__� __qualname__�__doc__rR r r r r_ r r
r r1 �property�boolr2 r3 r7 r r8 r9 r; r rA rI rK rH r rW r[ r r] rY r� r� rg r� rh r r� r� r" r" r" r# r sR & u
-r c sd e Zd ZdZeeee f d�dd�Zee d�dd�Z dd�dd �Z
ee d�� fd
d � Z � Z S ) �NotebookFormatterz�
Internal class for formatting output data in html for display in Jupyter
Notebooks. This class is intended for functionality specific to
DataFrame._repr_html_() and DataFrame.to_html(notebook=True)
)r c s � fdd�t � j�D �S )Nc s i | ]}� j j|�|�qS r" )r% r� )r rV )r0 r" r# r$ ; s z;NotebookFormatter._get_formatted_values.<locals>.<dictcomp>)r� r; )r0 r" )r0 r# r� : s z'NotebookFormatter._get_formatted_valuesc C s
| j j� S )N)r( rp )r0 r" r" r# r8 = s z/NotebookFormatter._get_columns_formatted_valuesNc s� d}d}d� ddg}t | jt�r<|jd� | jrF|jd� n
|jd� djt� fdd�|��}tdj|||f��}| j|� d S )Nz <style scoped>z </style>zM .dataframe %s {
%s: %s;
}�tbody tr th:only-of-type�vertical-align�middle� tbody tr th�top� thead tr th�
text-align�left�thead tr:last-of-type th�right�thead thz
c s � | S )Nr" )�t)�template_selectr" r# �<lambda>X s z/NotebookFormatter.write_style.<locals>.<lambda>�
)r� r� r� )r� r� r� )r� r� r� )r� r� r� )r� r� r� ) r r( r r? r2 rf r� r rA )r0 Ztemplate_firstZ
template_lastZ
element_propsZ template_midru r" )r� r# � write_style@ s
z NotebookFormatter.write_stylec s, | j d� | j� t� j� | j d� | jS )Nz<div>z</div>)rA r� �superr[ r) )r0 )� __class__r" r# r[ \ s
zNotebookFormatter.render)
r� r� r� r� r r r r_ r� r8 r� r[ �
__classcell__r" r" )r� r# r� 3 s
r� N) r� �textwrapr Ztypingr r r r r r r r
r r Zpandas._configr
Z pandas._libsr Zpandasr r Zpandas.io.commonr Zpandas.io.formats.formatr r r r Zpandas.io.formats.printingr r r� r" r" r" r# �<module> s 0 |