diff options
author | Diogo Sousa <montezdesousa@gmail.com> | 2024-05-27 16:37:49 +0100 |
---|---|---|
committer | Diogo Sousa <montezdesousa@gmail.com> | 2024-05-27 16:37:49 +0100 |
commit | 795e91f09915936c78f88fe67282ec8dfec166b9 (patch) | |
tree | cbd115d36c7a4e6371100540791e86cf8bfa7a10 | |
parent | 6159c6ec9d05b083a42601610461f3d7a2545449 (diff) |
feat: setup stdout logger
-rw-r--r-- | openbb_platform/core/openbb_core/app/logs/handlers_manager.py | 6 | ||||
-rw-r--r-- | openbb_platform/providers/binance/openbb_binance/models/crypto_historical.py | 11 |
2 files changed, 11 insertions, 6 deletions
diff --git a/openbb_platform/core/openbb_core/app/logs/handlers_manager.py b/openbb_platform/core/openbb_core/app/logs/handlers_manager.py index 2d9a0efb927..4ad9f929d2e 100644 --- a/openbb_platform/core/openbb_core/app/logs/handlers_manager.py +++ b/openbb_platform/core/openbb_core/app/logs/handlers_manager.py @@ -45,8 +45,11 @@ class HandlersManager: def _add_stdout_handler(self): """Add a stdout handler.""" handler = logging.StreamHandler(sys.stdout) - formatter = FormatterWithExceptions(settings=self._settings) + formatter = logging.Formatter( + "%(asctime)s - %(levelname)s - %(name)s - %(message)s" + ) handler.setFormatter(formatter) + handler.setLevel(logging.DEBUG) logging.getLogger().addHandler(handler) def _add_stderr_handler(self): @@ -68,6 +71,7 @@ class HandlersManager: handler = PathTrackingFileHandler(settings=self._settings) formatter = FormatterWithExceptions(settings=self._settings) handler.setFormatter(formatter) + handler.setLevel(logging.INFO) logging.getLogger().addHandler(handler) def update_handlers(self, settings: LoggingSettings): diff --git a/openbb_platform/providers/binance/openbb_binance/models/crypto_historical.py b/openbb_platform/providers/binance/openbb_binance/models/crypto_historical.py index 4aca9311fb8..13dc85358f7 100644 --- a/openbb_platform/providers/binance/openbb_binance/models/crypto_historical.py +++ b/openbb_platform/providers/binance/openbb_binance/models/crypto_historical.py @@ -6,16 +6,17 @@ from datetime import datetime, timedelta from typing import Any, AsyncGenerator, AsyncIterator, Dict, Optional import websockets +from openbb_core.provider.abstract.fetcher import Fetcher from openbb_core.provider.standard_models.crypto_historical import ( CryptoHistoricalData, CryptoHistoricalQueryParams, ) from pydantic import Field -from openbb_platform.core.openbb_core.provider.abstract.fetcher import Fetcher - # pylint: disable=unused-argument, arguments-differ +logger = logging.getLogger(__name__) + class BinanceCryptoHistoricalQueryParams(CryptoHistoricalQueryParams): """Binance Crypto Historical Query Params.""" @@ -66,17 +67,17 @@ class BinanceCryptoHistoricalFetcher(Fetcher): async with websockets.connect( f"wss://stream.binance.com:9443/ws/{query.symbol.lower()}@miniTicker" ) as websocket: - logging.info("Connected to WebSocket server.") + logger.info("Connected to WebSocket server.") end_time = datetime.now() + timedelta(seconds=query.lifetime) try: while datetime.now() < end_time: chunk = await websocket.recv() yield json.loads(chunk) except websockets.exceptions.ConnectionClosed as e: - logging.error("WebSocket connection closed.") + logger.error("WebSocket connection closed.") raise e finally: - logging.info("WebSocket connection closed.") + logger.info("WebSocket connection closed.") @staticmethod async def atransform_data( |