From 74d3bc154bf6573884487f2ad44ac78da4f6613e Mon Sep 17 00:00:00 2001
From: akamhy <64683866+akamhy@users.noreply.github.com>
Date: Thu, 7 May 2020 15:34:41 +0530
Subject: [PATCH] fix issue with py2.7

---
 waybackpy/wrapper.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/waybackpy/wrapper.py b/waybackpy/wrapper.py
index 47e86bb..855692c 100644
--- a/waybackpy/wrapper.py
+++ b/waybackpy/wrapper.py
@@ -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())))