use traverse_obj with mutations variable

This commit is contained in:
jakeogh 2024-05-14 17:39:44 -07:00
parent cf9751afd4
commit f09e66bb3f
1 changed files with 2 additions and 6 deletions

View File

@ -3587,6 +3587,7 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
raise
is_forced_continuation = False
continuation = None
mutations = traverse_obj(response, ('frameworkUpdates', 'entityBatchUpdate', 'mutations', ..., {dict}))
for continuation_items in traverse_obj(response, continuation_items_path, expected_type=list, default=[]):
if is_first_continuation:
continuation = extract_header(continuation_items)
@ -3595,12 +3596,7 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
break
continue
if 'frameworkUpdates' in response:
_iterator = extract_thread(continuation_items, response['frameworkUpdates']['entityBatchUpdate']['mutations'])
else:
_iterator = extract_thread(continuation_items, None)
for entry in _iterator:
for entry in extract_thread(continuation_items, mutations):
if not entry:
return
yield entry