Ashish/yt-dlp
1
1
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-06-10 10:50:49 +00:00
Code Issues 7 Projects Releases Wiki Activity

move continue block, filter() comment_keys for None

Browse Source
This commit is contained in:
jakeogh 2024-05-14 18:05:58 -07:00
parent 4701ad6c2e
commit 6f5c669fcc
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
yt_dlp/extractor/youtube.py
View File

@ -3453,9 +3453,6 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
expected_type=dict, default={})
comment = self._extract_comment_old(comment_renderer, parent)
if not comment:
continue
comment_id = comment['id']
# new comment format
else:
@ -3464,13 +3461,14 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
or traverse_obj(content, ('commentViewModel', {dict})))
if not view_model:
continue
comment_id = view_model['commentId']
comment_key = view_model.get('commentKey')
toolbar_state_key = view_model.get('toolbarStateKey')
entities = traverse_obj(entity_payloads, lambda _, v: v["entityKey"] in [comment_key, toolbar_state_key])
comment_keys = list(filter(None, [view_model.get('commentKey'), view_model.get('toolbarStateKey')]))
entities = traverse_obj(entity_payloads, lambda _, v: v['entityKey'] in comment_keys)
comment = self._extract_comment(view_model, entities, parent)
if not comment:
continue
comment_id = comment['id']
if comment.get('is_pinned'):
tracker['pinned_comment_ids'].add(comment_id)
# Sometimes YouTube may break and give us infinite looping comments.