20251222

Server/create_user.py

@@ -0,0 +1,69 @@
+import sys
+import os
+
+# Add current directory to sys.path
+sys.path.append(os.getcwd())
+
+from app.db import SessionLocal
+from app.services.auth_service import auth_service
+from app.schemas.auth import UserRegister
+from app.core.security import get_password_hash
+from app.models.user import User
+
+def create_user(username, password, email=None):
+    db = SessionLocal()
+    try:
+        # Check if user exists
+        existing = db.query(User).filter(User.username == username).first()
+        if existing:
+            print(f"❌ 用户名 '{username}' 已存在")
+            return
+
+        # Prepare registration data
+        # We bypass the code verification by not providing code, 
+        # but we set email if provided.
+        # However, auth_service.register sets is_verified=False by default if code is missing.
+        # We might want to manually set is_verified=True after registration for convenience.
+        
+        register_data = UserRegister(
+            username=username,
+            password=password,
+            confirm_password=password,
+            email=email,
+            device_id="local_script"
+        )
+        
+        # Call register service
+        try:
+            token = auth_service.register(db, register_data)
+            print(f"✅ 用户 '{username}' 注册成功！")
+            
+            # Manually verify the user for local convenience
+            user = db.query(User).filter(User.username == username).first()
+            if user:
+                user.is_verified = True
+                user.permissions = "admin" # Grant admin permissions for local test user
+                db.commit()
+                print(f"✅ 已自动验证邮箱并赋予 admin 权限")
+                
+        except Exception as e:
+            print(f"❌ 注册失败: {e}")
+            
+    finally:
+        db.close()
+
+if __name__ == "__main__":
+    if len(sys.path) < 2:
+        print("Usage: python create_user.py [username] [password]")
+    
+    username = "admin"
+    password = "password123"
+    email = "admin@example.com"
+    
+    if len(sys.argv) > 1:
+        username = sys.argv[1]
+    if len(sys.argv) > 2:
+        password = sys.argv[2]
+        
+    print(f"正在创建用户: {username} ...")
+    create_user(username, password, email)