From a4fda300582b1da19773e893e0b2962daf35ef87 Mon Sep 17 00:00:00 2001
From: Danglewood <85772166+deeleeramone@users.noreply.github.com>
Date: Tue, 17 Dec 2024 16:56:34 -0800
Subject: [PATCH] lint

---
 .../provider/utils/websockets/database.py     | 27 ++++++++++++++-----
 .../websockets/openbb_websockets/client.py    |  5 +++-
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/openbb_platform/core/openbb_core/provider/utils/websockets/database.py b/openbb_platform/core/openbb_core/provider/utils/websockets/database.py
index cefb2545a1e..e1aa70bb99a 100644
--- a/openbb_platform/core/openbb_core/provider/utils/websockets/database.py
+++ b/openbb_platform/core/openbb_core/provider/utils/websockets/database.py
@@ -105,7 +105,7 @@ class Database:
         import aiosqlite
 
         try:
-            if os.path.exists(self.results_file):
+            if self.results_file is not None and os.path.exists(self.results_file):  # type: ignore
                 async with aiosqlite.connect(
                     self.results_file, loop=self.loop, **self.kwargs
                 ) as conn:
@@ -139,7 +139,10 @@ class Database:
                 await conn.commit()
                 self.table_exists = True
         except Exception as e:
-            msg = f"Unexpected error while creating SQLite database -> {e.__class__.__name__}: {e}"
+            msg = (
+                "Unexpected error while creating SQLite database ->"
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             self._exception = e
             raise OpenBBError(msg) from e
@@ -221,7 +224,10 @@ class Database:
         try:
             run_async(self._write_to_db, message)
         except Exception as e:  # pylint: disable=broad-except
-            msg = f"Unexpected error while writing to SQLite database -> {e.__class__.__name__}: {e}"
+            msg = (
+                "Unexpected error while writing to SQLite database ->"
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             self._exception = e
             raise OpenBBError(msg) from e
@@ -272,7 +278,10 @@ class Database:
                 )
             )
         except Exception as e:
-            msg = f"Unexpected error while deserializing row -> {e.__class__.__name__}: {e}"
+            msg = (
+                "Unexpected error while deserializing row -> "
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             self._exception = e
             raise OpenBBError(msg) from e
@@ -282,7 +291,10 @@ class Database:
         try:
             return run_async(self._fetch_all, limit)
         except Exception as e:
-            msg = f"Unexpected error while reading from SQLite database -> {e.__class__.__name__}: {e}"
+            msg = (
+                "Unexpected error while reading from SQLite database ->"
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             self._exception = e
             raise OpenBBError(msg) from e
@@ -402,6 +414,9 @@ class Database:
         try:
             run_async(self._clear_results)
         except Exception as e:  # pylint: disable=broad-except
-            msg = f"Error clearing results: {e.__class__.__name__}: {e}"
+            msg = (
+                "Error clearing results: "
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             raise OpenBBError(msg) from e
diff --git a/openbb_platform/extensions/websockets/openbb_websockets/client.py b/openbb_platform/extensions/websockets/openbb_websockets/client.py
index 1b14d3dce88..f32bb7ce1ec 100644
--- a/openbb_platform/extensions/websockets/openbb_websockets/client.py
+++ b/openbb_platform/extensions/websockets/openbb_websockets/client.py
@@ -167,7 +167,10 @@ class WebSocketClient:  # pylint: disable=too-many-instance-attributes
                 data_model=self.data_model,
             )
         except Exception as e:  # pylint: disable=broad-except
-            msg = f"Unexpected error setting up the SQLite database and table -> {e.__class___.__name__}: {e}"
+            msg = (
+                "Unexpected error setting up the SQLite database and table ->"
+                f" {e.__class__.__name__ if hasattr(e, "__class__") else e.__name__}: {e}"
+            )
             self.logger.error(msg)
             self._exception = OpenBBError(msg)