fixed python 3 compatibility

pytube/api.py

@@ -3,9 +3,13 @@ from __future__ import unicode_literals
 from .exceptions import MultipleObjectsReturned, YouTubeError
 from .models import Video
 from .utils import safe_filename
-from urllib import urlencode
-from urllib2 import urlopen
-from urlparse import urlparse, parse_qs, unquote
+try:
+    from urllib import urlencode
+    from urllib2 import urlopen
+    from urlparse import urlparse, parse_qs, unquote
+except ImportError:
+    from urllib.parse import urlencode, urlparse, parse_qs, unquote
+    from urllib.request import urlopen
 
 import re
 
@@ -262,6 +266,4 @@ class YouTube(object):
             attr = YT_ENCODING.get(itag, None)
             if not attr:
                 return itag, None
-            data = {}
-            map(lambda k, v: data.update({k: v}), YT_ENCODING_KEYS, attr)
-            return itag, data
+            return itag, dict(zip(YT_ENCODING_KEYS, attr))

pytube/models.py

@@ -1,7 +1,10 @@
 from __future__ import unicode_literals
 from os.path import normpath, isfile
 from os import remove
-from urllib2 import urlopen
+try:
+    from urllib2 import urlopen
+except ImportError:
+    from urllib.request import urlopen
 from sys import exit
 
 
@@ -48,7 +51,8 @@ class Video(object):
 
         # Check for conflicting filenames
         if isfile(fullpath):
-            print "\n\nError: Conflicting filename: '{}'.\n\n".format(self.filename)
+            print("\n\nError: Conflicting filename:'{}'.\n\n".format(
+                  self.filename))
             exit(1)
 
         response = urlopen(self.url)
@@ -58,8 +62,9 @@ class Video(object):
         self._bytes_received = 0
         try:
             with open(fullpath, 'wb') as dst_file:
-        # Print downloading message
-                print "\nDownloading: '{0}.{1}' (Bytes: {2})\n\n".format(self.filename, self.extension, file_size)
+                # Print downloading message
+                print("\nDownloading: '{0}.{1}' (Bytes: {2})\n\n".format(
+                      self.filename, self.extension, file_size))
 
                 while True:
                     self._buffer = response.read(chunk_size)
@@ -75,16 +80,20 @@ class Video(object):
 
         # Catch possible exceptions occurring during download
         except IOError:
-            print """\n\nError: Failed to open file.\nCheck that: ('{0}'), is a valid pathname.
-                \nOr that ('{1}.{2}') is a valid filename.\n\n""".format(path, self.filename, self.extension)
+            print("\n\nError: Failed to open file.\n" \
+                  "Check that: ('{0}'), is a valid pathname.\n\n" \
+                  "Or that ('{1}.{2}') is a valid filename.\n\n".format(
+                        path, self.filename, self.extension))
             exit(2)
 
         except BufferError:
-            print "\n\nError: Failed on writing buffer.\nFailed to write video to file.\n\n"
+            print("\n\nError: Failed on writing buffer.\n" \
+                  "Failed to write video to file.\n\n")
             exit(1)
 
         except KeyboardInterrupt:
-            print "\n\nInterrupt signal given.\nDeleting incomplete video ('{0}.{1}').\n\n".format(self.filename, self.extension)
+            print("\n\nInterrupt signal given.\nDeleting incomplete video" \
+                  "('{0}.{1}').\n\n".format(self.filename, self.extension))
             remove(fullpath)
             exit(1)
 

pytube/utils.py

@@ -41,4 +41,4 @@ def print_status(progress, file_size):
     percent = progress * 100. / file_size
     status = r"{0:10d}  [{1:3.2f}%]".format(progress, percent)
     status = status + chr(8) * (len(status) + 1)
-    print status,
+    print(status,)