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

feat: refactor shared tool API to utilize ToolSerializer for create, update, and import operations

Browse Source
This commit is contained in:
CaptainB 2025-06-05 11:10:25 +08:00
parent f5bc003aaf
commit 5cb7fc8d9a
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -210,7 +210,7 @@ class ToolSerializer(serializers.Serializer):
user_id=self.data.get('user_id'), user_id=self.data.get('user_id'),
input_field_list=instance.get('input_field_list', []), input_field_list=instance.get('input_field_list', []),
init_field_list=instance.get('init_field_list', []), init_field_list=instance.get('init_field_list', []),
scope=ToolScope.WORKSPACE, scope=instance.get('scope', ToolScope.WORKSPACE),
folder_id=instance.get('folder_id', 'root'), folder_id=instance.get('folder_id', 'root'),
is_active=False) is_active=False)
tool.save() tool.save()
@ -371,7 +371,7 @@ class ToolSerializer(serializers.Serializer):
# #
@transaction.atomic @transaction.atomic
def import_(self): def import_(self, scope=ToolScope.WORKSPACE):
self.is_valid() self.is_valid()
user_id = self.data.get('user_id') user_id = self.data.get('user_id')
@ -389,7 +389,7 @@ class ToolSerializer(serializers.Serializer):
user_id=user_id, user_id=user_id,
input_field_list=tool.get('input_field_list'), input_field_list=tool.get('input_field_list'),
init_field_list=tool.get('init_field_list', []), init_field_list=tool.get('init_field_list', []),
scope=ToolScope.WORKSPACE, scope=scope,
is_active=False is_active=False
) )
tool_model.save() tool_model.save()

5
apps/tools/views/tool.py
View File

@ -10,6 +10,7 @@ from common.constants.permission_constants import PermissionConstants
from common.result import result from common.result import result
from tools.api.tool import ToolCreateAPI, ToolEditAPI, ToolReadAPI, ToolDeleteAPI, ToolTreeReadAPI, ToolDebugApi, \ from tools.api.tool import ToolCreateAPI, ToolEditAPI, ToolReadAPI, ToolDeleteAPI, ToolTreeReadAPI, ToolDebugApi, \
ToolExportAPI, ToolImportAPI, ToolPageAPI, PylintAPI, EditIconAPI ToolExportAPI, ToolImportAPI, ToolPageAPI, PylintAPI, EditIconAPI
from tools.models import ToolScope
from tools.serializers.tool import ToolSerializer, ToolTreeSerializer from tools.serializers.tool import ToolSerializer, ToolTreeSerializer
@ -30,7 +31,7 @@ class ToolView(APIView):
def post(self, request: Request, workspace_id: str): def post(self, request: Request, workspace_id: str):
return result.success(ToolSerializer.Create( return result.success(ToolSerializer.Create(
data={'user_id': request.user.id, 'workspace_id': workspace_id} data={'user_id': request.user.id, 'workspace_id': workspace_id}
).insert(request.data)) ).insert({**request.data, 'scope': ToolScope.WORKSPACE}))
@extend_schema( @extend_schema(
methods=['GET'], methods=['GET'],
@ -155,7 +156,7 @@ class ToolView(APIView):
def post(self, request: Request, workspace_id: str): def post(self, request: Request, workspace_id: str):
return result.success(ToolSerializer.Import( return result.success(ToolSerializer.Import(
data={'workspace_id': workspace_id, 'file': request.FILES.get('file'), 'user_id': request.user.id} data={'workspace_id': workspace_id, 'file': request.FILES.get('file'), 'user_id': request.user.id}
).import_()) ).import_(ToolScope.WORKSPACE))
class Export(APIView): class Export(APIView):
authentication_classes = [TokenAuth] authentication_classes = [TokenAuth]