From 554b81b59adb7d886af222ceb4fe4382f3dcac87 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Tue, 13 May 2025 11:22:06 +0800 Subject: [PATCH] fix: update has_permissions decorator formatting for knowledge document endpoints --- apps/knowledge/views/document.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/apps/knowledge/views/document.py b/apps/knowledge/views/document.py index 3c678d2e8..b2e61b3c0 100644 --- a/apps/knowledge/views/document.py +++ b/apps/knowledge/views/document.py @@ -6,7 +6,7 @@ from rest_framework.views import APIView from common.auth import TokenAuth from common.auth.authentication import has_permissions -from common.constants.permission_constants import PermissionConstants +from common.constants.permission_constants import PermissionConstants, CompareConstants from common.result import result from knowledge.api.document import DocumentSplitAPI, DocumentBatchAPI, DocumentBatchCreateAPI, DocumentCreateAPI, \ DocumentReadAPI, DocumentEditAPI, DocumentDeleteAPI, TableDocumentCreateAPI, QaDocumentCreateAPI, \ @@ -121,10 +121,10 @@ class DocumentView(APIView): responses=DocumentSplitAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def post(self, request: Request, workspace_id: str, knowledge_id: str): split_data = {'file': request.FILES.getlist('file')} request_data = request.data @@ -262,10 +262,10 @@ class DocumentView(APIView): responses=DocumentBatchCreateAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def put(self, request: Request, workspace_id: str, knowledge_id: str): return result.success(DocumentSerializers.Batch( data={'knowledge_id': knowledge_id, 'workspace_id': workspace_id} @@ -284,10 +284,10 @@ class DocumentView(APIView): responses=DocumentBatchAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def put(self, request: Request, workspace_id: str, knowledge_id: str): return result.success(DocumentSerializers.Batch( data={'knowledge_id': knowledge_id, 'workspace_id': workspace_id} @@ -306,10 +306,10 @@ class DocumentView(APIView): responses=DocumentBatchAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def put(self, request: Request, workspace_id: str, knowledge_id: str): return result.success(DocumentSerializers.Batch( data={'workspace_id': workspace_id, 'knowledge_id': knowledge_id} @@ -327,10 +327,10 @@ class DocumentView(APIView): responses=BatchRefreshAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def put(self, request: Request, workspace_id: str, knowledge_id: str): return result.success( DocumentSerializers.Batch( @@ -350,10 +350,10 @@ class DocumentView(APIView): responses=BatchGenerateRelatedAPI.get_response(), tags=[_('Knowledge Base/Documentation')] # type: ignore ) - @has_permissions([ + @has_permissions( PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(), PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(), - ]) + ) def put(self, request: Request, workspace_id: str, knowledge_id: str): return result.success(DocumentSerializers.BatchGenerateRelated( data={'workspace_id': workspace_id, 'knowledge_id': knowledge_id}