o
    d[                     @  s   d Z ddlmZ ddl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 ddlmZmZ ddlmZ dd	lmZmZ dd
lmZ G dd dZG dd deZdS )z,
Module for formatting output data in HTML.
    )annotations)dedent)AnyFinalHashableIterableMappingcast)
get_option)lib)
MultiIndexoption_context)is_url)DataFrameFormatterget_level_lengths)pprint_thingc                   @  sB  e Zd ZU dZdZded< 				dTdUddZdVddZdWddZe	dXddZ
e	dXddZe	dXdd Ze	dYd"d#ZdZd%d&Ze	dXd'd(Ze	dYd)d*Zd[d\d/d0Z	d]d^d3d4Zd_d`d5d6Z	dadbd9d:Z	+	+				+dcddd?d@Zd[dedAdBZdedCdDZdedEdFZdedGdHZdfdJdKZdedLdMZdgdPdQZdgdRdSZdS )h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.
       r   indent_deltaNF	formatterr   classes(str | list[str] | tuple[str, ...] | Noneborderint | bool | Nonetable_id
str | Nonerender_linksboolreturnNonec                 C  s   || _ || _| j j| _| j jj| _g | _| j j| _| j j| _| j j| _|d u s+|du r3t	t
td}n|s7d }|| _|| _|| _dd | j j D | _d S )NTzdisplay.html.borderc                 S  s*   i | ]\}}|t |tr| d n|qS )Zpx)
isinstanceint).0columnvalue r%   J/app/.heroku/python/lib/python3.10/site-packages/pandas/io/formats/html.py
<dictcomp>G   s    z*HTMLFormatter.__init__.<locals>.<dictcomp>)fmtr   frametr_framecolumnselements	bold_rowsescapeZshow_dimensionsr	   r!   r
   r   r   r   	col_spaceitems)selfr   r   r   r   r   r%   r%   r&   __init__-   s$   




zHTMLFormatter.__init__strc                 C  s2   |   }tdd |D rdd |D }d|S )Nc                 s  s    | ]}t |tV  qd S N)r    r3   r"   xr%   r%   r&   	<genexpr>N   s    z*HTMLFormatter.to_string.<locals>.<genexpr>c                 S  s   g | ]}t |qS r%   )r3   r5   r%   r%   r&   
<listcomp>O   s    z+HTMLFormatter.to_string.<locals>.<listcomp>
)renderanyjoin)r1   linesr%   r%   r&   	to_stringL   s   
zHTMLFormatter.to_string	list[str]c              	   C  sH   |    | jr!td}| dt| j d| dt| jj d | jS )N   z<p>z rows  z columns</p>)_write_tableshould_show_dimensionschrwritelenr)   r+   r,   )r1   Zbyr%   r%   r&   r:   R   s   $zHTMLFormatter.renderc                 C     | j jS r4   )r(   rC   r1   r%   r%   r&   rC   ]      z$HTMLFormatter.should_show_dimensionsc                 C  rG   r4   )r(   show_row_idx_namesrH   r%   r%   r&   rJ   a   rI   z HTMLFormatter.show_row_idx_namesc                 C  rG   r4   )r(   show_col_idx_namesrH   r%   r%   r&   rK   e   rI   z HTMLFormatter.show_col_idx_namesr!   c                 C  s    | j jr	| jjjS | jrdS dS )N   r   )r(   indexr)   nlevelsrK   rH   r%   r%   r&   
row_levelsi   s
   
