X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/a0d30541ee9349b8c62200c9aa6d4dbb2873fccb..170e50a0efa18ed0f5f2ff644793469894983029:/paperdoorknob.py?ds=sidebyside

diff --git a/paperdoorknob.py b/paperdoorknob.py
index 0c6ec85..fb2df1f 100644
--- 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')