panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Model get in migration file

v3.2
zhangzhanwei 2025-10-30 14:25:12 +08:00 committed by zhanweizhang7
parent 543f83a822
commit 274dfdae5c
2 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
apps/system_manage/migrations/0003_alter_workspaceuserresourcepermission_target.py
View File

@ -6,21 +6,18 @@ from django.db.models import QuerySet
from common.constants.permission_constants import WorkspaceUserRoleMapping from common.constants.permission_constants import WorkspaceUserRoleMapping
from common.utils.common import group_by from common.utils.common import group_by
from application.models import ApplicationFolder
from knowledge.models import KnowledgeFolder
from tools.models import ToolFolder
from system_manage.models import WorkspaceUserResourcePermission
from users.models import User
def delete_auth(folder_model): def delete_auth(apps,folder_model):
QuerySet(WorkspaceUserResourcePermission).filter(target__in=QuerySet(folder_model).values_list('id')).delete() workspace_user_resource_permission_model = apps.get_model('system_manage', 'WorkspaceUserResourcePermission')
QuerySet(workspace_user_resource_permission_model).filter(target__in=QuerySet(folder_model).values_list('id')).delete()
def get_workspace_user_resource_permission_list(auth_target_type, workspace_user_role_mapping_model_workspace_dict, def get_workspace_user_resource_permission_list(apps, auth_target_type, workspace_user_role_mapping_model_workspace_dict,
folder_model): folder_model):
workspace_user_resource_permission_model = apps.get_model('system_manage', 'WorkspaceUserResourcePermission')
return reduce(lambda x, y: [*x, *y], [ return reduce(lambda x, y: [*x, *y], [
[WorkspaceUserResourcePermission(target=f.id, workspace_id=f.workspace_id, user_id=wurm.user_id, [workspace_user_resource_permission_model(target=f.id, workspace_id=f.workspace_id, user_id=wurm.user_id,
auth_target_type=auth_target_type, auth_type="RESOURCE_PERMISSION_GROUP", auth_target_type=auth_target_type, auth_type="RESOURCE_PERMISSION_GROUP",
permission_list=['VIEW','MANAGE'] if wurm.user_id == f.user_id else ['VIEW']) for wurm in permission_list=['VIEW','MANAGE'] if wurm.user_id == f.user_id else ['VIEW']) for wurm in
workspace_user_role_mapping_model_workspace_dict.get(f.workspace_id, [])] for f in workspace_user_role_mapping_model_workspace_dict.get(f.workspace_id, [])] for f in
@ -30,31 +27,38 @@ def get_workspace_user_resource_permission_list(auth_target_type, workspace_user
def auth_folder(apps, schema_editor): def auth_folder(apps, schema_editor):
from common.database_model_manage.database_model_manage import DatabaseModelManage from common.database_model_manage.database_model_manage import DatabaseModelManage
DatabaseModelManage.init() DatabaseModelManage.init()
user_model = apps.get_model('users', 'User')
application_folder_model = apps.get_model('application', 'ApplicationFolder')
knowledge_folder_model = apps.get_model('knowledge', 'KnowledgeFolder')
tool_folder_model = apps.get_model('tools', 'ToolFolder')
workspace_user_resource_permission_model = apps.get_model('system_manage', 'WorkspaceUserResourcePermission')
workspace_user_role_mapping_model = DatabaseModelManage.get_model("workspace_user_role_mapping") workspace_user_role_mapping_model = DatabaseModelManage.get_model("workspace_user_role_mapping")
if workspace_user_role_mapping_model is None: if workspace_user_role_mapping_model is None:
workspace_user_role_mapping_model_workspace_dict = { workspace_user_role_mapping_model_workspace_dict = {
'default': [WorkspaceUserRoleMapping('default', '', u.id) for u in QuerySet(User).all()]} 'default': [WorkspaceUserRoleMapping('default', '', u.id) for u in QuerySet(user_model).all()]}
else: else:
workspace_user_role_mapping_model_workspace_dict = group_by( workspace_user_role_mapping_model_workspace_dict = group_by(
[v for v in {str(wurm.user_id) + str(wurm.workspace_id): wurm for wurm in [v for v in {str(wurm.user_id) + str(wurm.workspace_id): wurm for wurm in
QuerySet(workspace_user_role_mapping_model)}.values()], QuerySet(workspace_user_role_mapping_model)}.values()],
lambda item: item.workspace_id) lambda item: item.workspace_id)
workspace_user_resource_permission_list = get_workspace_user_resource_permission_list("APPLICATION", workspace_user_resource_permission_list = get_workspace_user_resource_permission_list(apps,"APPLICATION",
workspace_user_role_mapping_model_workspace_dict, workspace_user_role_mapping_model_workspace_dict,
ApplicationFolder) application_folder_model)
workspace_user_resource_permission_list += get_workspace_user_resource_permission_list("TOOL", workspace_user_resource_permission_list += get_workspace_user_resource_permission_list(apps,"TOOL",
workspace_user_role_mapping_model_workspace_dict, workspace_user_role_mapping_model_workspace_dict,
ToolFolder) knowledge_folder_model)
workspace_user_resource_permission_list += get_workspace_user_resource_permission_list("KNOWLEDGE", workspace_user_resource_permission_list += get_workspace_user_resource_permission_list(apps,"KNOWLEDGE",
workspace_user_role_mapping_model_workspace_dict, workspace_user_role_mapping_model_workspace_dict,
KnowledgeFolder) tool_folder_model)
delete_auth(ApplicationFolder) delete_auth(apps,application_folder_model)
delete_auth(ToolFolder) delete_auth(apps,knowledge_folder_model)
delete_auth(KnowledgeFolder) delete_auth(apps,tool_folder_model)
QuerySet(WorkspaceUserResourcePermission).bulk_create(workspace_user_resource_permission_list) QuerySet(workspace_user_resource_permission_model).bulk_create(workspace_user_resource_permission_list)
class Migration(migrations.Migration): class Migration(migrations.Migration):

2
ui/src/views/application-overview/component/LimitDialog.vue
View File

@ -14,7 +14,7 @@
v-model="form.access_num" v-model="form.access_num"
:min="0" :min="0"
:step="1" :step="1"
:max="10000" :max="10000000"
:value-on-clear="0" :value-on-clear="0"
controls-position="right" controls-position="right"
style="width: 268px" style="width: 268px"