]> git.scottworley.com Git - paperdoorknob/blob - domfilter.py
Apply bare-\emph fix to \st too
[paperdoorknob] / domfilter.py
1 # paperdoorknob: Print glowfic
2 #
3 # This program is free software: you can redistribute it and/or modify it
4 # under the terms of the GNU General Public License as published by the
5 # Free Software Foundation, version 3.
6
7
8 from typing import Any, Callable, List, Tuple
9
10 from bs4.element import Tag
11
12
13 def _changeTag(tag: Tag, new_name: str) -> Tag:
14 tag.name = new_name
15 return tag
16
17
18 DOMFilters: List[Tuple[str, Callable[[Tag], Any]]] = [
19 ("NoEdit", lambda x: [eb.decompose() for eb in x.find_all("div", class_="post-edit-box")]),
20 ("NoFooter", lambda x: [foot.decompose() for foot in x.find_all("div", class_="post-footer")]),
21 ("Strike", lambda x: [_changeTag(span, 's')
22 for span in x.find_all("span", style="text-decoration: line-through;")]),
23 ]
24
25
26 class DOMFilterError(Exception):
27 pass
28
29
30 def ApplyDOMFilters(filter_list: str, tag: Tag) -> None:
31 for filter_name in filter_list.split(','):
32 filters = [f for (name, f) in DOMFilters if name == filter_name]
33 if len(filters) == 0:
34 raise DOMFilterError(f"Unknown DOM filter: {filter_name}")
35 if len(filters) > 1:
36 raise DOMFilterError(
37 f"Multiple DOM filters with the same name!?: {filter_name}")
38 filters[0](tag)