X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/6409066b6abbe205fa64f844a01516f3f3de2553..a0d30541ee9349b8c62200c9aa6d4dbb2873fccb:/paperdoorknob.py

diff --git a/paperdoorknob.py b/paperdoorknob.py
index b88c02d..0c6ec85 100644
--- a/paperdoorknob.py
+++ b/paperdoorknob.py
@@ -7,6 +7,9 @@
 
 from argparse import ArgumentParser
 import os.path
+
+from typing import Iterable
+
 from bs4 import BeautifulSoup
 from bs4.element import Tag
 import requests
@@ -25,6 +28,11 @@ class Post:
         assert isinstance(text, Tag)
         return text
 
+    def replies(self) -> Iterable[Tag]:
+        replies = self._html.find_all("div", class_="post-reply")
+        assert all(isinstance(r, Tag) for r in replies)
+        return replies
+
 
 def command_line_parser() -> ArgumentParser:
     parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')