mirror of
https://github.com/quay/quay.git
synced 2025-04-18 10:44:06 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
69 lines
1.4 KiB
Python
69 lines
1.4 KiB
Python
from bs4 import BeautifulSoup, NavigableString, Tag
|
|
|
|
_NEWLINE_INDICATOR = "<<<newline>>>"
|
|
|
|
|
|
def _bold(elem):
|
|
elem.replace_with("*%s*" % elem.text)
|
|
|
|
|
|
def _unordered_list(elem):
|
|
constructed = ""
|
|
for child in elem.children:
|
|
if child.name == "li":
|
|
constructed += "* %s\n" % child.text
|
|
elem.replace_with(constructed)
|
|
|
|
|
|
def _horizontal_rule(elem):
|
|
elem.replace_with("%s\n" % ("-" * 80))
|
|
|
|
|
|
def _anchor(elem):
|
|
elem.replace_with("[%s](%s)" % (elem.text, elem["href"]))
|
|
|
|
|
|
def _table(elem):
|
|
elem.replace_with("%s%s" % (elem.text, _NEWLINE_INDICATOR))
|
|
|
|
|
|
_ELEMENT_REPLACER = {
|
|
"b": _bold,
|
|
"strong": _bold,
|
|
"ul": _unordered_list,
|
|
"hr": _horizontal_rule,
|
|
"a": _anchor,
|
|
"table": _table,
|
|
}
|
|
|
|
|
|
def _collapse_whitespace(text):
|
|
new_lines = []
|
|
lines = text.split("\n")
|
|
for line in lines:
|
|
if not line.strip():
|
|
continue
|
|
|
|
new_lines.append(line.strip().replace(_NEWLINE_INDICATOR, "\n"))
|
|
|
|
return "\n".join(new_lines)
|
|
|
|
|
|
def html2text(html):
|
|
soup = BeautifulSoup(html, "html5lib")
|
|
_html2text(soup)
|
|
return _collapse_whitespace(soup.text)
|
|
|
|
|
|
def _html2text(elem):
|
|
for child in elem.children:
|
|
if isinstance(child, Tag):
|
|
_html2text(child)
|
|
elif isinstance(child, NavigableString):
|
|
# No changes necessary
|
|
continue
|
|
|
|
if elem.parent:
|
|
if elem.name in _ELEMENT_REPLACER:
|
|
_ELEMENT_REPLACER[elem.name](elem)
|