From 56a2ae4b2048bd903a978e63a132b713511c638b Mon Sep 17 00:00:00 2001 From: GuoQing Zhang Date: Wed, 3 Jul 2024 17:03:50 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E9=BB=98=E8=AE=A4=E7=BB=99=E6=99=AE?= =?UTF-8?q?=E9=80=9A=E7=94=A8=E6=88=B7=E6=8E=88=E6=9D=83=E4=B8=80=E4=BA=9B?= =?UTF-8?q?=E8=8F=9C=E5=8D=95=E6=A0=8F=E7=9A=84=E6=9F=A5=E7=9C=8B=E6=9D=83?= =?UTF-8?q?=E9=99=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/bisheng/database/base.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/backend/bisheng/database/base.py b/src/backend/bisheng/database/base.py index 5945aeb11..336c8b65d 100644 --- a/src/backend/bisheng/database/base.py +++ b/src/backend/bisheng/database/base.py @@ -27,6 +27,7 @@ def init_default_data(): from bisheng.database.models.flow_version import FlowVersion from bisheng.database.models.user_role import UserRoleDao from bisheng.database.models.group import Group, DefaultGroup + from bisheng.database.models.role_access import RoleAccess, AccessType if redis_client.setNx('init_default_data', '1'): try: @@ -35,12 +36,19 @@ def init_default_data(): db_role = session.exec(select(Role).limit(1)).all() if not db_role: # 初始化系统配置, 管理员拥有所有权限 - db_role = Role(id=AdminRole, role_name='系统管理员', remark='系统所有权限管理员', group_id=DefaultGroup) + db_role = Role(id=AdminRole, role_name='系统管理员', remark='系统所有权限管理员', + group_id=DefaultGroup) session.add(db_role) - db_role_normal = Role(id=DefaultRole, role_name='普通用户', remark='默认用户', group_id=DefaultGroup) + db_role_normal = Role(id=DefaultRole, role_name='普通用户', remark='默认用户', + group_id=DefaultGroup) session.add(db_role_normal) + # 给普通用户赋予 构建、知识、模型菜单栏的查看权限 + session.add_all([ + RoleAccess(role_id=DefaultRole, type=AccessType.WEB_MENU.value, third_id='build'), + RoleAccess(role_id=DefaultRole, type=AccessType.WEB_MENU.value, third_id='knowledge'), + RoleAccess(role_id=DefaultRole, type=AccessType.WEB_MENU.value, third_id='model'), + ]) session.commit() - # 添加默认用户组 group = session.exec(select(Group).limit(1)).all() if not group: