logger.pyโข2.17 kB
"""
Logging utility for JetsonMCP
Provides centralized logging configuration for the entire application.
"""
import logging
import sys
from pathlib import Path
from typing import Optional
def setup_logger(
log_level: str = "INFO",
log_file: Optional[str] = None,
enable_console: bool = True,
) -> None:
"""
Set up logging configuration for JetsonMCP.
Args:
log_level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
log_file: Optional path to log file
enable_console: Whether to enable console logging
"""
# Convert log level string to logging constant
numeric_level = getattr(logging, log_level.upper(), logging.INFO)
# Create formatter
formatter = logging.Formatter(
fmt="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
)
# Get root logger
root_logger = logging.getLogger()
root_logger.setLevel(numeric_level)
# Clear existing handlers
root_logger.handlers.clear()
# Add console handler if enabled
if enable_console:
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(numeric_level)
console_handler.setFormatter(formatter)
root_logger.addHandler(console_handler)
# Add file handler if specified
if log_file:
try:
# Create log directory if it doesn't exist
log_path = Path(log_file)
log_path.parent.mkdir(parents=True, exist_ok=True)
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(numeric_level)
file_handler.setFormatter(formatter)
root_logger.addHandler(file_handler)
except Exception as e:
# If file logging fails, log to console
root_logger.warning(f"Failed to set up file logging: {e}")
# Set specific logger levels
logging.getLogger("paramiko").setLevel(logging.WARNING)
logging.getLogger("asyncssh").setLevel(logging.WARNING)
# Log startup message
logger = logging.getLogger(__name__)
logger.info(f"JetsonMCP logging initialized at {log_level} level")