修改登录,添加超级管理员权限

This commit is contained in:
dxc
2026-02-25 11:02:06 +08:00
parent 948149cd44
commit 1c3f116c50
5 changed files with 180 additions and 109 deletions

View File

@ -3,10 +3,11 @@ from app.models.system import SysUser
from app.extensions import db
from flask_jwt_extended import create_access_token
from app.utils.constants import UserRole
from datetime import timedelta # [修改点1] 引入 timedelta 用于设置过期时间
from datetime import timedelta
class AuthService:
# 硬编码的超级管理员凭证
# 硬编码的“初始”超级管理员凭证 (用于系统初始化或数据库被锁死时)
SUPER_ADMIN_USER = "IRIS"
SUPER_ADMIN_PASS = "licahk"
@ -19,20 +20,22 @@ class AuthService:
user_id = None
user_info = {}
# 1. 优先检查硬编码的超级管理员
# 1. 优先检查硬编码的超级管理员 (IRIS)
if username == AuthService.SUPER_ADMIN_USER:
if password == AuthService.SUPER_ADMIN_PASS:
# 显式指定角色为 SUPER_ADMIN
user_role = UserRole.SUPER_ADMIN
user_id = 0 # 虚拟ID
user_id = 0 # 虚拟ID区分于数据库ID
user_info = {
'username': username,
'role': user_role,
'department': 'System'
'department': 'System',
'status': 'active'
}
else:
raise ValueError("密码错误")
# 2. 如果不是 IRIS,检查数据库用户
# 2. 如果不是硬编码用户,检查数据库用户
else:
user = SysUser.query.filter_by(username=username).first()
@ -49,8 +52,7 @@ class AuthService:
user_id = user.id
user_info = user.to_dict()
# 3. 生成 Token
# [修改点2] 增加 expires_delta 参数,设置为 7 天(可根据需要修改为 days=1 或 days=30
# 3. 生成 Token (设置为7天过期)
access_token = create_access_token(
identity=user_id,
additional_claims={'role': user_role, 'username': username},
@ -65,18 +67,31 @@ class AuthService:
@staticmethod
def create_user(data, operator_role):
"""
创建新用户 (仅限管理员使用)
创建新用户
权限控制:只有超级管理员(SUPER_ADMIN) 或 主管(SUPERVISOR) 可以创建
"""
if operator_role not in [UserRole.SUPER_ADMIN, UserRole.SUPERVISOR]:
raise Exception("权限不足:只有超级管理员或主管可以创建新用户")
# 检查用户名是否重复
if SysUser.query.filter_by(username=data.get('username')).first():
raise Exception("用户名已存在")
role = data.get('role')
valid_roles = [v for k, v in UserRole.__dict__.items() if not k.startswith('__')]
# 验证角色合法性
# [核心修复] 过滤掉 __开头的属性 和 ROLE_MAP 字典,只保留字符串类型的角色定义
valid_roles = [
v for k, v in UserRole.__dict__.items()
if not k.startswith('__') and isinstance(v, str)
]
if role not in valid_roles:
raise Exception(f"角色无效,可选角色: {valid_roles}")
raise Exception(f"角色无效")
# 如果操作者只是 SUPERVISOR (主管),不允许他创建 SUPER_ADMIN (超管)
if operator_role == UserRole.SUPERVISOR and role == UserRole.SUPER_ADMIN:
raise Exception("权限不足:主管无法创建超级管理员")
email = data.get('email', '')
if email and SysUser.query.filter_by(email=email).first():
@ -99,8 +114,9 @@ class AuthService:
@staticmethod
def update_user(user_id, data, operator_role):
"""
[新增] 更新用户信息
更新用户信息
"""
# 权限校验
if operator_role not in [UserRole.SUPER_ADMIN, UserRole.SUPERVISOR]:
raise Exception("权限不足:只有超级管理员或主管可以修改用户信息")
@ -108,18 +124,30 @@ class AuthService:
if not user:
raise Exception("用户不存在")
# 1. 更新基本信息
# 1. 更新角色 (Role)
if 'role' in data:
valid_roles = [v for k, v in UserRole.__dict__.items() if not k.startswith('__')]
if data['role'] not in valid_roles:
raise Exception(f"角色无效")
user.role = data['role']
# [核心修复] 同样添加类型检查
valid_roles = [
v for k, v in UserRole.__dict__.items()
if not k.startswith('__') and isinstance(v, str)
]
new_role = data['role']
if new_role not in valid_roles:
raise Exception(f"角色无效")
# 防止主管把自己或其他用户提升为超管
if operator_role == UserRole.SUPERVISOR and new_role == UserRole.SUPER_ADMIN:
raise Exception("权限不足:主管无法分配超级管理员权限")
user.role = new_role
# 2. 更新部门
if 'department' in data:
user.department = data['department']
# 3. 更新邮箱
if 'email' in data:
# 如果修改了邮箱,且新邮箱已被其他人使用
email = data['email']
if email and email != user.email:
existing = SysUser.query.filter_by(email=email).first()
@ -127,7 +155,11 @@ class AuthService:
raise Exception("该邮箱已被其他用户使用")
user.email = email
# 2. 如果提供了密码,则重置密码;否则保持原密码
# 4. 更新状态 (Status) - 例如禁用用户
if 'status' in data:
user.status = data['status']
# 5. 更新密码
new_password = data.get('password')
if new_password and str(new_password).strip():
if len(new_password) < 6:
@ -140,14 +172,16 @@ class AuthService:
@staticmethod
def get_all_users():
"""获取所有系统用户"""
# 按照 ID 倒序排列
users = SysUser.query.order_by(SysUser.id.desc()).all()
return [user.to_dict() for user in users]
@staticmethod
def delete_user(user_id, operator_role):
"""删除用户"""
if operator_role not in [UserRole.SUPER_ADMIN, UserRole.SUPERVISOR]:
raise Exception("权限不足")
# 只有超级管理员可以执行物理删除
if operator_role != UserRole.SUPER_ADMIN:
raise Exception("权限不足:只有超级管理员可以删除用户,建议使用禁用功能")
user = SysUser.query.get(user_id)
if not user: