初始项目

setup.py

@@ -0,0 +1,167 @@
+#!/usr/bin/env python3
+"""
+数据库初始化脚本
+用于创建数据库表结构
+"""
+
+import asyncio
+import sys
+import os
+
+# 添加项目根目录到Python路径
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from app.database import init_db, engine
+from app.config import settings
+from sqlalchemy import text
+
+
+async def create_database():
+    """创建数据库（如果不存在）"""
+    try:
+        # 从DATABASE_URL中提取数据库名
+        db_url = settings.database_url
+        if "postgresql+asyncpg://" in db_url:
+            # 解析数据库URL获取数据库名
+            # 格式: postgresql+asyncpg://user:password@host:port/dbname
+            import re
+            match = re.match(r'.+/([^/]+)$', db_url)
+            if match:
+                db_name = match.group(1)
+                # 创建不包含数据库名的连接URL
+                base_url = db_url.rsplit('/', 1)[0]
+                
+                # 连接到postgres默认数据库创建新数据库
+                from sqlalchemy.ext.asyncio import create_async_engine
+                temp_engine = create_async_engine(f"{base_url}/postgres")
+                
+                async with temp_engine.begin() as conn:
+                    # 检查数据库是否已存在
+                    result = await conn.execute(
+                        text("SELECT 1 FROM pg_database WHERE datname = :db_name"),
+                        {"db_name": db_name}
+                    )
+                    exists = result.scalar()
+                    
+                    if not exists:
+                        await conn.execute(text(f"CREATE DATABASE {db_name}"))
+                        print(f"✅ 数据库 '{db_name}' 创建成功")
+                    else:
+                        print(f"✅ 数据库 '{db_name}' 已存在")
+                
+                await temp_engine.dispose()
+        
+    except Exception as e:
+        print(f"❌ 创建数据库失败: {e}")
+        return False
+    
+    return True
+
+
+async def init_tables():
+    """初始化数据库表结构"""
+    try:
+        print("🔄 开始初始化数据库表结构...")
+        
+        # 初始化所有表
+        await init_db()
+        
+        print("✅ 数据库表结构初始化成功")
+        
+        # 验证表是否创建成功
+        async with engine.begin() as conn:
+            result = await conn.execute(text("""
+                SELECT table_name 
+                FROM information_schema.tables 
+                WHERE table_schema = 'public' 
+                ORDER BY table_name
+            """))
+            tables = [row[0] for row in result.fetchall()]
+            
+            print(f"📋 已创建的表: {', '.join(tables)}")
+            
+            # 检查必要的表
+            required_tables = ['callback_logs', 'external_api_logs']
+            missing_tables = [table for table in required_tables if table not in tables]
+            
+            if missing_tables:
+                print(f"❌ 缺少必要的表: {', '.join(missing_tables)}")
+                return False
+            else:
+                print("✅ 所有必要的表都已创建")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ 初始化数据库表结构失败: {e}")
+        return False
+
+
+async def verify_connection():
+    """验证数据库连接"""
+    try:
+        print("🔄 验证数据库连接...")
+        
+        async with engine.begin() as conn:
+            result = await conn.execute(text("SELECT version()"))
+            version = result.scalar()
+            print(f"✅ 数据库连接成功")
+            print(f"📊 PostgreSQL版本: {version}")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ 数据库连接失败: {e}")
+        print("\n💡 请检查以下配置:")
+        print(f"   - DATABASE_URL: {settings.database_url}")
+        print("   - PostgreSQL服务是否运行")
+        print("   - 用户名密码是否正确")
+        print("   - 网络连接是否正常")
+        return False
+
+
+async def main():
+    """主函数"""
+    print("🚀 开始数据库初始化...")
+    print(f"📋 配置信息:")
+    print(f"   - 数据库URL: {settings.database_url}")
+    print(f"   - Redis URL: {settings.redis_url}")
+    print(f"   - 应用名称: {settings.app_name}")
+    print(f"   - 外部API URL: {settings.external_api_url}")
+    print()
+    
+    # 验证连接
+    if not await verify_connection():
+        sys.exit(1)
+    
+    print()
+    
+    # 创建数据库（如果需要）
+    if not await create_database():
+        sys.exit(1)
+    
+    print()
+    
+    # 初始化表结构
+    if not await init_tables():
+        sys.exit(1)
+    
+    print()
+    print("🎉 数据库初始化完成！")
+    print("\n📝 下一步:")
+    print("   1. 配置 .env 文件中的数据库连接信息")
+    print("   2. 确保 Redis 服务正在运行")
+    print("   3. 运行应用: python main.py")
+    print(f"   4. 访问API文档: http://localhost:8000/docs")
+    print(f"   5. 接口地址: POST /ai-talk/callback/{{siteId}}")
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("\n⚠️  用户中断操作")
+        sys.exit(1)
+    except Exception as e:
+        print(f"\n❌ 初始化过程中发生错误: {e}")
+        sys.exit(1)