Merge pull request #177 from JC-Chung/JC-Chung-patch-1

[YouTube] Q&A message didn't have `header`
xenova · Nov 29, 2022 · 60a5a64 · 60a5a64
2 parents 657e56e + abfe92b
commit 60a5a64
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/chat_downloader/sites/youtube.py b/chat_downloader/sites/youtube.py
@@ -1758,18 +1758,21 @@ def _get_chat_messages(self, initial_info, ytcfg, params):
 
                             header = original_item[original_message_type].get(
                                 'header')
-                            parsed_header = self._parse_item(
-                                header, offset=offset)
-                            header_message = parsed_header.get('message')
+                            if header:
+                                parsed_header = self._parse_item(
+                                    header, offset=offset)
+                                header_message = parsed_header.get('message')
 
                             contents = original_item[original_message_type].get(
                                 'contents')
                             parsed_contents = self._parse_item(
                                 contents, offset=offset)
 
-                            data.update(parsed_header)
+                            if header:
+                                data.update(parsed_header)
                             data.update(parsed_contents)
-                            data['header_message'] = header_message
+                            if header:
+                                data['header_message'] = header_message
                         else:
                             debug_log(
                                 'No bannerRenderer item',