jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 12:40:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

more linting

Browse Source
This commit is contained in:
Danglewood
2025-01-06 19:37:32 -08:00
parent 58dadc5b1e
commit 854052d43a
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
openbb_platform/core/openbb_core/provider/utils/websockets/broadcast.py
View File

@@ -215,7 +215,7 @@ class BroadcastServer: # pylint: disable=too-many-instance-attributes
async def stream_results( # noqa: PLR0915 # pylint: disable=too-many-branches
self,
sql: str = None,
sql: Optional[str] = None,
replay: bool = False,
):
"""Continuously read the database and send new messages as JSON via WebSocket."""
@@ -283,7 +283,7 @@ class BroadcastServer: # pylint: disable=too-many-instance-attributes
continue
for row in rows:
last_id = row[0] if row[0] > last_id else last_id
await self.websocket.send_json(
await self.websocket.send_json( # type: ignore
json.dumps(json.loads(row[1]))
)
if self.replay is True: # type: ignore

15
openbb_platform/core/openbb_core/provider/utils/websockets/database.py
View File

@@ -244,7 +244,7 @@ class Database:
if name == "read":
if ":" not in self.results_file: # type: ignore
results_file = ( # type: ignore
"file:"
"file:" # type: ignore
+ (
self.results_file # type: ignore
if self.results_file.startswith("/") # type: ignore
@@ -506,6 +506,7 @@ class Database:
except Exception as e: # pylint: disable=broad-except
msg = f"{e.__class__.__name__ if hasattr(e, '__class__') else e}: {e.args}"
self.logger.error(msg)
return []
async def _clear_results(self):
"""Clear the results from the SQLite database."""
@@ -951,17 +952,17 @@ class DatabaseWriter:
prune_thread = threading.Thread(target=self._run_prune_event)
prune_thread.daemon = True
prune_thread.name = "WebSocketPruneThread"
self.prune_thread = prune_thread
self.prune_thread.start()
self.prune_thread = prune_thread # type: ignore
self.prune_thread.start() # type: ignore
finally:
self.prune_thread.join(timeout=1)
self.prune_thread.join(timeout=1) # type: ignore
def stop_prune_task(self):
"""Public method to stop the background pruning task."""
if hasattr(self, "prune_thread") and self.prune_thread:
self.prune_thread.join(timeout=1)
if self.prune_thread.is_alive():
kill_thread(self.prune_thread)
self.prune_thread.join(timeout=1) # type: ignore
if self.prune_thread.is_alive(): # type: ignore
kill_thread(self.prune_thread) # type: ignore
self._prune_running = False
self.prune_thread = None