git.scottworley.com Git - paperdoorknob/blob - paperdoorknob.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 argparse import ArgumentParser
   9 import requests
  10 import requests_cache
  11
  12
  13 def command_line_parser() -> ArgumentParser:
  14     parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')
  15     parser.add_argument(
  16         '--timeout',
  17         help='How long to wait for HTTP requests, in seconds',
  18         default=30)
  19     parser.add_argument('url', help='URL to retrieve')
  20     return parser
  21
  22
  23 def fetch(url: str, session: requests.Session, timeout: int) -> None:
  24     with session.get(url, timeout=timeout) as r:
  25         r.raise_for_status()
  26
  27
  28 def main() -> None:
  29     args = command_line_parser().parse_args()
  30     with requests_cache.CachedSession() as session:
  31         fetch(args.url, session, args.timeout)
  32
  33
  34 if __name__ == '__main__':
  35     main()