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
  11
  12 def command_line_parser() -> ArgumentParser:
  13     parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')
  14     parser.add_argument(
  15         '--timeout',
  16         help='How long to wait for HTTP requests, in seconds',
  17         default=30)
  18     parser.add_argument('url', help='URL to retrieve')
  19     return parser
  20
  21
  22 def fetch(url: str, timeout: int) -> None:
  23     with requests.session() as s:
  24         with s.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     fetch(args.url, args.timeout)
  31
  32
  33 if __name__ == '__main__':
  34     main()