U
    \> e%"  ã                   @  sž   d Z ddlmZ ddlmZ ddlmZ ddlm  m	Z ddl
Z
ddlmZmZmZ erfdd	lmZ dZdZdZG d
d„ deƒZG dd„ deƒZdd„ ZdS )zˆ
Added parsing of tables to Python-Markdown.

See the [documentation](https://Python-Markdown.github.io/extensions/tables)
for details.
é    )Úannotationsé   )Ú	Extensioné   )ÚBlockProcessorN)ÚTYPE_CHECKINGÚAnyÚSequence)Úblockparserc                      s¦   e Zd ZdZe d¡Ze d¡Zdddœ‡ fdd„Zd	d
ddœdd„Z	d	dddœdd„Z
d	dddœdd„Zd
d	dddœdd„Zd
ddœdd„Zd
ddœdd„Z‡  ZS ) ÚTableProcessorz Process Tables. z"(?:(\\\\)|(\\`+)|(`+)|(\\\|)|(\|))z(?<!\\)(?:\\\\)*\|$zblockparser.BlockParserzdict[str, Any])ÚparserÚconfigc                   s"   d| _ d| _|| _tƒ  |¡ d S )NFÚ )ÚborderÚ	separatorr   ÚsuperÚ__init__)Úselfr   r   ©Ú	__class__© ú>/tmp/pip-unpacked-wheel-wj0og6ym/markdown/extensions/tables.pyr   ,   s    zTableProcessor.__init__zetree.ElementÚstrÚbool)ÚparentÚblockÚreturnc           	      C  s   d}dd„ |  d¡D ƒ}t|ƒdkr|d }t| _| d¡rL|  jtO  _| j |¡dk	rj|  jtO  _|  	|¡}t|ƒ}|dk}|sØ|dkrØ| jrØt
dt|ƒƒD ]2}||  d¡}|sÎ| j || ¡dk	}|s¤ qØq¤|r|  	|d ¡}t|ƒ|kotd	 |¡ƒtd
ƒk}|r|| _|S )z©
        Ensure first two rows (column header and separator row) are valid table rows.

        Keep border check and separator row do avoid repeating the work.
        Fc                 S  s   g | ]}|  d ¡‘qS )ú )Ústrip)Ú.0Úrowr   r   r   Ú
<listcomp>:   s     z'TableProcessor.test.<locals>.<listcomp>Ú
r   r   ú|Nr   z|:- )ÚsplitÚlenÚ	PIPE_NONEr   Ú
startswithÚ	PIPE_LEFTÚRE_END_BORDERÚsearchÚ
PIPE_RIGHTÚ
_split_rowÚrangeÚsetÚjoinr   )	r   r   r   Zis_tableÚrowsZheader0r    Zrow0_lenÚindexr   r   r   Útest3   s2    

$zTableProcessor.testz	list[str]ÚNone)r   Úblocksr   c                 C  s  |  d¡ d¡}|d  d¡}t|ƒdk r.g n
|dd… }g }| jD ]d}| d¡}| d¡rr| d¡rr| d¡ qD| d¡rˆ| d	¡ qD| d¡rž| d
¡ qD| d¡ qDt 	|d¡}t 	|d¡}	|  
||	|¡ t 	|d¡}
t|ƒdkrö|  |
|¡ n|D ]}|  
| d¡|
|¡ qúdS )z& Parse a table block and build table. r   r"   r   é   r   Nú:ÚcenterÚleftÚrightÚtableÚtheadÚtbody)Úpopr$   r   r%   r   r'   ÚendswithÚappendÚetreeÚ
SubElementÚ
_build_rowÚ_build_empty_row)r   r   r4   r   Úheaderr0   ÚalignÚcr:   r;   r<   r    r   r   r   ÚrunW   s*    



