From 5ad2886bcb64126e9bcb9fa244f3afb527827eab Mon Sep 17 00:00:00 2001
From: akamhy <64683866+akamhy@users.noreply.github.com>
Date: Mon, 4 May 2020 09:49:40 +0530
Subject: [PATCH] python 2.x support

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

diff --git a/waybackpy/wrapper.py b/waybackpy/wrapper.py
index fa935da..f2dcf72 100644
--- a/waybackpy/wrapper.py
+++ b/waybackpy/wrapper.py
@@ -1,8 +1,12 @@
 # -*- coding: utf-8 -*-
 
 from datetime import datetime
-from urllib.request import Request, urlopen
-import urllib.error
+try:
+    from urllib.request import Request, urlopen
+    from urllib.error import HTTPError
+except ImportError:
+    from urllib2 import Request, urlopen, HTTPError
+
 
 default_UA = "waybackpy python package"
 
@@ -45,7 +49,7 @@ def save(url,UA=default_UA):
         raise InvalidUrlError("'%s' is not a vaild url." % url)
     try:
         response = urlopen(req) #nosec
-    except urllib.error.HTTPError as e:
+    except HTTPError as e:
         if e.code == 502:
             raise PageNotSavedError(e)
         elif e.code == 429:
@@ -72,9 +76,8 @@ def near(
     hdr = { 'User-Agent' : '%s' % UA }
     req = Request(request_url, headers=hdr)
     response = urlopen(req) #nosec
-    encoding = response.info().get_content_charset('utf8')
     import json
-    data = json.loads(response.read().decode(encoding))
+    data = json.loads(response.read().decode('utf8'))
     if not data["archived_snapshots"]:
         raise ArchiveNotFound("'%s' is not yet archived." % url)