init

src/database/database.py

@@ -0,0 +1,99 @@
+# database/database.py
+import sys
+import os
+from pathlib import Path
+
+# 添加项目根目录到Python路径
+project_root = Path(__file__).parent.parent.parent
+sys.path.insert(0, str(project_root))
+
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
+from sqlalchemy import text
+from sqlalchemy.orm import sessionmaker
+from config.settings import settings
+import asyncio
+from utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+class DatabaseManager:
+    """数据库管理器"""
+    
+    def __init__(self):
+        self.engine = None
+        self.async_session = None
+        self.is_connected = False
+    
+    async def connect(self):
+        """连接数据库"""
+        if not settings.DATABASE_URL:
+            logger.warning("未配置数据库连接，跳过数据库初始化")
+            self.is_connected = False
+            return
+            
+        try:
+            # 创建异步引擎
+            self.engine = create_async_engine(
+                settings.DATABASE_URL,
+                echo=settings.DEBUG,
+                pool_size=20,
+                max_overflow=30,
+                pool_recycle=3600
+            )
+            
+            # 创建异步会话工厂
+            self.async_session = async_sessionmaker(
+                self.engine,
+                class_=AsyncSession,
+                expire_on_commit=False
+            )
+            
+            # 测试连接
+            async with self.engine.begin() as conn:
+                await conn.execute(text("SELECT 1"))
+            
+            self.is_connected = True
+            logger.info("数据库连接成功")
+            
+        except Exception as e:
+            logger.error(f"数据库连接失败: {e}")
+            self.is_connected = False
+            raise
+    
+    async def disconnect(self):
+        """断开数据库连接"""
+        if self.engine:
+            await self.engine.dispose()
+            self.is_connected = False
+            logger.info("数据库连接已断开")
+    
+    async def get_session(self) -> AsyncSession:
+        """获取数据库会话"""
+        if not self.is_connected:
+            await self.connect()
+        
+        return self.async_session()
+    
+    async def create_tables(self):
+        """创建数据库表"""
+        from models.database import Base
+        
+        try:
+            async with self.engine.begin() as conn:
+                await conn.run_sync(Base.metadata.create_all)
+            logger.info("数据库表创建成功")
+        except Exception as e:
+            logger.error(f"数据库表创建失败: {e}")
+            raise
+
+# 全局数据库管理器实例
+db_manager = DatabaseManager()
+
+# 数据库依赖注入
+async def get_db_session():
+    """获取数据库会话的依赖函数"""
+    session = await db_manager.get_session()
+    try:
+        yield session
+    finally:
+        await session.close()

src/database/init_db.py

@@ -0,0 +1,25 @@
+# database/init_db.py
+import asyncio
+from database.database import db_manager
+from utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+async def init_database():
+    """初始化数据库"""
+    try:
+        # 连接数据库
+        await db_manager.connect()
+        
+        # 创建表
+        await db_manager.create_tables()
+        
+        logger.info("数据库初始化完成")
+        return True
+        
+    except Exception as e:
+        logger.error(f"数据库初始化失败: {e}")
+        return False
+
+if __name__ == "__main__":
+    asyncio.run(init_database())

src/database/migrate_db.py

@@ -0,0 +1,67 @@
+"""数据库迁移脚本 - 删除旧表并重新创建"""
+import asyncio
+import sys
+import os
+from pathlib import Path
+
+# 添加项目根目录到路径
+project_root = Path(__file__).parent.parent.parent
+sys.path.insert(0, str(project_root))
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+# 设置环境变量确保正确导入
+os.environ['PYTHONPATH'] = str(project_root) + os.pathsep + str(Path(__file__).parent.parent)
+
+from src.database.database import db_manager
+from src.models.database import Base
+from src.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+async def migrate_database():
+    """迁移数据库：删除所有表并重新创建"""
+    try:
+        # 连接数据库
+        await db_manager.connect()
+
+        # 删除所有表
+        logger.info("正在删除所有数据库表...")
+        async with db_manager.engine.begin() as conn:
+            await conn.run_sync(Base.metadata.drop_all)
+
+        # 重新创建所有表
+        logger.info("正在创建所有数据库表...")
+        async with db_manager.engine.begin() as conn:
+            await conn.run_sync(Base.metadata.create_all)
+
+        logger.info("数据库迁移完成！")
+
+        return True
+
+    except Exception as e:
+        logger.error(f"数据库迁移失败: {e}")
+        return False
+    finally:
+        await db_manager.disconnect()
+
+
+if __name__ == "__main__":
+    import sys
+
+    # 检查命令行参数
+    if len(sys.argv) > 1 and sys.argv[1] == '--force':
+        confirm = 'yes'
+    else:
+        print("=== 数据库迁移 ===")
+        print("警告：这将删除所有数据库表和数据！")
+        confirm = input("确认继续？(yes/no): ")
+
+    if confirm.lower() == 'yes':
+        asyncio.run(migrate_database())
+    else:
+        print("已取消迁移")
+
+
+
+