zHTMLFormatter.row_levelsr   c                 C  s   | j S r4   )r+   rH   r%   r%   r&   _get_columns_formatted_valuesx   s   z+HTMLFormatter._get_columns_formatted_valuesc                 C  rG   r4   )r(   is_truncatedrH   r%   r%   r&   rQ   {   rI   zHTMLFormatter.is_truncatedc                 C  s   t | jjjS r4   )rF   r(   r*   r+   rH   r%   r%   r&   ncols   s   zHTMLFormatter.ncolsr   sr   indentc                 C  s    t |}| jd| |  d S )NrA   )r   r,   append)r1   rS   rT   rsr%   r%   r&   rE      s   zHTMLFormatter.writeheadertagsc                 C  sH   | j |d}|r|dur|pd}|d| d7 }| j|d||d dS )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 zstyle="min-width: z;"thkindrT   rX   )r/   get_write_cell)r1   rS   rW   rT   rX   r/   r%   r%   r&   write_th   s
   zHTMLFormatter.write_thc                 C  s   | j |d||d d S )Ntdr[   )r^   )r1   rS   rT   rX   r%   r%   r&   write_td   s   zHTMLFormatter.write_tdr`   r\   c           
      C  s   |d urd| d| d}nd| d}| j rdddd}ni }t||d }| jrBt|rBt|i d }|d	| d
7 }d}	nd}	| | | |	 d| d| d S )N<rA   >z&amp;z&lt;z&gt;)&rb   rc   )Zescape_charsz	<a href="z" target="_blank">z</a>rY   z</)r.   r   stripr   r   rE   )
r1   rS   r\   rT   rX   Z	start_tagescrV   Zrs_unescapedZend_ar%   r%   r&   r^      s   $zHTMLFormatter._write_celllinealigndict[int, str] | Nonenindex_levelsc                 C  s   |d u ri }|d u r|  d| n
|  d| d| ||7 }t|D ]%\}}	||d }
|s6| jr@||k r@| j|	|||
d q#| j|	||
d q#||8 }|  d| d S )Nz<tr>z<tr style="text-align: z;">)rT   rW   rX   )rX   z</tr>)rE   	enumerater]   r-   r_   ra   )r1   rg   rT   r   rW   rh   rX   rj   irS   Zval_tagr%   r%   r&   write_tr   s   
zHTMLFormatter.write_trc              	   C  s  dg}t d}|s|d | jd ur7t| jtr| j | _t| jttfs1tdt	| j |
| j | jd u r?d}nd| j d}| jd u rNd}nd| j d}| d	| d
d| d| d| | jjso| jrw| || j  | || j  | d| d S )NZ	dataframezdisplay.html.use_mathjaxZtex2jax_ignorez.classes must be a string, list, or tuple, not rY   z id=""z	 border="z<tablez class="rA   rc   z</table>)r
   rU   r   r    r3   splitlisttuple	TypeErrortypeextendr   r   rE   r<   r(   rW   rJ   _write_headerr   _write_body)r1   rT   Z_classesZuse_mathjaxZ
id_sectionZborder_attrr%   r%   r&   rB      s8   



zHTMLFormatter._write_tablec                 C  sJ  | j j}t| jtrWd}| j jrtj}nd}| jj|ddd}t	||}t
|d }tt||D ]!\}\}	}
|r| j j}| j jri }t|	 D ]^\}}||krZ|||d < n7|| |kr|d ||< ||kry|
d | d |
|d   }
n|
d | |
|d  f |
|d   }
n|||< || |krd||< |
d | d |
|d   }
qK|}	t
|d }||krd|	|< n1i }t|	 D ]\}}||kr|||d < q|||< qd||< |}	|
d | dg |
|d   }
dg| jd  }| j js| jr| j jr| jj| }|t|pd n|d i }t
|}t|
D ]'\}}||	v r<|	| dkr;|j|	| d||< nq!|d7 }|| q!| j||| j|d	d
 q2d S dg| jd  }| j jsh| jr}| j jrx|| jjpud n|d ||   | j j}|r| j| j j }||d | j||| jd	|d d S )Nz colspan="{span:d}" halign="left"FsparsifyZadjoinnamesrL   )...rz   rY   spanT)rX   rW   )rW   rh   )r(   is_truncated_horizontallyr    r+   r   rx   r   
no_defaultformatr   rF   rk   zip
tr_col_numrp   r0   rO   rM   rK   Zshow_index_namesry   rU   r   rm   r   namert   rP   justifyinsert)r1   rT   r}   templatesentinellevelslevel_lengths	inner_lvllnumrecordsvaluesZins_colZrecs_newtagr|   rowr   rX   jrl   vrh   r%   r%   r&   _write_col_header  s   






T

zHTMLFormatter._write_col_headerc                 C  sL   | j j}dd | jjjD dg| j|rdnd   }| j||| jdd d S )Nc                 S  s   g | ]
}|d ur
|ndqS )NrY   r%   r5   r%   r%   r&   r8     s    z3HTMLFormatter._write_row_header.<locals>.<listcomp>rY   rL   r   T)rW   )r(   r}   r)   rM   ry   rR   rm   r   )r1   rT   r}   r   r%   r%   r&   _write_row_header}  s
   zHTMLFormatter._write_row_headerc                 C  sJ   |  d| | jjr| || j  | jr| || j  |  d| d S )Nz<thead>z</thead>)rE   r(   rW   r   r   rJ   r   )r1   rT   r%   r%   r&   ru     s   zHTMLFormatter._write_headerdict[int, list[str]]c                   sH   t dd   fddt jD }W d    |S 1 sw   Y  |S )Nzdisplay.max_colwidthc                      i | ]	}| j |qS r%   r(   Z
format_colr"   rl   rH   r%   r&   r'         z7HTMLFormatter._get_formatted_values.<locals>.<dictcomp>)r   rangerR   )r1   
fmt_valuesr%   rH   r&   _get_formatted_values  s   
z#HTMLFormatter._get_formatted_valuesc                 C  s`   |  d| |  }| jjrt| jjtr| ||| j  n	| 	||| j  |  d| d S )Nz<tbody>z</tbody>)
rE   r   r(   rM   r    r)   r   _write_hierarchical_rowsr   _write_regular_rows)r1   rT   r   r%   r%   r&   rv     s   zHTMLFormatter._write_bodyr   Mapping[int, list[str]]c                   s&  | j j}| j j}t| j j}| j jr,| j d}|d ur%| j jj|}n| j jj }g }t	|D ]^|rO| j j
krOdgt| }	| j|	|| jd | jd g }| j jr]||  n| jre|d | fddt	| jD  |r| j j| j }
||
d | j||| jd | jd q2d S )N	__index__rz   rX   rj   rY   c                 3      | ]	} |  V  qd S r4   r%   r"   r   r   rl   r%   r&   r7         z4HTMLFormatter._write_regular_rows.<locals>.<genexpr>)r(   r}   is_truncated_verticallyrF   r*   rM   _get_formattermapr   r   
tr_row_numrm   r   rO   rU   rK   rt   rR   r   r   )r1   r   rT   r}   r   nrowsr(   Zindex_valuesr   str_sep_rowZ
dot_col_ixr%   r   r&   r     s@   
 z!HTMLFormatter._write_regular_rowsc              	     s  d}| j j}| j j}| j j}t|}t|jtsJ |jjdddd}t	t
