]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob.py
fetch: Cache
[paperdoorknob] / 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()