[sites:reddit] Retry when an invalid socket url is returned

xenova · Jul 23, 2021 · cd0949e · cd0949e
1 parent 71d70a7
commit cd0949e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/chat_downloader/sites/reddit.py b/chat_downloader/sites/reddit.py
@@ -376,9 +376,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://'):
-                    raise RedditError(
-                        'Invalid websocket URL: {}'.format(socket_url))
+                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)
+                    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)