Merge pull request #2 from Edanflame/main

修改on_error函数
vnpy · Sep 24, 2021 · 93c8512 · 93c8512
2 parents a9b96f1 + 921a906
commit 93c8512
Showing 1 changed file with 16 additions and 7 deletions.
diff --git a/vnpy_websocket/websocket_client.py b/vnpy_websocket/websocket_client.py
@@ -129,16 +129,25 @@ def on_packet(self, packet: dict):
         """收到数据回调"""
         pass
 
-    def on_error(self, exception_type: type, exception_value: Exception, tb):
+    def on_error(
+        self,
+        exception_type: type,
+        exception_value: Exception,
+        tb
+    ) -> None:
         """触发异常回调"""
-        sys.stderr.write(
-            self.exception_detail(exception_type, exception_value, tb)
-        )
-        return sys.excepthook(exception_type, exception_value, tb)
+        try:
+            print("WebsocketClient on error" + "-" * 10)
+            print(self.exception_detail(exception_type, exception_value, tb))
+        except Exception:
+            traceback.print_exc()
 
     def exception_detail(
-        self, exception_type: type, exception_value: Exception, tb
-    ):
+        self,
+        exception_type: type,
+        exception_value: Exception,
+        tb
+    ) -> str:
         """异常信息格式化"""
         text = "[{}]: Unhandled WebSocket Error:{}\n".format(
             datetime.now().isoformat(), exception_type