028371a0e1
- Introduced a modular repository structure by creating separate repository classes for `Bot`, `MarketData`, and `RawTrade`, improving code organization and maintainability. - Updated the `DatabaseOperations` class to utilize the new repository classes, enhancing the abstraction of database interactions. - Refactored the `.env` file to update database connection parameters and add new logging and health monitoring configurations. - Modified the `okx_config.json` to change default timeframes for trading pairs, aligning with updated requirements. - Added comprehensive unit tests for the new repository classes, ensuring robust functionality and reliability. These changes improve the overall architecture of the database layer, making it more scalable and easier to manage.
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"""Base repository for all other repository classes."""
|
|
|
|
import logging
|
|
from contextlib import contextmanager
|
|
from typing import Optional
|
|
|
|
from ..connection import get_db_manager
|
|
|
|
|
|
class DatabaseOperationError(Exception):
|
|
"""Custom exception for database operation errors."""
|
|
pass
|
|
|
|
|
|
class BaseRepository:
|
|
"""Base class for all repository classes."""
|
|
|
|
def __init__(self, logger: Optional[logging.Logger] = None):
|
|
"""Initialize repository with optional logger."""
|
|
self.logger = logger
|
|
self._db_manager = get_db_manager()
|
|
self._db_manager.initialize()
|
|
|
|
def log_info(self, message: str) -> None:
|
|
"""Log info message if logger is available."""
|
|
if self.logger:
|
|
self.logger.info(message)
|
|
|
|
def log_debug(self, message: str) -> None:
|
|
"""Log debug message if logger is available."""
|
|
if self.logger:
|
|
self.logger.debug(message)
|
|
|
|
def log_error(self, message: str) -> None:
|
|
"""Log error message if logger is available."""
|
|
if self.logger:
|
|
self.logger.error(message)
|
|
|
|
@contextmanager
|
|
def get_session(self):
|
|
"""Get database session with automatic cleanup."""
|
|
if not self._db_manager:
|
|
raise DatabaseOperationError("Database manager not initialized")
|
|
|
|
with self._db_manager.get_session() as session:
|
|
yield session |