mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2024-12-29 11:30:39 +00:00
Code Issues Projects Releases Wiki Activity

Python formatting

Browse source
This commit is contained in:
Mouse Reeve 2022-02-03 13:19:56 -08:00
parent 1f6ecc39ac
commit cae7191a2b
2 changed files with 3 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
bookwyrm/sanitize_html.py
View file

@ -22,9 +22,7 @@ class InputHtmlParser(HTMLParser): # pylint: disable=abstract-method
"ol",
"li",
]
self.allowed_attrs = [
"href", "rel", "src", "alt"
]
self.allowed_attrs = ["href", "rel", "src", "alt"]
self.tag_stack = []
self.output = []
# if the html appears invalid, we just won't allow any at all
@ -36,7 +34,7 @@ class InputHtmlParser(HTMLParser): # pylint: disable=abstract-method
allowed_attrs = " ".join(
f'{a}="{v}"' for a, v in attrs if a in self.allowed_attrs
)
reconstructed = f'<{tag}'
reconstructed = f"<{tag}"
if allowed_attrs:
reconstructed += " " + allowed_attrs
reconstructed += ">"

5
bookwyrm/tests/test_sanitize_html.py
View file

@ -37,10 +37,7 @@ class Sanitizer(TestCase):
parser = InputHtmlParser()
parser.feed(input_text)
output = parser.get_output()
self.assertEqual(
output,
'<a href="fish.com">yes </a> <i>html</i>'
)
self.assertEqual(output, '<a href="fish.com">yes </a> <i>html</i>')
def test_invalid_html(self):
"""remove all html when the html is malformed"""