[youtube] Fix search continuations

This commit is contained in:
pukkandan 2021-02-11 17:10:38 +05:30
parent cc2db87805
commit a96c6d154a
1 changed files with 6 additions and 6 deletions

View File

@ -3181,6 +3181,12 @@ class YoutubeSearchIE(SearchInfoExtractor, YoutubeBaseInfoExtractor):
# So we search through all entries till we find them. # So we search through all entries till we find them.
continuation_token = None continuation_token = None
for slr_content in slr_contents: for slr_content in slr_contents:
if continuation_token is None:
continuation_token = try_get(
slr_content,
lambda x: x['continuationItemRenderer']['continuationEndpoint']['continuationCommand']['token'],
compat_str)
isr_contents = try_get( isr_contents = try_get(
slr_content, slr_content,
lambda x: x['itemSectionRenderer']['contents'], lambda x: x['itemSectionRenderer']['contents'],
@ -3202,12 +3208,6 @@ class YoutubeSearchIE(SearchInfoExtractor, YoutubeBaseInfoExtractor):
if total == n: if total == n:
return return
if continuation_token is None:
continuation_token = try_get(
slr_content,
lambda x: x['continuationItemRenderer']['continuationEndpoint']['continuationCommand']['token'],
compat_str)
if not continuation_token: if not continuation_token:
break break
data['continuation'] = continuation_token data['continuation'] = continuation_token