Fix zoom chat downloader

xenova · May 5, 2023 · 05f3826 · 05f3826
1 parent 14b9521
commit 05f3826
Showing 1 changed file with 30 additions and 11 deletions.
diff --git a/chat_downloader/sites/zoom.py b/chat_downloader/sites/zoom.py
@@ -10,7 +10,8 @@
 from ..utils.core import (
     time_to_seconds,
     regex_search,
-    ensure_seconds
+    ensure_seconds,
+    multi_get,
 )
 from ..errors import (
     SiteError,
@@ -34,6 +35,7 @@ class ZoomChatDownloader(BaseChatDownloader):
 
     _ZOOM_HOMEPAGE = 'https://zoom.us/'
     _ZOOM_PATH_TEMPLATE = 'rec/play/{id}'
+    _ZOOM_API_TEMPLATE = 'nws/recording/1.0/play/info/{file_id}'
 
     _INITIAL_INFO_REGEX = r'(?s)window\.__data__\s*=\s*({.+?});'
     _CHAT_MESSAGES_REGEX = r'window\.__data__\.chatList\.push\((\{[\s\S]+?\})\)'
@@ -43,7 +45,7 @@ class ZoomChatDownloader(BaseChatDownloader):
     }
 
     _REMAPPING = {
-        'username': 'author_name',
+        'userName': 'author_name',
         'time': 'time_text',
         'content': 'message',
     }
@@ -116,16 +118,34 @@ def get_chat_by_video_id(self, video_id, params, base_url=_ZOOM_HOMEPAGE):
                 raise ParsingError('Error parsing video')
 
         initial_info = self._parse_js_dict(json_string)
-
         video_type = 'video' if initial_info.get('isVideo') else 'not_video'
 
-        return Chat(
-            self._get_chat_messages(page_data, params),
+        file_id = initial_info.get('fileId')
+        if not file_id:
+            raise ParsingError('Error parsing video. Unable to find file ID.')
+
+        api_url = base_url + self._ZOOM_API_TEMPLATE.format(file_id=file_id)
+
+        api_data = self._session_get_json(api_url)
 
-            title=initial_info.get('topic'),
+        if api_data.get('errorCode') != 0:
+            raise ZoomError(
+                f'An error occured: {api_data.get("errorMessage")} ({api_data.get("errorCode")})')
+
+        result = api_data.get('result')
+        if not result:
+            raise ZoomError(
+                f'Unable to find chat messages for video {video_id}')
+
+        chat_messages = result.get('meetingChatList') or []
+        title = multi_get(result, 'meet', 'topic')
+        return Chat(
+            self._get_chat_messages(chat_messages, params),
+            title=title,
             video_type=video_type,
-            start_time=initial_info.get('fileStartTime'),
-            id=initial_info.get('recordingId'),
+            start_time=result.get('fileStartTime'),
+            id=video_id,
+            duration=result.get('duration'),
         )
 
     def _parse_js_dict(self, json_string):
@@ -136,12 +156,11 @@ def _parse_js_dict(self, json_string):
         result = re.sub(r":\s+'(.*)'", ": \"\\g<1>\"", result, 0, re.MULTILINE)
         return json.loads(result)
 
-    def _get_chat_messages(self, page_data, params):
+    def _get_chat_messages(self, messages, params):
         start_time = ensure_seconds(params.get('start_time'), 0)
         end_time = ensure_seconds(params.get('end_time'), float('inf'))
 
-        for item in re.findall(self._CHAT_MESSAGES_REGEX, page_data):
-            data = self._parse_js_dict(item)
+        for data in messages:
             data = r.remap_dict(data, self._REMAPPING)
 
             # Process time inforamtion