| }| j jrhtj}	|jj|	ddd}
t|
|	}t|d }|r| j j}d}t|D ]\}}i }t	| D ]{\}}||krl|||d < n<|| |kr|d ||< |st	||d  }d|d< ||t| d}nt	|| }d||| < t|||< n|||< || |krd||< |dkr||tdgt|  q]|rt	|| }d||| < t|||< q]|||< qQd|| |<  D ]
} | |d q|d7 }t|D ]lg }i }d}d}t
|| D ]+\}}|v r%| dkr$|j| d	||< n|d7 }q
|d7 }|| q
| fd
dt| jD  |rV|| j| | j j d | j||| j|t|
| d qd S g }tt|D ]a|r| j jkrdgt| }| j||| jd | jd t	t
|jjdddd }g }||  | fddt| jD  |r|| j| j j d | j||| jd |jjd qpd S )Nzrowspan="{span}" valign="top"Frw   rL   rz   Tr   r{   c                 3  r   r4   r%   r   r   r%   r&   r7   !  r   z9HTMLFormatter._write_hierarchical_rows.<locals>.<genexpr>r   c                 3  r   r4   r%   r   r   r%   r&   r7   ?  r   )r(   r}   r   r*   rF   r    rM   r   r   rp   r   rx   r   r~   r   r   rk   r0   r   rq   r   rU   rt   rR   rO   r   rm   r   rN   )r1   r   rT   r   r}   r   r)   r   Z
idx_valuesr   r   r   r   Zins_rowZinsertedr   r   Zrec_newr   r|   Zdot_rowZix_colr   rX   Zsparse_offsetr   r   r   r%   r   r&   r     s   



 
 
