[sites:reddit] Improved VideoNotFound error message

xenova · Jul 23, 2021 · 36b704b · 36b704b
1 parent cd0949e
commit 36b704b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/chat_downloader/sites/reddit.py b/chat_downloader/sites/reddit.py
@@ -377,10 +377,10 @@ def get_chat_by_post_id(self, post_id, params, attempt_number=0, initial_info=No
             if is_live and socket_url:  # live stream
 
                 if not socket_url.startswith('wss://') or 'wss.redditmedia.com' not in socket_url:
-                    self.retry(attempt_number, text='Invalid websocket URL: {}'.format(socket_url), **params)
+                    self.retry(attempt_number, text='Invalid websocket URL: {}'.format(
+                        socket_url), **params)
                     return self.get_chat_by_post_id(post_id, params, attempt_number + 1)
 
-
                 chat_item.chat = self._get_chat_messages_by_socket(
                     socket_url, params)
 
@@ -399,7 +399,7 @@ def get_chat_by_post_id(self, post_id, params, attempt_number=0, initial_info=No
             raise RedditError(data)
 
         elif status == 'video not found':
-            raise VideoNotFound(status_message)
+            raise VideoNotFound('Video not found')
 
         else:  # Unknown
             raise UnexpectedError('Info: {}'.format(initial_info))