初步开发完成

2026-01-26 23:45:58 +08:00
parent 1fe6708e11
commit 989a0227c7
58 changed files with 150371 additions and 217 deletions
@@ -0,0 +1,116 @@
+#!/usr/bin/env python3
+"""
+创建测试用户脚本
+在数据库中创建测试用户：admin, market, other
+"""
+
+import asyncio
+import sys
+from pathlib import Path
+
+# 添加项目路径
+sys.path.insert(0, str(Path(__file__).parent))
+
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
+from sqlalchemy import text
+
+# 数据库配置
+DATABASE_URL = 'mysql+aiomysql://root:rootpassword@localhost:3306/project_manager?charset=utf8mb4'
+
+# 创建引擎
+engine = create_async_engine(DATABASE_URL, echo=False)
+AsyncSessionLocal = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+
+
+def hash_password(password: str) -> str:
+    """使用bcrypt哈希密码"""
+    import bcrypt
+    salt = bcrypt.gensalt(rounds=12)
+    hashed = bcrypt.hashpw(password.encode("utf-8"), salt)
+    return hashed.decode("utf-8")
+
+
+async def create_test_users():
+    """创建测试用户"""
+    async with AsyncSessionLocal() as session:
+        # 检查是否已有测试用户
+        result = await session.execute(text("SELECT username FROM users WHERE username IN ('market', 'other')"))
+        existing = result.fetchall()
+
+        if len(existing) >= 2:
+            print("测试用户已存在，跳过创建")
+            return
+
+        print("开始创建测试用户...")
+
+        # 检查market用户是否存在
+        result = await session.execute(text("SELECT id FROM users WHERE username = 'market'"))
+        market_exists = result.fetchone()
+
+        if not market_exists:
+            # 插入市场部用户
+            await session.execute(text("""
+                INSERT INTO users (username, password_hash, real_name, department, role, phone, is_active)
+                VALUES (:username, :password_hash, :real_name, :department, :role, :phone, :is_active)
+            """), {
+                'username': 'market',
+                'password_hash': hash_password('market123'),
+                'real_name': '市场部用户',
+                'department': '市场部',
+                'role': 'market',
+                'phone': '13800000002',
+                'is_active': True
+            })
+            print("  - 市场部用户创建成功")
+        else:
+            print("  - 市场部用户已存在")
+
+        # 检查other用户是否存在
+        result = await session.execute(text("SELECT id FROM users WHERE username = 'other'"))
+        other_exists = result.fetchone()
+
+        if not other_exists:
+            # 插入其他部门用户
+            await session.execute(text("""
+                INSERT INTO users (username, password_hash, real_name, department, role, phone, is_active)
+                VALUES (:username, :password_hash, :real_name, :department, :role, :phone, :is_active)
+            """), {
+                'username': 'other',
+                'password_hash': hash_password('other123'),
+                'real_name': '其他部门用户',
+                'department': '其他部门',
+                'role': 'other',
+                'phone': '13800000003',
+                'is_active': True
+            })
+            print("  - 其他部门用户创建成功")
+        else:
+            print("  - 其他部门用户已存在")
+
+        await session.commit()
+        print("\n✅ 测试用户创建完成！")
+        print("\n用户账号信息：")
+        print("  - admin / admin123  (系统管理员)")
+        print("  - market / market123 (市场部)")
+        print("  - other / other123  (其他部门)")
+
+
+async def main():
+    """主函数"""
+    print("="*60)
+    print("创建测试用户")
+    print("="*60)
+
+    try:
+        await create_test_users()
+    except Exception as e:
+        print(f"❌ 错误: {e}")
+        import traceback
+        traceback.print_exc()
+    finally:
+        await engine.dispose()
+
+
+if __name__ == '__main__':
+    asyncio.run(main())
+