from argparse import ArgumentParser
import os.path
from bs4 import BeautifulSoup
+from bs4.element import Tag
import requests
import requests_cache
from xdg_base_dirs import xdg_cache_home
+class Post:
+ def __init__(self, html: BeautifulSoup) -> None:
+ self._html = html
+
+ def text(self) -> Tag:
+ body = self._html.body
+ assert body
+ text = body.find_next("div", class_="post-post")
+ assert isinstance(text, Tag)
+ return text
+
+
def command_line_parser() -> ArgumentParser:
parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')
parser.add_argument(
def main() -> None:
args = command_line_parser().parse_args()
with requests_cache.CachedSession(args.cache_path, cache_control=True) as session:
- fetch(args.url, session, args.timeout)
+ html = fetch(args.url, session, args.timeout)
+ Post(html)
if __name__ == '__main__':
return 200
def do_GET(self) -> None:
- body = b'<html><body>This is glowfic</body></html>'
+ body = b'''<html>
+ <body>
+ <div class="post-container post-post">
+ This is glowfic
+ </div>
+ </body>
+</html>'''
self.send_response(self._response_code())
self.send_header("Content-type", "text/html")
self.send_header("Content-Length", str(len(body)))
def testFetchConents(self) -> None:
with requests.session() as s:
- doc = paperdoorknob.fetch(
- f"http://localhost:{self._port()}", s, TIMEOUT)
- body = doc.body
- assert body
- self.assertEqual(body.text, "This is glowfic")
+ post = paperdoorknob.Post(paperdoorknob.fetch(
+ f"http://localhost:{self._port()}", s, TIMEOUT))
+ self.assertEqual(post.text().text.strip(), "This is glowfic")
def testFetchErrors(self) -> None:
with requests.session() as s: