feat(danmaku): try to resolve TimeoutError

Signed-off-by: hldh214 <[email protected]>
hldh214 · Jan 10, 2024 · a6e84d9 · a6e84d9
1 parent 3f91c0e
commit a6e84d9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/recorder/danmaku/douyin/dylr.py b/recorder/danmaku/douyin/dylr.py
@@ -3,7 +3,6 @@
 import gzip
 import logging
 import time
-import traceback
 
 import pymongo.errors
 import websockets
@@ -94,9 +93,8 @@ async def subscribe(room_id):
     ):
         try:
             await consumer_handler(websocket, room_id)
-        except websockets.WebSocketException:
-            logging.warning('WebSocketException excepted: ' + traceback.format_exc())
-            raise
+        except websockets.WebSocketException as e:
+            logging.warning('WebSocketException excepted: ' + str(e))
 
 
 async def _main(room_id, interval):