62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import logging
|
|
from pathlib import Path
|
|
from typing import Optional, Union
|
|
|
|
from core.config.config_manager import ConfigManager
|
|
from core.config.logging_config import LoggingConfig
|
|
from core.config.pipeline_config import PipelineConfig
|
|
|
|
config_manager = ConfigManager()
|
|
|
|
|
|
def get_config() -> PipelineConfig:
|
|
"""Get the global configuration instance"""
|
|
return config_manager.get_config()
|
|
|
|
|
|
def load_config(config_path: Optional[Union[str, Path]] = None) -> PipelineConfig:
|
|
"""Load configuration from specified path"""
|
|
if config_path:
|
|
return config_manager.load_config(Path(config_path))
|
|
return config_manager.get_config()
|
|
|
|
|
|
def setup_logging(config: PipelineConfig):
|
|
"""Setup logging based on configuration"""
|
|
|
|
# Create logs directory
|
|
log_dir = config.paths.logs_dir
|
|
log_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Setup logging configuration
|
|
log_level = getattr(logging, config.logging.level.upper(), logging.INFO)
|
|
|
|
# Create formatter
|
|
formatter = logging.Formatter(config.logging.format)
|
|
|
|
# Setup root logger
|
|
root_logger = logging.getLogger()
|
|
root_logger.setLevel(log_level)
|
|
|
|
# Clear existing handlers
|
|
root_logger.handlers.clear()
|
|
|
|
# Console handler
|
|
if config.logging.console_logging:
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(formatter)
|
|
root_logger.addHandler(console_handler)
|
|
|
|
# File handler
|
|
if config.logging.file_logging:
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
log_file_path = log_dir / config.logging.log_file
|
|
file_handler = RotatingFileHandler(
|
|
log_file_path,
|
|
maxBytes=config.logging.max_log_size,
|
|
backupCount=config.logging.backup_count,
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
root_logger.addHandler(file_handler)
|