diff options
author | Sergey M․ <dstftw@gmail.com> | 2021-02-10 22:28:58 +0700 |
---|---|---|
committer | Sergey M․ <dstftw@gmail.com> | 2021-02-10 22:28:58 +0700 |
commit | a4c7ed6b1e9100be8ef65c44e7e6e43b9314ff5c (patch) | |
tree | 65e4c2c0488be2ab0113e2c1ef0805429a646f16 | |
parent | 7f8b8bc418b8831ea1c2ae8de64e3bf0e8b707f8 (diff) |
[youtube:tab] Improve grid continuation extraction (closes #28130)
-rw-r--r-- | youtube_dl/extractor/youtube.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/youtube_dl/extractor/youtube.py b/youtube_dl/extractor/youtube.py index 346311d9b..c78996629 100644 --- a/youtube_dl/extractor/youtube.py +++ b/youtube_dl/extractor/youtube.py @@ -2374,9 +2374,9 @@ class YoutubeTabIE(YoutubeBaseInfoExtractor): next_continuation = cls._extract_next_continuation_data(renderer) if next_continuation: return next_continuation - contents = renderer.get('contents') - if not isinstance(contents, list): - return + contents = [] + for key in ('contents', 'items'): + contents.extend(try_get(renderer, lambda x: x[key], list) or []) for content in contents: if not isinstance(content, dict): continue @@ -2509,6 +2509,13 @@ class YoutubeTabIE(YoutubeBaseInfoExtractor): continuation_item = continuation_items[0] if not isinstance(continuation_item, dict): continue + renderer = continuation_item.get('gridVideoRenderer') + if renderer: + grid_renderer = {'items': continuation_items} + for entry in self._grid_entries(grid_renderer): + yield entry + continuation = self._extract_continuation(grid_renderer) + continue renderer = continuation_item.get('playlistVideoRenderer') or continuation_item.get('itemSectionRenderer') if renderer: video_list_renderer = {'contents': continuation_items} |