util.py

import functools

from bs4 import Tag

# Unicode LINE SEPARATOR character
LINE_SEPARATOR = "\u2028"
# Tags within which we should not be replacing content
VERBATIM_TAGS = ("pre", "code")
# Tag within which we should not perform text substitutions
LINK_TAG = "link"


def keep_verbatim(tag: Tag) -> bool:
    return tag.name in VERBATIM_TAGS or any(
        filter(lambda t: t.name in VERBATIM_TAGS, tag.parents)
    )


def is_link_component(tag: Tag) -> bool:
    return tag.name == LINK_TAG or any(
        filter(lambda t: t.name == LINK_TAG, tag.parents)
    )


__html_escape_lut = str.maketrans(
    {"&": "&amp;", "<": "&lt;", ">": "&gt;", '"': "&quot;", "'": "&#39;"}
)


@functools.lru_cache()
def html_escape(value):
    return value.translate(__html_escape_lut)