mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-16 21:33:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

use get_first() and remove .get()

Browse source
This commit is contained in:
jakeogh 2024-05-14 18:55:02 -07:00
parent 6f5c669fcc
commit f6ced29596
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
yt_dlp/extractor/youtube.py
View file

@ -3308,9 +3308,11 @@ def _extract_heatmap(self, data):
})) or None
def _extract_comment(self, view_model, entities, parent=None):
comment_entity_payload = traverse_obj(entities, (..., 'payload', 'commentEntityPayload', {dict}), get_all=False)
toolbar_entity_payload = traverse_obj(entities, (..., 'payload', 'engagementToolbarStateEntityPayload', {dict}), get_all=False)
comment_id = comment_entity_payload.get('properties').get('commentId')
comment_entity_payload = get_first(entities, ('payload', 'commentEntityPayload', {dict}))
toolbar_entity_payload = get_first(entities, ('payload', 'engagementToolbarStateEntityPayload', {dict})) or {}
comment_id = traverse_obj(comment_entity_payload, ('properties', 'commentId', {str}))
if not comment_id:
return
info = {
'id': comment_id,