From 402a71f3e6598e87952aa5c99f1a1dfe71c86800 Mon Sep 17 00:00:00 2001 From: Danglewood <85772166+deeleeramone@users.noreply.github.com> Date: Sat, 9 Nov 2024 17:48:50 -0800 Subject: [PATCH] use sys.stdout.write instead of logger.error for unexpected error --- .../websockets/openbb_websockets/client.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/openbb_platform/extensions/websockets/openbb_websockets/client.py b/openbb_platform/extensions/websockets/openbb_websockets/client.py index 1c43612da30..b045aec1896 100644 --- a/openbb_platform/extensions/websockets/openbb_websockets/client.py +++ b/openbb_platform/extensions/websockets/openbb_websockets/client.py @@ -217,14 +217,15 @@ class WebSocketClient: if ( "server rejected" in output.lower() or "PROVIDER ERROR" in output - or "Unexpected error" in output + or "unexpected error" in output.lower() ): - err = ChildProcessError(output) - self._exception = err - self.logger.error(output) self._psutil_process.kill() self._process.wait() self._thread.join() + err = ChildProcessError(output) + self._exception = err + sys.stdout.write(msg + "\n") + sys.stdout.flush() break output = clean_message(output) @@ -265,9 +266,9 @@ class WebSocketClient: output = None if output: - if "ERROR:" in output: + if output.startswith("ERROR:"): output = output.replace("ERROR:", "BROADCAST ERROR:") + "\n" - if "INFO:" in output: + elif output.startswith("INFO:"): output = output.replace("INFO:", "BROADCAST INFO:") + "\n" output = output[0] if isinstance(output, tuple) else output output = clean_message(output)