]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob.py
b7e4349b98ab1d97ae2ff0ad6f50e4740bb6440f
[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 os.path
10 import requests
11 import requests_cache
12 from xdg_base_dirs import xdg_cache_home
13
14
15 def command_line_parser() -> ArgumentParser:
16 parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic')
17 parser.add_argument(
18 '--cache_path',
19 metavar='PATH',
20 help='Where to keep the http cache (instead of %(default)s)',
21 default=os.path.join(xdg_cache_home(), "paperdoorknob"))
22 parser.add_argument(
23 '--timeout',
24 help='How long to wait for HTTP requests, in seconds',
25 default=30)
26 parser.add_argument('url', help='URL to retrieve')
27 return parser
28
29
30 def fetch(url: str, session: requests.Session, timeout: int) -> None:
31 with session.get(url, timeout=timeout) as r:
32 r.raise_for_status()
33
34
35 def main() -> None:
36 args = command_line_parser().parse_args()
37 with requests_cache.CachedSession(args.cache_path, cache_control=True) as session:
38 fetch(args.url, session, args.timeout)
39
40
41 if __name__ == '__main__':
42 main()