2.7 compat

pytube/compat.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# flake8: noqa
+import sys
+
+python_version = sys.version_info[0]
+
+if python_version == 2:
+    from urllib import urlencode
+    from urllib2 import quote
+    from urllib2 import unquote
+    from urllib2 import urlopen
+    from urlparse import parse_qsl
+
+elif python_version == 3:
+    from urllib.parse import parse_qsl
+    from urllib.parse import quote
+    from urllib.parse import unquote
+    from urllib.parse import urlencode
+    from urllib.request import urlopen

pytube/extract.py

@@ -6,10 +6,10 @@ pytube.extract
 """
 import json
 import re
-from urllib.parse import parse_qsl
-from urllib.parse import quote
-from urllib.parse import urlencode
 
+from pytube.compat import parse_qsl
+from pytube.compat import quote
+from pytube.compat import urlencode
 from pytube.helpers import memoize
 
 

pytube/mixins.py

@@ -4,10 +4,9 @@ pytube.mixins
 ~~~~~~~~~~~~~
 
 """
-from urllib.parse import parse_qsl
-from urllib.parse import unquote
-
 from pytube import cipher
+from pytube.compat import parse_qsl
+from pytube.compat import unquote
 
 
 def apply_signature(video_info, fmt, js):

pytube/request.py

@@ -5,7 +5,7 @@ pytube.request
 
 Implements a simple wrapper around urlopen.
 """
-from urllib.request import urlopen
+from pytube.compat import urlopen
 
 
 def get(url, headers=False, streaming=False, chunk_size=8 * 1024):