zTableProcessor.runzSequence[str | None])r   rE   r   c                 C  s2   t  |d¡}t|ƒ}|r.t  |d¡ |d8 }qdS )zBuild an empty row.ÚtrÚtdr   N)r@   rA   r%   )r   r   rE   rH   Úcountr   r   r   rC   v   s
    zTableProcessor._build_empty_row)r    r   rE   r   c           
   	   C  sª   t  |d¡}d}|jdkrd}|  |¡}t|ƒD ]t\}}t  ||¡}	z||  d¡|	_W n tk
rr   d|	_Y nX |r0| jd r|	 	d|¡ q0|	 	d	d
|› d¡ q0dS )z) Given a row of text, build table cells. rH   rI   r;   Úthr   r   Úuse_align_attributerE   Ústyleztext-align: ú;N)
r@   rA   Útagr,   Ú	enumerater   ÚtextÚ
IndexErrorr   r.   )
r   r    r   rE   rH   rO   ÚcellsÚiÚarF   r   r   r   rB   ~   s    


zTableProcessor._build_row)r    r   c                 C  s4   | j r*| d¡r|dd… }| j d|¡}|  |¡S )z) split a row of text into list of cells. r#   r   Nr   )r   r'   r)   ÚsubÚ_split)r   r    r   r   r   r,   “   s
    
zTableProcessor._split_rowc                 C  s  g }g }g }g }g }g }| j  |¡D ]¢}| d¡rl| t| d¡ƒd ¡ | | d¡| d¡d df¡ q$| d¡r¬| t| d¡ƒ¡ | | d¡| d¡d df¡ q$| d¡r$| | d¡¡ q$d}	t|ƒ}
|	|
k rnzn||	 ||	 d  }|dkrt‚||	d d…  |¡d }| ||	 d ||	|  d f¡ |	|d 7 }	W qÔ tk
rj   |	d7 }	Y qÔX qÔ|D ]j}d}|D ]J}||d k r˜ qÊn.|d |  kr¸|d kr~n n
d} qÊq~|sr| |¡ qrd}	|D ] }| ||	|… ¡ |d }	qæ| ||	d… ¡ |S )	z: split a row of text with some code into a list of cells. r   r   r5   r   é   NFT)	ÚRE_CODE_PIPESÚfinditerÚgroupr?   r%   ÚstartÚendÚ
ValueErrorr1   )r   r    ÚelementsZpipesZticsZ
tic_pointsZ
tic_regionZ
good_pipesÚmÚposZtic_lenZtic_sizer1   ÚpipeZ	throw_outZregionr   r   r   rW   ›   sV    
"
"


"$
zTableProcessor._split)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚreÚcompilerY   r)   r   r2   rG   rC   rB   r,   rW   Ú__classcell__r   r   r   r   r   &   s   

$r   c                      s(   e Zd ZdZ‡ fdd„Zdd„ Z‡  ZS )ÚTableExtensionz Add tables to Markdown. c                   s    dddgi| _ tƒ jf |Ž d S )NrL   Fz-True to use align attribute instead of style.)r   r   r   )r   Úkwargsr   r   r   r   ç   s     ÿzTableExtension.__init__c                 C  s<   d|j kr|j  d¡ t|j|  ¡ ƒ}|jj |dd¡ dS )z7 Add an instance of `TableProcessor` to `BlockParser`. r#   r:   éK   N)ZESCAPED_CHARSr?   r   r   Z
getConfigsÚblockprocessorsÚregister)r   ZmdÚ	processorr   r   r   ÚextendMarkdownï   s    
zTableExtension.extendMarkdown)rc   rd   re   rf   r   rp   ri   r   r   r   r   rj   ä   s   rj   c                  K  s
   t f | ŽS )N)rj   )rk   r   r   r   ÚmakeExtension÷   s    rq   )rf   Ú
__future__r   r   r   rm   r   Úxml.etree.ElementTreer@   ÚElementTreerg   Útypingr   r   r	   r
   r&   r(   r+   r   rj   rq   r   r   r   r   Ú<module>   s    ?