浏览代码

[hotstar] fix extraction in python 2(closes #17696)

Remita Amine 7 年之前
父节点
当前提交
05e7c184da
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      youtube_dl/extractor/hotstar.py

+ 6 - 1
youtube_dl/extractor/hotstar.py

@@ -11,6 +11,7 @@ from ..utils import (
     determine_ext,
     ExtractorError,
     int_or_none,
+    try_get,
 )
 
 
@@ -72,7 +73,11 @@ class HotStarIE(HotStarBaseIE):
         app_state = self._parse_json(self._search_regex(
             r'<script>window\.APP_STATE\s*=\s*({.+?})</script>',
             webpage, 'app state'), video_id)
-        video_data = list(app_state.values())[0]['initialState']['contentData']['content']
+        video_data = {}
+        for v in app_state.values():
+            content = try_get(v, lambda x: x['initialState']['contentData']['content'], dict)
+            if content and content.get('contentId') == video_id:
+                video_data = content
 
         title = video_data['title']