fix issue with py2.7

This commit is contained in:
akamhy 2020-05-07 15:34:41 +05:30 committed by GitHub
parent a8e94dfb25
commit 74d3bc154b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -112,6 +112,11 @@ def total_archives(url, UA=default_UA):
hdr = { 'User-Agent' : '%s' % UA }
request_url = "https://web.archive.org/cdx/search/cdx?url=%s&output=json" % clean_url(url)
req = Request(request_url, headers=hdr) # nosec
with urlopen(req) as response: # nosec
data = json.loads(response.read())
return (len(data))
try:
response = urlopen(req) #nosec
except HTTPError as e:
if e.code == 502:
raise BadGateWay(e)
elif e.code == 503:
raise WaybackUnavailable(e)
return (len(json.loads(response.read())))