"""
Logging configuration for AWS MCP Server
"""
import logging
import sys
from pathlib import Path
def setup_logging(
level: str = "INFO",
format_string: str | None = None,
include_timestamp: bool = True,
log_file: str | None = None,
max_bytes: int = 10 * 1024 * 1024, # 10MB
backup_count: int = 5,
) -> logging.Logger:
"""
Set up logging for the AWS MCP Server
Args:
level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
format_string: Custom format string for log messages
include_timestamp: Whether to include timestamp in logs
log_file: Path to log file (if None, only console logging)
max_bytes: Maximum size of log file before rotation
backup_count: Number of backup files to keep
Returns:
Configured logger instance
"""
# Create logger
logger = logging.getLogger("aws_mcp_server")
logger.setLevel(getattr(logging, level.upper()))
# Remove existing handlers to avoid duplicates
for handler in logger.handlers[:]:
logger.removeHandler(handler)
# Create formatter
if format_string is None:
if include_timestamp:
format_string = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
else:
format_string = "%(levelname)s - %(message)s"
formatter = logging.Formatter(format_string)
# Create console handler
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(getattr(logging, level.upper()))
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
# Create file handler if log_file is specified
if log_file:
from logging.handlers import RotatingFileHandler
# Create log directory if it doesn't exist
log_path = Path(log_file)
log_path.parent.mkdir(parents=True, exist_ok=True)
file_handler = RotatingFileHandler(
log_file, maxBytes=max_bytes, backupCount=backup_count
)
file_handler.setLevel(getattr(logging, level.upper()))
# Always use timestamp format for file logging
file_formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
file_handler.setFormatter(file_formatter)
logger.addHandler(file_handler)
# Prevent propagation to root logger
logger.propagate = False
return logger
def get_logger(name: str = "aws_mcp_server") -> logging.Logger:
"""
Get a logger instance
Args:
name: Logger name (defaults to aws_mcp_server)
Returns:
Logger instance
"""
return logging.getLogger(name)
