Strip edit-boxes and footers

[paperdoorknob] / paperdoorknob.py

diff --git a/paperdoorknob.py b/paperdoorknob.py
index 0c6ec856b50377f1b895a2dfe2d9681badf93dba..fb2df1f1cc1c7443bc6e07018bad7148eb12a5c6 100644 (file)
--- a/paperdoorknob.py
+++ b/paperdoorknob.py
@@ -6,6 +6,7 @@
 
 
 from argparse import ArgumentParser
+import itertools
 import os.path
 
 from typing import Iterable
@@ -20,6 +21,10 @@ from xdg_base_dirs import xdg_cache_home
 class Post:
     def __init__(self, html: BeautifulSoup) -> None:
         self._html = html
+        for eb in self._html.find_all("div", class_="post-edit-box"):
+            eb.decompose()
+        for footer in self._html.find_all("div", class_="post-footer"):
+            footer.decompose()
 
     def text(self) -> Tag:
         body = self._html.body
@@ -33,6 +38,9 @@ class Post:
         assert all(isinstance(r, Tag) for r in replies)
         return replies
 
+    def entries(self) -> Iterable[Tag]:
+        return itertools.chain([self.text()], self.replies())
+
 
 def command_line_parser() -> ArgumentParser:
     parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')