z&HTMLFormatter._write_hierarchical_rows)NNNF)r   r   r   r   r   r   r   r   r   r   r   r   )r   r3   r   r?   )r   r   )r   r!   )r   r   )r   )rS   r   rT   r!   r   r   )Fr   N)
rS   r   rW   r   rT   r!   rX   r   r   r   )r   N)rS   r   rT   r!   rX   r   r   r   )r`   r   N)
rS   r   r\   r3   rT   r!   rX   r   r   r   )r   r   FNNr   )rg   r   rT   r!   r   r!   rW   r   rh   r   rX   ri   rj   r!   r   r   )rT   r!   r   r   r   r   )r   r   rT   r!   r   r   )__name__
__module____qualname____doc__r   __annotations__r2   r>   r:   propertyrC   rJ   rK   rO   rP   rQ   rR   rE   r_   ra   r^   rm   rB   r   r   ru   r   rv   r   r   r%   r%   r%   r&   r   !   sX   
 	


!
%
w



-r   c                      s@   e Zd ZdZdddZdddZdd
dZd fd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   r   c                   s    fddt  jD S )Nc                   r   r%   r   r   rH   r%   r&   r'   S  r   z;NotebookFormatter._get_formatted_values.<locals>.<dictcomp>)r   rR   rH   r%   rH   r&   r   R  s   z'NotebookFormatter._get_formatted_valuesr?   c                 C  s
   | j  S r4   )r+   r   rH   r%   r%   r&   rP   U  s   
z/NotebookFormatter._get_columns_formatted_valuesr   c                   s   d}d}d ddg}t | jtr|d | jr|d n|d d	t fd
d|}td|||f}| | d S )Nz            <style scoped>z            </style>zM                .dataframe %s {
                    %s: %s;
                })ztbody tr th:only-of-typevertical-alignmiddle)ztbody tr thr   top)zthead tr th
text-alignleft)zthead tr:last-of-type thr   right)zthead thr   r   z

c                   s    |  S r4   r%   )tZtemplate_selectr%   r&   <lambda>p  s    z/NotebookFormatter.write_style.<locals>.<lambda>r9   )	r    r+   r   rU   rJ   r<   r   r   rE   )r1   Ztemplate_firstZtemplate_lastZelement_propsZtemplate_midr   r%   r   r&   write_styleX  s"   

zNotebookFormatter.write_stylec                   s,   |  d |   t   |  d | jS )Nz<div>z</div>)rE   r   superr:   r,   rH   	__class__r%   r&   r:   t  s
   


zNotebookFormatter.renderr   r   )r   r   )	r   r   r   r   r   rP   r   r:   __classcell__r%   r%   r   r&   r   K  s    


r   N)r   
__future__r   textwrapr   typingr   r   r   r   r   r	   Zpandas._configr
   Zpandas._libsr   Zpandasr   r   Zpandas.io.commonr   Zpandas.io.formats.formatr   r   Zpandas.io.formats.printingr   r   r   r%   r%   r%   r&   <module>   s      	    .