@gkAjdZddlZddlmZmZmZmZddlmZddl m Z m Z m Z dZ d dZGdd eZy) zLConvert HTML with simple text formatting to text with ANSI escape sequences.N) HTMLParserStringIOname2codepointunichr)compact_empty_lines)ANSI_COLOR_CODES ANSI_RESET ansi_style) HTMLConverter html_to_ansic*t|}||S)a Convert HTML with simple text formatting to text with ANSI escape sequences. :param data: The HTML to convert (a string). :param callback: Optional callback to pass to :class:`HTMLConverter`. :returns: Text with ANSI escape sequences (a string). Please refer to the documentation of the :class:`HTMLConverter` class for details about the conversion process (like which tags are supported) and an example with a screenshot. )callback)r )datar converters P/var/www/openai/venv/lib/python3.12/site-packages/humanfriendly/terminal/html.pyr r sx0I T?ceZdZdZdZ dZdZedZdZ ddZ d Z d Z d Z d Zd ZdZdZdZdZdZdZy)r a Convert HTML with simple text formatting to text with ANSI escape sequences. The following text styles are supported: - Bold: ````, ```` and ```` - Italic: ````, ```` and ```` - Strike-through: ````, ```` and ```` - Underline: ````, ```` and ```` Colors can be specified as follows: - Foreground color: ```` - Background color: ```` Here's a small demonstration: .. code-block:: python from humanfriendly.text import dedent from humanfriendly.terminal import html_to_ansi print(html_to_ansi(dedent(''' Hello world! Is this thing on? I guess I can underline or strike-through text? And what about color? '''))) rainbow_colors = [ '#FF0000', '#E2571E', '#FF7F00', '#FFFF00', '#00FF00', '#96BF33', '#0000FF', '#4B0082', '#8B00FF', '#FFFFFF', ] html_rainbow = "".join('o' % c for c in rainbow_colors) print(html_to_ansi("Let's try a rainbow: %s" % html_rainbow)) Here's what the results look like: .. image:: images/html-to-ansi.png Some more details: - Nested tags are supported, within reasonable limits. - Text in ```` and ``
`` tags will be highlighted in a
      different color from the main text (currently this is yellow).

    - ``TEXT`` is converted to the format "TEXT (URL)" where
      the uppercase symbols are highlighted in light blue with an underline.

    - ``
``, ``
`` and ``
`` tags are considered block level tags
      and are wrapped in vertical whitespace to prevent their content from
      "running into" surrounding text. This may cause runs of multiple empty
      lines to be emitted. As a *workaround* the :func:`__call__()` method
      will automatically call :func:`.compact_empty_lines()` on the generated
      output before returning it to the caller. Of course this won't work
      when `output` is set to something like :data:`sys.stdout`.

    - ``
`` is converted to a single plain text line break.

    Implementation notes:

    - A list of dictionaries with style information is used as a stack where
      new styling can be pushed and a pop will restore the previous styling.
      When new styling is pushed, it is merged with (but overrides) the current
      styling.

    - If you're going to be converting a lot of HTML it might be useful from
      a performance standpoint to re-use an existing :class:`HTMLConverter`
      object for unrelated HTML fragments, in this case take a look at the
      :func:`__call__()` method (it makes this use case very easy).

    .. versionadded:: 4.15
       :class:`humanfriendly.terminal.HTMLConverter` was added to the
       `humanfriendly` package during the initial development of my new
       `chat-archive `_ project, whose
       command line interface makes for a great demonstration of the
       flexibility that this feature provides (hint: check out how the search
       keyword highlighting combines with the regular highlighting).
    )divpprec|jdd|_|jdd|_tj|g|i|y)a
        Initialize an :class:`HTMLConverter` object.

        :param callback: Optional keyword argument to specify a function that
                         will be called to process text fragments before they
                         are emitted on the output stream. Note that link text
                         and preformatted text fragments are not processed by
                         this callback.
        :param output: Optional keyword argument to redirect the output to the
                       given file-like object. If this is not given a new
                       :class:`~python3:io.StringIO` object is created.
        rNoutput)poprrr__init__)selfargskws   rrzHTMLConverter.__init__{s@z40
ffXt,D.4.2.rc|j|j||jt|jt
r#t
|jjSy)a
        Reset the parser, convert some HTML and get the text with ANSI escape sequences.

        :param data: The HTML to convert to text (a string).
        :returns: The converted text (only in case `output` is
                  a :class:`~python3:io.StringIO` object).
        N)resetfeedclose
isinstancerrrgetvaluerrs  r__call__zHTMLConverter.__call__sL	

		$

dkk8,&t{{';';'=>>-rc<|jr|jdSiS)z?Get the current style from the top of the stack (a dictionary).)stackrs r
current_stylezHTMLConverter.current_styles"&tzz"~33rct|jr&|jjtg|_tj|y)ab
        Close previously opened ANSI escape sequences.

        This method overrides the same method in the superclass to ensure that
        an :data:`.ANSI_RESET` code is emitted when parsing reaches the end of
        the input but a style is still active. This is intended to prevent
        malformed HTML from messing up terminal output.
        N)anyr(rwriter	rr!r)s rr!zHTMLConverter.closes6tzz?KKj)DJrNc|jjt||jn|}|r%|jjt	di|yy)a)
        Emit an ANSI escape sequence for the given or current style to the output stream.

        :param style: A dictionary with arguments for :func:`.ansi_style()` or
                      :data:`None`, in which case the style at the top of the
                      stack is emitted.
        N)rr-r	r*r
)rstyles  r
emit_stylezHTMLConverter.emit_stylesI	
*%&+m""KKj1512rc	|jjt|jdrt	|dddyt	|y)z
        Process a decimal or hexadecimal numeric character reference.

        :param value: The decimal or hexadecimal value (a string).
        xN)rr-r
startswithint)rvalues  rhandle_charrefzHTMLConverter.handle_charrefsD	
&u7G7G7LU12Y!3]^RUV[R\]^rc|jr||_n,|jr |jdk(r|j|}|jj|y)zZ
        Process textual data.

        :param data: The decoded text (a string).
        rN)link_url	link_textrpreformatted_text_levelrr-r$s  rhandle_datazHTMLConverter.handle_datasJ=="DN
]]t;;q@==&D$rc|dvrD|j}|jr|jjd|j}|dk(r|j|j|j
r|j
|n|j
||jjd|j
||jj|j|j
|j
||jjdn|j
||dvr|xjdzc_
||jvr|jjdy	y	)
zf
        Process the end of an HTML tag.

        :param tag: The name of the tag (a string).
        )abcodedelemiinsrsstrongspanur'r@z ())rBrr4

N)r*r(r
urls_matchr<r;r1rr-
render_urlr=
BLOCK_TAGS)rtag	old_style	new_styles    r
handle_endtagzHTMLConverter.handle_endtags``**Izz

r"**Icz??4>>4==AOOI.OOI.KK%%d+OOI.KK%%doodmm&DEOOI.KK%%c*	*o%,,1,$//!KKf%"rcZ|jjtt|y)z|
        Process a named character reference.

        :param name: The name of the character reference (a string).
        N)rr-rr)rnames  rhandle_entityrefzHTMLConverter.handle_entityrefs 	
&!567rc"||jvr|jjd|dk(r-|jdddt	d|Dd|_y*|dk(s|d	k(r|jd
y*|dk(r|jjdy*|d
k(s|dk(r(|jd|xjdz
c_y*|dk(s|dk(r|jdy*|dk(s|dk(r|jdy*|dk(s|dk(r|jdy*|dk(ri}t	d|Dd}|jdD]}|jd\}}}|j}|j}|dk(r|j||d <R|d!k(r|j||d!<l|d"k(r|d#k(rd|d#<||d$k(r|d%k(rd|d%<|d&k(r|d'k(rd|d(<|d&k(s|d)k(sd|d)<|jd+i|y*y*),z
        Process the start of an HTML tag.

        :param tag: The name of the tag (a string).
        :param attrs: A list of tuples with two strings each.
        rLr@blueT)colorbright	underlinec32K|]\}}|dk(s|yw)hrefNr/.0nvs   r	z0HTMLConverter.handle_starttag..s!E1f!
rArH)boldbr
rBryellow)rYr4rCrG)strike_throughrDrE)italicrFrJ)r[rIc32K|]\}}|dk(s|yw)r0Nr/r^s   rrbz0HTMLConverter.handle_starttag.. s``) when the link text is that same URL.
        )r)rr@rAs   rrMzHTMLConverter.urls_matchs%!!!$(:(:1(===rN)__name__
__module____qualname____doc__rOrr%propertyr*r!r1r9r>rSrVrzrrtrorNrrMr/rrrr%svOb%JE/&?44
3_ $&@81'f+!F#.&&>rrr)rr~humanfriendly.compatrrrrhumanfriendly.textrhumanfriendly.terminalrr	r
__all__rrr/rrrs<S
NM2KK,
 B>JB>r