"""Logging configuration for the Beijing Car Quota Lottery MCP Server."""
import logging
import sys
from pathlib import Path
from typing import Optional
def setup_logging(
level: str = "INFO",
log_file: Optional[Path] = None,
format_string: Optional[str] = None
) -> None:
"""
Setup logging configuration for the application.
Args:
level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
log_file: Optional file path to write logs to
format_string: Custom format string for log messages
"""
if format_string is None:
format_string = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
# Configure root logger
logging.basicConfig(
level=getattr(logging, level.upper()),
format=format_string,
handlers=[]
)
# Create console handler
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(getattr(logging, level.upper()))
console_formatter = logging.Formatter(format_string)
console_handler.setFormatter(console_formatter)
# Add console handler to root logger
root_logger = logging.getLogger()
root_logger.addHandler(console_handler)
# Add file handler if specified
if log_file:
log_file.parent.mkdir(parents=True, exist_ok=True)
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(getattr(logging, level.upper()))
file_formatter = logging.Formatter(format_string)
file_handler.setFormatter(file_formatter)
root_logger.addHandler(file_handler)
# Set specific logger levels
logging.getLogger("uvicorn").setLevel(logging.INFO)
logging.getLogger("fastapi").setLevel(logging.INFO)
logging.getLogger("httpx").setLevel(logging.WARNING)
logging.getLogger("crawl4ai").setLevel(logging.WARNING)