24024
d4b28a731a
- 新增 SystemNotification 模型，实现系统通知的数据存储 - 管理后台新增通知相关接口，支持通知的增删改查 - 用户端新增接口，获取最新激活通知并支持标记已读 - 在前端首页添加全局通知弹窗，实现通知自动轮询及已读同步 - 生成历史记录中兼容支持图片缩略图及新旧图片格式 - 优化后台图片同步逻辑，新增缩略图生成与存储 - 支持上传参考图的拖拽、粘贴、多文件上传及排序功能 - 增加购买积分页面入口及菜单项，调整菜单结构 - 日志系统由 Redis 列表迁移为有序集合，保留 30 天日志 - 优化日志页面样式，提升可读性及滚动体验 - 调整部分模板布局为自定义滚动条容器，增强视觉一致性
58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
from app import app
|
|
from extensions import db
|
|
from models import User, Role, Permission
|
|
|
|
def init_rbac():
|
|
with app.app_context():
|
|
print("🚀 正在初始化 RBAC 系统...")
|
|
|
|
# 1. 创建基础权限
|
|
perms = {
|
|
'view_logs': '查看系统日志',
|
|
'manage_rbac': '管理角色与权限',
|
|
'manage_users': '管理用户信息',
|
|
'manage_dicts': '管理系统字典',
|
|
'manage_notifications': '管理系统通知',
|
|
'manage_system': '系统最高权限'
|
|
}
|
|
perm_objs = {}
|
|
for code, desc in perms.items():
|
|
p = Permission.query.filter_by(name=code).first()
|
|
if not p:
|
|
p = Permission(name=code, description=desc)
|
|
db.session.add(p)
|
|
perm_objs[code] = p
|
|
|
|
db.session.commit()
|
|
|
|
# 2. 创建基础角色
|
|
# 超级管理员角色
|
|
admin_role = Role.query.filter_by(name='超级管理员').first()
|
|
if not admin_role:
|
|
admin_role = Role(name='超级管理员', description='系统最高权限持有者')
|
|
admin_role.permissions = list(perm_objs.values())
|
|
db.session.add(admin_role)
|
|
else:
|
|
# 确保现有超级管理员拥有所有新权限
|
|
admin_role.permissions = list(perm_objs.values())
|
|
|
|
# 普通用户角色
|
|
user_role = Role.query.filter_by(name='普通用户').first()
|
|
if not user_role:
|
|
user_role = Role(name='普通用户', description='常规功能使用者')
|
|
db.session.add(user_role)
|
|
|
|
db.session.commit()
|
|
|
|
# 3. 为现有用户分配超级管理员角色（作为测试）
|
|
# 请根据实际情况修改
|
|
first_user = User.query.first()
|
|
if first_user:
|
|
first_user.role = admin_role
|
|
db.session.commit()
|
|
print(f"✅ 已将用户 {first_user.phone} 设为超级管理员")
|
|
|
|
print("✨ RBAC 初始化完成")
|
|
|
|
if __name__ == '__main__':
|
|
init_rbac() |