73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
设置admin用户为管理员
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|||
|
|
|
|||
|
|
from app.core.database import SessionLocal
|
|||
|
|
from app.models.user import User
|
|||
|
|
from app.models.permission import Role
|
|||
|
|
|
|||
|
|
def set_admin():
|
|||
|
|
"""设置admin用户为管理员"""
|
|||
|
|
db = SessionLocal()
|
|||
|
|
try:
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("设置admin用户为管理员")
|
|||
|
|
print("=" * 60)
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 查找admin用户
|
|||
|
|
admin_user = db.query(User).filter(User.username == "admin").first()
|
|||
|
|
|
|||
|
|
if not admin_user:
|
|||
|
|
print("❌ 未找到admin用户，请先创建admin用户")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
print(f"找到用户: {admin_user.username} (ID: {admin_user.id})")
|
|||
|
|
print(f"当前角色: {admin_user.role}")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 设置role字段为admin
|
|||
|
|
admin_user.role = "admin"
|
|||
|
|
print("✅ 已将role字段设置为admin")
|
|||
|
|
|
|||
|
|
# 如果存在admin角色，也分配给用户
|
|||
|
|
admin_role = db.query(Role).filter(Role.name == "admin").first()
|
|||
|
|
if admin_role:
|
|||
|
|
# 检查用户是否已经有admin角色
|
|||
|
|
if admin_role not in admin_user.roles:
|
|||
|
|
admin_user.roles.append(admin_role)
|
|||
|
|
print("✅ 已分配admin角色给用户")
|
|||
|
|
else:
|
|||
|
|
print("ℹ️ 用户已有admin角色")
|
|||
|
|
else:
|
|||
|
|
print("ℹ️ admin角色不存在（可能需要先运行init_rbac_data.py）")
|
|||
|
|
|
|||
|
|
db.commit()
|
|||
|
|
print()
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("✅ admin用户已设置为管理员！")
|
|||
|
|
print("=" * 60)
|
|||
|
|
print()
|
|||
|
|
print("用户信息：")
|
|||
|
|
print(f" 用户名: {admin_user.username}")
|
|||
|
|
print(f" 邮箱: {admin_user.email}")
|
|||
|
|
print(f" 角色: {admin_user.role}")
|
|||
|
|
if admin_user.roles:
|
|||
|
|
print(f" RBAC角色: {', '.join([r.name for r in admin_user.roles])}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
db.rollback()
|
|||
|
|
print(f"❌ 设置失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
set_admin()
|