1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
# SPDX-License-Identifier: MIT
# SPDX-FileCopyrightText: © 2017 Tobias Gruetzmacher
import gzip
import os.path
import re
from functools import lru_cache
from responses import GET, add
def _file(name):
return os.path.join(os.path.dirname(__file__), 'responses', name)
@lru_cache()
def content(name, ext: str = 'html'):
with gzip.open(_file(f'{name}.{ext}.gz'), 'r') as f:
return f.read()
@lru_cache()
def _img(name):
with open(_file(name + '.png'), 'rb') as f:
return f.read()
def page(url, pagename):
add(GET, url, content(pagename))
def json(url, pagename):
add(GET, url, content(pagename, 'json'))
def png(url, name='empty'):
add(GET, url, _img(name), content_type='image/png')
def jpeg(url, name='empty'):
add(GET, url, _img(name), content_type='image/jpeg')
def xkcd():
page('https://xkcd.com/', 'xkcd-1899')
for num in (302, 303, 1898, 1899):
page('https://xkcd.com/%i/' % num, 'xkcd-%i' % num)
png(re.compile(r'https://imgs\.xkcd\.com/.*\.png'))
|
