diff options
Diffstat (limited to 'venv/lib/python3.7/site-packages/pip-10.0.1-py3.7.egg/pip/_vendor/cachecontrol/_cmd.py')
-rw-r--r-- | venv/lib/python3.7/site-packages/pip-10.0.1-py3.7.egg/pip/_vendor/cachecontrol/_cmd.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/venv/lib/python3.7/site-packages/pip-10.0.1-py3.7.egg/pip/_vendor/cachecontrol/_cmd.py b/venv/lib/python3.7/site-packages/pip-10.0.1-py3.7.egg/pip/_vendor/cachecontrol/_cmd.py new file mode 100644 index 0000000..10bc01e --- /dev/null +++ b/venv/lib/python3.7/site-packages/pip-10.0.1-py3.7.egg/pip/_vendor/cachecontrol/_cmd.py | |||
@@ -0,0 +1,60 @@ | |||
1 | import logging | ||
2 | |||
3 | from pip._vendor import requests | ||
4 | |||
5 | from pip._vendor.cachecontrol.adapter import CacheControlAdapter | ||
6 | from pip._vendor.cachecontrol.cache import DictCache | ||
7 | from pip._vendor.cachecontrol.controller import logger | ||
8 | |||
9 | from argparse import ArgumentParser | ||
10 | |||
11 | |||
12 | def setup_logging(): | ||
13 | logger.setLevel(logging.DEBUG) | ||
14 | handler = logging.StreamHandler() | ||
15 | logger.addHandler(handler) | ||
16 | |||
17 | |||
18 | def get_session(): | ||
19 | adapter = CacheControlAdapter( | ||
20 | DictCache(), | ||
21 | cache_etags=True, | ||
22 | serializer=None, | ||
23 | heuristic=None, | ||
24 | ) | ||
25 | sess = requests.Session() | ||
26 | sess.mount('http://', adapter) | ||
27 | sess.mount('https://', adapter) | ||
28 | |||
29 | sess.cache_controller = adapter.controller | ||
30 | return sess | ||
31 | |||
32 | |||
33 | def get_args(): | ||
34 | parser = ArgumentParser() | ||
35 | parser.add_argument('url', help='The URL to try and cache') | ||
36 | return parser.parse_args() | ||
37 | |||
38 | |||
39 | def main(args=None): | ||
40 | args = get_args() | ||
41 | sess = get_session() | ||
42 | |||
43 | # Make a request to get a response | ||
44 | resp = sess.get(args.url) | ||
45 | |||
46 | # Turn on logging | ||
47 | setup_logging() | ||
48 | |||
49 | # try setting the cache | ||
50 | sess.cache_controller.cache_response(resp.request, resp.raw) | ||
51 | |||
52 | # Now try to get it | ||
53 | if sess.cache_controller.cached_request(resp.request): | ||
54 | print('Cached!') | ||
55 | else: | ||
56 | print('Not cached :(') | ||
57 | |||
58 | |||
59 | if __name__ == '__main__': | ||
60 | main() | ||