import logging
import logging.handlers
import os
from datetime import datetime
from pathlib import Path

def setup_logging():
    """配置日志系统"""
    # 创建logs目录
    log_dir = Path("logs")
    log_dir.mkdir(exist_ok=True)

    # 配置根日志记录器
    root_logger = logging.getLogger()
    root_logger.setLevel(logging.INFO)

    # 清除现有的处理器
    for handler in root_logger.handlers[:]:
        root_logger.removeHandler(handler)

    # 创建格式化器
    formatter = logging.Formatter(
        '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
        datefmt='%Y-%m-%d %H:%M:%S'
    )

    # 1. 控制台处理器
    console_handler = logging.StreamHandler()
    console_handler.setLevel(logging.INFO)
    console_handler.setFormatter(formatter)
    root_logger.addHandler(console_handler)

    # 2. 应用日志文件处理器（按日期滚动）
    app_log_file = log_dir / "app.log"
    app_handler = logging.handlers.TimedRotatingFileHandler(
        app_log_file,
        when='midnight',
        interval=1,
        backupCount=30,
        encoding='utf-8'
    )
    app_handler.setLevel(logging.INFO)
    app_handler.setFormatter(formatter)
    root_logger.addHandler(app_handler)

    # 3. 错误日志文件处理器
    error_log_file = log_dir / "error.log"
    error_handler = logging.handlers.TimedRotatingFileHandler(
        error_log_file,
        when='midnight',
        interval=1,
        backupCount=30,
        encoding='utf-8'
    )
    error_handler.setLevel(logging.ERROR)
    error_handler.setFormatter(formatter)
    root_logger.addHandler(error_handler)

    # 4. 任务调度器专用日志处理器
    scheduler_log_file = log_dir / "scheduler.log"
    scheduler_handler = logging.handlers.TimedRotatingFileHandler(
        scheduler_log_file,
        when='midnight',
        interval=1,
        backupCount=30,
        encoding='utf-8'
    )
    scheduler_handler.setLevel(logging.INFO)
    scheduler_handler.setFormatter(formatter)

    # 为调度器设置专用logger
    scheduler_logger = logging.getLogger('apscheduler')
    scheduler_logger.addHandler(scheduler_handler)
    scheduler_logger.setLevel(logging.INFO)

    # 为应用调度器模块设置专用logger
    app_scheduler_logger = logging.getLogger('app.utils.scheduler')
    app_scheduler_logger.addHandler(scheduler_handler)
    app_scheduler_logger.setLevel(logging.INFO)

    # 5. API访问日志处理器
    access_log_file = log_dir / "access.log"
    access_handler = logging.handlers.TimedRotatingFileHandler(
        access_log_file,
        when='midnight',
        interval=1,
        backupCount=30,
        encoding='utf-8'
    )
    access_formatter = logging.Formatter(
        '%(asctime)s - %(message)s',
        datefmt='%Y-%m-%d %H:%M:%S'
    )
    access_handler.setFormatter(access_formatter)

    # 为uvicorn访问日志设置处理器
    uvicorn_access_logger = logging.getLogger("uvicorn.access")
    uvicorn_access_logger.addHandler(access_handler)
    uvicorn_access_logger.setLevel(logging.INFO)

    # 6. 数据库日志处理器
    db_log_file = log_dir / "database.log"
    db_handler = logging.handlers.TimedRotatingFileHandler(
        db_log_file,
        when='midnight',
        interval=1,
        backupCount=30,
        encoding='utf-8'
    )
    db_handler.setLevel(logging.INFO)
    db_handler.setFormatter(formatter)

    # 为SQLAlchemy设置日志
    sqlalchemy_logger = logging.getLogger('sqlalchemy.engine')
    sqlalchemy_logger.addHandler(db_handler)
    sqlalchemy_logger.setLevel(logging.WARNING)  # 只记录警告和错误

    print(f"日志系统已配置，日志文件保存在: {log_dir.absolute()}")
    print("日志文件:")
    print(f"  - 应用日志: {app_log_file}")
    print(f"  - 错误日志: {error_log_file}")
    print(f"  - 调度器日志: {scheduler_log_file}")
    print(f"  - 访问日志: {access_log_file}")
    print(f"  - 数据库日志: {db_log_file}")

def get_logger(name: str = None):
    """获取日志记录器"""
    return logging.getLogger(name or __name__)