zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: streamline permission handling and remove unused cache logic in tool management

Browse Source
This commit is contained in:
CaptainB 2025-07-03 17:16:16 +08:00
parent c8d901ba4d
commit f186b25afe
1 changed files with 5 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
apps/tools/serializers/tool.py
View File

@ -7,7 +7,6 @@ import re
import uuid_utils.compat as uuid import uuid_utils.compat as uuid
from django.core import validators from django.core import validators
from django.core.cache import cache
from django.db import transaction from django.db import transaction
from django.db.models import QuerySet, Q from django.db.models import QuerySet, Q
from django.http import HttpResponse from django.http import HttpResponse
@ -16,8 +15,6 @@ from pylint.lint import Run
from pylint.reporters import JSON2Reporter from pylint.reporters import JSON2Reporter
from rest_framework import serializers, status from rest_framework import serializers, status
from common.constants.cache_version import Cache_Version
from common.constants.permission_constants import ResourceAuthType, ResourcePermission
from common.database_model_manage.database_model_manage import DatabaseModelManage from common.database_model_manage.database_model_manage import DatabaseModelManage
from common.db.search import page_search, native_page_search from common.db.search import page_search, native_page_search
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -512,19 +509,11 @@ class ToolSerializer(serializers.Serializer):
tool.save() tool.save()
# 自动授权给创建者 # 自动授权给创建者
WorkspaceUserResourcePermission( UserResourcePermissionSerializer(data={
target=tool_id, 'workspace_id': self.data.get('workspace_id'),
auth_target_type=AuthTargetType.TOOL, 'user_id': self.data.get('user_id'),
permission_list=[ResourcePermission.VIEW, ResourcePermission.MANAGE], 'auth_target_type': AuthTargetType.TOOL.value
workspace_id=self.data.get('workspace_id'), }).auth_resource(str(tool_id))
user_id=self.data.get('user_id'),
auth_type=ResourceAuthType.RESOURCE_PERMISSION_GROUP
).save()
# 刷新缓存
version = Cache_Version.PERMISSION_LIST.get_version()
key = Cache_Version.PERMISSION_LIST.get_key(user_id=self.data.get('user_id'))
cache.delete(key, version=version)
return ToolModelSerializer(tool).data return ToolModelSerializer(tool).data