o
    ˷e
                     @   s   d dl m  mZ d dlmZ d dlmZmZ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
edefddZe ZdS )    N)	Formatter)AnyListTupleUnion   )FormattedTextStyleAndTextTuplesHTMLc                   @   sf   e Zd ZdZdeddfddZdefddZdefd	d
Zde	de	dd fddZ
de	dd f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``.
    valuereturnNc                    st   || _ td| d}g g g g  dtf fdddtdd f fdd| t| _d S )	Nz<html-root>z</html-root>r   c                     sR   g } r|  dd  r|  dd    r$|  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 Y/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/formatted_text/html.pyget_current_style(   s   
z(HTML.__init__.<locals>.get_current_stylenodec                    s   | j D ]s}|j|jkr |jf q|jdv}d }}|j D ]\}}|dkr-|}|dkr3|}|dkr9|}q#d|v rBtdd|v rJtd|rR|j |rY| |r` | | |rj	  |rp	  |rv 	  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)

childNodesnodeType	TEXT_NODEr   datanodeName
attributesitems
ValueErrorpop)r   childadd_to_name_stackr   r   kvr   r   r   r   process_noderesultr   r   r-   4   s@   



z#HTML.__init__.<locals>.process_node)r   minidomparseStringstrr   r   formatted_text)selfr   documentr   r,   r   __init__   s    ,zHTML.__init__c                 C   s   d| j dS )NzHTML())r   r3   r   r   r   __repr__d      zHTML.__repr__c                 C   s   | j S N)r2   r7   r   r   r   __pt_formatted_text__g   s   zHTML.__pt_formatted_text__argskwargsc                 O   s   t t| j||S )zc
        Like `str.format`, but make sure that the arguments are properly
        escaped.
        )r
   	FORMATTERvformatr   )r3   r<   r=   r   r   r   formatj   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 r:   )html_escape).0ir   r   r   	<genexpr>x   s    zHTML.__mod__.<locals>.<genexpr>)
isinstancetupler
   r   )r3   r   r   r   r   __mod__q   s   
zHTML.__mod__)__name__
__module____qualname____doc__r1   r5   r8   r	   r;   objectr@   rG   r   r   r   r   r
   
   s    Ec                   @   s"   e Zd ZdededefddZdS )HTMLFormatterr   format_specr   c                 C   s   t t||S r:   )rA   r@   )r3   r   rN   r   r   r   format_field}   r9   zHTMLFormatter.format_fieldN)rH   rI   rJ   rL   r1   rO   r   r   r   r   rM   |   s    rM   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;)rE   r1   replace)rP   r   r   r   rA      s   

rA   )xml.dom.minidomdomr/   stringr   typingr   r   r   r   baser   r	   __all__r
   rM   rL   r1   rA   r>   r   r   r   r   <module>   s    r
