U
    d                     @  s   d dl mZ d dlm  mZ d dlmZ d dlmZm	Z	 ddl
mZmZ dgZG dd dZG d	d
 d
eZdddddZe ZdS )    )annotationsN)	Formatter)AnyList   )FormattedTextStyleAndTextTuplesHTMLc                   @  s^   e Zd ZdZdddddZdddd	Zd
dddZddd dddZdd dddZdS )r	   a  
    HTML formatted text.
    Take something HTML-like, for use as a formatted string.

    ::

        # Turn something into red.
        HTML('<style fg="ansired" bg="#00ff44">...</style>')

        # Italic, bold, underline and strike.
        HTML('<i>...</i>')
        HTML('<b>...</b>')
        HTML('<u>...</u>')
        HTML('<s>...</s>')

    All HTML elements become available as a "class" in the style sheet.
    E.g. ``<username>...</username>`` can be styled, by setting a style for
    ``username``.
    strNone)valuereturnc                   sr   || _ td| d}g g g g  dd fddddd	 fd
d| t| _d S )Nz<html-root>z</html-root>r
   r   c                    sR   g } r|  dd  r2|  dd    rH|  d d   d| S )z$Build style string for current node.zclass:,zfg:zbg: )appendjoin)parts)bg_stackfg_stack
name_stack F/tmp/pip-unpacked-wheel-4x_7prb2/prompt_toolkit/formatted_text/html.pyget_current_style*   s    z(HTML.__init__.<locals>.get_current_styler   r   )noder   c                   s   | j D ]}|j|jkr* |jf q|jdk}d }}|j D ],\}}|dkrZ|}|dkrf|}|dkrF|}qFd|krtdd|krtd|r|j |r| |r | | |rԈ	  |r	  |r 	  qd	S )
zProcess node recursively.)z	#documentz	html-rootstyle fgbgcolorr   z "fg" attribute contains a space.z "bg" attribute contains a space.N)

childNodesZnodeTypeZ	TEXT_NODEr   datanodeName
attributesitems
ValueErrorpop)r   childZadd_to_name_stackr   r   kvr   r   r   r   process_noderesultr   r   r,   6   s:    



z#HTML.__init__.<locals>.process_node)r   minidomparseStringr   formatted_text)selfr   documentr   r+   r   __init__!   s    ,zHTML.__init__r   c                 C  s   d| j dS )NzHTML())r   r1   r   r   r   __repr__f   s    zHTML.__repr__r   c                 C  s   | j S N)r0   r5   r   r   r   __pt_formatted_text__i   s    zHTML.__pt_formatted_text__object)argskwargsr   c                 O  s   t t| j||S )zc
        Like `str.format`, but make sure that the arguments are properly
        escaped.
        )r	   	FORMATTERvformatr   )r1   r:   r;   r   r   r   formatl   s    zHTML.formatc                 C  s0   t |ts|f}tdd |D }t| j| S )z+
        HTML('<b>%s</b>') % value
        c                 s  s   | ]}t |V  qd S r7   )html_escape).0ir   r   r   	<genexpr>z   s     zHTML.__mod__.<locals>.<genexpr>)
isinstancetupler	   r   )r1   r   r   r   r   __mod__s   s    
zHTML.__mod__N)	__name__
__module____qualname____doc__r3   r6   r8   r>   rE   r   r   r   r   r	      s   Ec                   @  s   e Zd ZddddddZdS )HTMLFormatterr9   r
   )r   format_specr   c                 C  s   t t||S r7   )r?   r>   )r1   r   rK   r   r   r   format_field   s    zHTMLFormatter.format_fieldN)rF   rG   rH   rL   r   r   r   r   rJ   ~   s   rJ   r9   r
   )textr   c                 C  s4   t | ts|  } | ddddddddS )	N&z&amp;<z&lt;>z&gt;"z&quot;)rC   r
   replace)rM   r   r   r   r?      s    
   r?   )
__future__r   Zxml.dom.minidomdomr.   stringr   typingr   r   baser   r   __all__r	   rJ   r?   r<   r   r   r   r   <module>   s   r