KCGL/inventory-backend/app/extensions.py

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from flask_cors import CORS
from flask_jwt_extended import JWTManager  # 确保引入了 JWTManager
from datetime import datetime, timezone, timedelta
import redis

# 1. 创建扩展实例（此时未绑定具体的 App）
db = SQLAlchemy()
migrate = Migrate()
cors = CORS()
jwt = JWTManager()  # 必须实例化

# Redis 客户端 (单设备登录互踢用)
redis_client = None


def beijing_time():
    """获取北京时间 (UTC+8)，剥离时区信息以兼容数据库 naive DateTime 字段"""
    return datetime.now(timezone(timedelta(hours=8))).replace(tzinfo=None)


# 2. 定义初始化函数 (供工厂函数 create_app 调用)
def init_extensions(app):
    """
    统一初始化所有 Flask 扩展
    """
    global redis_client

    # 初始化数据库
    db.init_app(app)

    # 初始化迁移工具
    migrate.init_app(app, db)

    # 初始化跨域设置 (允许 /api/* 路径被所有来源访问)
    cors.init_app(app, resources={r"/api/*": {"origins": "*"}})

    # 初始化 JWT (这一步至关重要，缺少它会导致 500 错误)
    jwt.init_app(app)

    # 初始化 Redis (单设备登录互踢)
    redis_url = app.config.get('REDIS_URL', 'redis://localhost:6379/0')
    try:
        redis_client = redis.from_url(redis_url, decode_responses=True)
        redis_client.ping()
        app.logger.info("✅ Redis connected successfully")
    except Exception as e:
        app.logger.warning(f"⚠️ Redis connection failed: {e}, single-device login will be disabled")

    # ★ 注册 SQLAlchemy 审计监听器
    # 必须在 db.init_app 之后调用，确保所有模型已映射
    try:
        from app.core.audit_listener import register_audit_listeners
        with app.app_context():
            count = register_audit_listeners(db)
            app.logger.info(f"✅ 审计监听器注册成功，共绑定 {count} 个模型")
    except Exception as e:
        app.logger.error(f"⚠️ 审计监听器注册失败: {e}")