👌 Changed imports for both Python 2 & 3 support

alexmercerind · Aug 7, 2020 · 1bfbbb3 · 1bfbbb3
1 parent bda9d4c
commit 1bfbbb3
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 5 deletions.
diff --git a/youtubesearchpython/__requesthandler.py b/youtubesearchpython/__requesthandler.py
@@ -1,12 +1,21 @@
-import urllib
+import sys
+
+#########python2#########
+if sys.version_info < (2, 8):
+    from urllib import urlencode, urlopen
+
+#########python3#########
+else:
+    from urllib.request import urlopen
+    from urllib.parse import urlencode
 
 
 class requesthandler:
 
     def request(self):
 
         try:
-            query = urllib.urlencode({
+            query = urlencode({
                 "search_query": self.keyword,
                 "page": self.offset,
                 "sp": self.searchPreferences
@@ -15,7 +24,7 @@ def request(self):
 
             #########Making Network Request#########
 
-            response = urllib.urlopen(request).read()
+            response = urlopen(request).read()
 
             self.page = response.decode('utf_8')
 

diff --git a/youtubesearchpython/videos__pagehandler.py b/youtubesearchpython/videos__pagehandler.py
@@ -1,4 +1,12 @@
-import HTMLParser
+import sys
+
+#########python2#########
+if sys.version_info < (2, 8):
+    import HTMLParser
+
+#########python3#########
+else:
+    import html
 
 
 class pagehandler:
@@ -25,7 +33,9 @@ def pageResponseHandler(self):
 
         self.pageSource = self.page.split()
 
-        html = HTMLParser.HTMLParser()
+        #########python2#########
+        if sys.version_info < (2, 8):
+            html = HTMLParser.HTMLParser()
 
         for index in range(0, len(self.pageSource) - 1, 1):