初始化

src/common/database/sqlalchemy_init.py

@@ -0,0 +1,158 @@
+"""SQLAlchemy数据库初始化模块
+
+替换Peewee的数据库初始化逻辑
+提供统一的数据库初始化接口
+"""
+
+from typing import Optional
+from sqlalchemy.exc import SQLAlchemyError
+from src.common.logger import get_logger
+from src.common.database.sqlalchemy_models import (
+    Base, get_engine, get_session, initialize_database
+)
+
+logger = get_logger("sqlalchemy_init")
+
+
+def initialize_sqlalchemy_database() -> bool:
+    """
+    初始化SQLAlchemy数据库
+    创建所有表结构
+    
+    Returns:
+        bool: 初始化是否成功
+    """
+    try:
+        logger.info("开始初始化SQLAlchemy数据库...")
+        
+        # 初始化数据库引擎和会话
+        engine, session_local = initialize_database()
+        
+        if engine is None:
+            logger.error("数据库引擎初始化失败")
+            return False
+        
+        logger.info("SQLAlchemy数据库初始化成功")
+        return True
+        
+    except SQLAlchemyError as e:
+        logger.error(f"SQLAlchemy数据库初始化失败: {e}")
+        return False
+    except Exception as e:
+        logger.error(f"数据库初始化过程中发生未知错误: {e}")
+        return False
+
+
+def create_all_tables() -> bool:
+    """
+    创建所有数据库表
+    
+    Returns:
+        bool: 创建是否成功
+    """
+    try:
+        logger.info("开始创建数据库表...")
+        
+        engine = get_engine()
+        if engine is None:
+            logger.error("无法获取数据库引擎")
+            return False
+        
+        # 创建所有表
+        Base.metadata.create_all(bind=engine)
+        
+        logger.info("数据库表创建成功")
+        return True
+        
+    except SQLAlchemyError as e:
+        logger.error(f"创建数据库表失败: {e}")
+        return False
+    except Exception as e:
+        logger.error(f"创建数据库表过程中发生未知错误: {e}")
+        return False
+
+
+def check_database_connection() -> bool:
+    """
+    检查数据库连接是否正常
+    
+    Returns:
+        bool: 连接是否正常
+    """
+    try:
+        session = get_session()
+        if session is None:
+            logger.error("无法获取数据库会话")
+            return False
+        
+        # 检查会话是否可用（如果能获取到会话说明连接正常）
+        if session is None:
+            logger.error("数据库会话无效")
+            return False
+        
+        session.close()
+        
+        logger.info("数据库连接检查通过")
+        return True
+        
+    except SQLAlchemyError as e:
+        logger.error(f"数据库连接检查失败: {e}")
+        return False
+    except Exception as e:
+        logger.error(f"数据库连接检查过程中发生未知错误: {e}")
+        return False
+
+
+def get_database_info() -> Optional[dict]:
+    """
+    获取数据库信息
+    
+    Returns:
+        dict: 数据库信息字典，包含引擎信息等
+    """
+    try:
+        engine = get_engine()
+        if engine is None:
+            return None
+        
+        info = {
+            'engine_name': engine.name,
+            'driver': engine.driver,
+            'url': str(engine.url).replace(engine.url.password or '', '***'),  # 隐藏密码
+            'pool_size': getattr(engine.pool, 'size', None),
+            'max_overflow': getattr(engine.pool, 'max_overflow', None),
+        }
+        
+        return info
+        
+    except Exception as e:
+        logger.error(f"获取数据库信息失败: {e}")
+        return None
+
+
+_database_initialized = False
+
+def initialize_database_compat() -> bool:
+    """
+    兼容性数据库初始化函数
+    用于替换原有的Peewee初始化代码
+    
+    Returns:
+        bool: 初始化是否成功
+    """
+    global _database_initialized
+    
+    if _database_initialized:
+        return True
+    
+    success = initialize_sqlalchemy_database()
+    if success:
+        success = create_all_tables()
+    
+    if success:
+        success = check_database_connection()
+    
+    if success:
+        _database_initialized = True
+    
+    return success