aboutsummaryrefslogtreecommitdiffstats
path: root/youtube/yt_data_extract
diff options
context:
space:
mode:
Diffstat (limited to 'youtube/yt_data_extract')
-rw-r--r--youtube/yt_data_extract/watch_extraction.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/youtube/yt_data_extract/watch_extraction.py b/youtube/yt_data_extract/watch_extraction.py
index 0970125..9dbb252 100644
--- a/youtube/yt_data_extract/watch_extraction.py
+++ b/youtube/yt_data_extract/watch_extraction.py
@@ -415,8 +415,14 @@ def extract_watch_info(polymer_json):
if error:
info['playability_error'] = error
+ player_response = top_level.get('playerResponse', {})
+
+ # usually, only the embedded one has the urls
player_args = deep_get(top_level, 'player', 'args', default={})
- player_response = json.loads(player_args['player_response']) if 'player_response' in player_args else {}
+ if 'player_response' in player_args:
+ embedded_player_response = json.loads(player_args['player_response'])
+ else:
+ embedded_player_response = {}
# captions
info['automatic_caption_languages'] = []
@@ -446,7 +452,9 @@ def extract_watch_info(polymer_json):
print('WARNING: Found non-translatable caption language')
# formats
- _extract_formats(info, player_response)
+ _extract_formats(info, embedded_player_response)
+ if not info['formats']:
+ _extract_formats(info, player_response)
# playability errors
_extract_playability_error(info, player_response)