Merge pull request #465 from dart-neitro/master

pytube/streams.py

@@ -165,6 +165,27 @@ class Stream(object):
             self._filesize = int(headers['content-length'])
         return self._filesize
 
+    @property
+    def title(self):
+        """Get title of video
+
+        :rtype: str
+        :returns:
+            Youtube video title
+        """
+        player_config_args = self.player_config_args or {}
+
+        if 'title' in player_config_args:
+            return player_config_args['title']
+
+        details = self.player_config_args.get(
+            'player_response', {}).get('videoDetails', {})
+
+        if 'title' in details:
+            return details['title']
+
+        return 'Unknown YouTube Video Title'
+
     @property
     def default_filename(self):
         """Generate filename based on the video title.
@@ -173,8 +194,8 @@ class Stream(object):
         :returns:
             An os file system compatible filename.
         """
-        title = self.player_config_args['title']
-        filename = safe_filename(title)
+
+        filename = safe_filename(self.title)
         return '{filename}.{s.subtype}'.format(filename=filename, s=self)
 
     def download(self, output_path=None, filename=None, filename_prefix=None):

tests/test_streams.py

@@ -19,6 +19,21 @@ def test_default_filename(cipher_signature):
     assert stream.default_filename == expected
 
 
+def test_title(cipher_signature):
+    expected = 'PSY - GANGNAM STYLE(강남스타일) M/V'
+    stream = cipher_signature.streams.first()
+    assert stream.title == expected
+
+    expected = 'PSY - GANGNAM STYLE(강남스타일)'
+    stream.player_config_args = {
+        'player_response': {'videoDetails': {'title': expected}}}
+    assert stream.title == expected
+
+    expected = 'Unknown YouTube Video Title'
+    stream.player_config_args = {}
+    assert stream.title == expected
+
+
 def test_download(cipher_signature, mocker):
     mocker.patch.object(request, 'get')
     request.get.side_effect = [