U
    ú±d«  ã                   @   sn   d Z ddlmZ ddlmZ ddlm  mZ ddlZdZ	dZ
dZG dd„ deƒZG d	d
„ d
eƒZdd„ ZdS )a‹  
Tables Extension for Python-Markdown
====================================

Added parsing of tables to Python-Markdown.

See <https://Python-Markdown.github.io/extensions/tables>
for documentation.

Original code Copyright 2009 [Waylan Limberg](http://achinghead.com)

All changes Copyright 2008-2014 The Python Markdown Project

License: [BSD](https://opensource.org/licenses/bsd-license.php)

é   )Ú	Extensioné   )ÚBlockProcessoré    Nc                       sd   e Zd ZdZe d¡Ze d¡Z‡ fdd„Zdd„ Z	dd	„ Z
d
d„ Zdd„ Zdd„ Zdd„ Z‡  ZS )ÚTableProcessorz Process Tables. z"(?:(\\\\)|(\\`+)|(`+)|(\\\|)|(\|))z(?<!\\)(?:\\\\)*\|$c                    s"   d| _ d| _|| _tƒ  |¡ d S )NFÚ )ÚborderÚ	separatorÚconfigÚsuperÚ__init__)ÚselfÚparserr
   ©Ú	__class__© ú>/tmp/pip-unpacked-wheel-muih9_xp/markdown/extensions/tables.pyr   !   s    zTableProcessor.__init__c           	      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   ÚparentÚblockZis_tableÚrowsZheader0r   Zrow0_lenÚindexr   r   r   Útest(   s2    

$zTableProcessor.testc                 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&   Úblocksr'   Úheaderr(   ÚalignÚcr0   r1   r2   r   r   r   r   ÚrunL   s*    



zTableProcessor.runc                 C   s2   t  |d¡}t|ƒ}|r.t  |d¡ |d8 }qdS )zBuild an empty row.ÚtrÚtdr   N)r6   r7   r   )r   r&   r<   r?   Úcountr   r   r   r9   k   s
    zTableProcessor._build_empty_rowc           
   	   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. r?   r@   r1   Úthr   r   Úuse_align_attributer<   Ústyleztext-align: ú;N)
r6   r7   Útagr"   Ú	enumerater   ÚtextÚ
IndexErrorr
   r$   )
r   r   r&   r<   r?   rF   ÚcellsÚiÚar=   r   r   r   r8   s   s    


zTableProcessor._build_rowc                 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   r+   r   é   NFT)	ÚRE_CODE_PIPESÚfinditerÚgroupr5   r   ÚstartÚendÚ
ValueErrorr)   )r   r   ÚelementsZpipesZticsZ
tic_pointsZ
tic_regionZ
good_pipesÚmÚposZtic_lenZtic_sizer)   ÚpipeZ	throw_outZregionr   r   r   rN      sV    
"
"


"$
zTableProcessor._split)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚreÚcompilerP   r   r   r*   r>   r9   r8   r"   rN   Ú__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 )NrC   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 )z3 Add an instance of TableProcessor to BlockParser. r   r0   éK   N)ZESCAPED_CHARSr5   r   r   Z
getConfigsÚblockprocessorsÚregister)r   ZmdÚ	processorr   r   r   ÚextendMarkdownã   s    
zTableExtension.extendMarkdown)rZ   r[   r\   r]   r   rg   r`   r   r   r   r   ra   Ù   s   ra   c                  K   s
   t f | ŽS )N)ra   )rb   r   r   r   ÚmakeExtensionë   s    rh   )r]   r   r   rd   r   Úxml.etree.ElementTreer6   ÚElementTreer^   r   r   r!   r   ra   rh   r   r   r   r   Ú<module>   s    ?