superset_tool logger rework
This commit is contained in:
@@ -16,7 +16,7 @@ from ...core.config_models import AppConfig, Environment, GlobalSettings
|
|||||||
from ...dependencies import get_config_manager
|
from ...dependencies import get_config_manager
|
||||||
from ...core.config_manager import ConfigManager
|
from ...core.config_manager import ConfigManager
|
||||||
from ...core.logger import logger
|
from ...core.logger import logger
|
||||||
from superset_tool.client import SupersetClient
|
from ...core.superset_client import SupersetClient
|
||||||
from superset_tool.models import SupersetConfig
|
from superset_tool.models import SupersetConfig
|
||||||
import os
|
import os
|
||||||
# [/SECTION]
|
# [/SECTION]
|
||||||
|
|||||||
@@ -11,11 +11,18 @@
|
|||||||
from typing import List, Dict, Optional, Tuple
|
from typing import List, Dict, Optional, Tuple
|
||||||
from superset_tool.client import SupersetClient as BaseSupersetClient
|
from superset_tool.client import SupersetClient as BaseSupersetClient
|
||||||
from superset_tool.models import SupersetConfig
|
from superset_tool.models import SupersetConfig
|
||||||
|
from backend.src.core.logger import logger
|
||||||
|
from superset_tool.utils.logger import SupersetLogger
|
||||||
# [/SECTION]
|
# [/SECTION]
|
||||||
|
|
||||||
# [DEF:SupersetClient:Class]
|
# [DEF:SupersetClient:Class]
|
||||||
# @PURPOSE: Extended SupersetClient for migration-specific operations.
|
# @PURPOSE: Extended SupersetClient for migration-specific operations.
|
||||||
class SupersetClient(BaseSupersetClient):
|
class SupersetClient(BaseSupersetClient):
|
||||||
|
def __init__(self, config: SupersetConfig):
|
||||||
|
# Initialize with the application's logger wrapped in SupersetLogger
|
||||||
|
# to ensure BeliefFormatter is used.
|
||||||
|
sl_logger = SupersetLogger(logger=logger)
|
||||||
|
super().__init__(config=config, logger=sl_logger)
|
||||||
|
|
||||||
# [DEF:SupersetClient.get_databases_summary:Function]
|
# [DEF:SupersetClient.get_databases_summary:Function]
|
||||||
# @PURPOSE: Fetch a summary of databases including uuid, name, and engine.
|
# @PURPOSE: Fetch a summary of databases including uuid, name, and engine.
|
||||||
|
|||||||
@@ -28,7 +28,11 @@ class SupersetLogger:
|
|||||||
# @PARAM: log_dir (Optional[Path]) - Директория для сохранения лог-файлов.
|
# @PARAM: log_dir (Optional[Path]) - Директория для сохранения лог-файлов.
|
||||||
# @PARAM: level (int) - Уровень логирования (e.g., `logging.INFO`).
|
# @PARAM: level (int) - Уровень логирования (e.g., `logging.INFO`).
|
||||||
# @PARAM: console (bool) - Флаг для включения вывода в консоль.
|
# @PARAM: console (bool) - Флаг для включения вывода в консоль.
|
||||||
def __init__(self, name: str = "superset_tool", log_dir: Optional[Path] = None, level: int = logging.INFO, console: bool = True) -> None:
|
def __init__(self, name: str = "superset_tool", log_dir: Optional[Path] = None, level: int = logging.INFO, console: bool = True, logger: Optional[logging.Logger] = None) -> None:
|
||||||
|
if logger:
|
||||||
|
self.logger = logger
|
||||||
|
return
|
||||||
|
|
||||||
self.logger = logging.getLogger(name)
|
self.logger = logging.getLogger(name)
|
||||||
self.logger.setLevel(level)
|
self.logger.setLevel(level)
|
||||||
self.logger.propagate = False
|
self.logger.propagate = False
|
||||||
|
|||||||
Reference in New Issue
Block a user