diff --git a/api/apps/document_app.py b/api/apps/document_app.py index 2b2147579..1906f9ccd 100644 --- a/api/apps/document_app.py +++ b/api/apps/document_app.py @@ -587,10 +587,11 @@ async def rm(): @validate_request("doc_ids", "run") async def run(): req = await get_request_json() + uid = current_user.id try: def _run_sync(): for doc_id in req["doc_ids"]: - if not DocumentService.accessible(doc_id, current_user.id): + if not DocumentService.accessible(doc_id, uid): return get_json_result(data=False, message="No authorization.", code=RetCode.AUTHENTICATION_ERROR) kb_table_num_map = {} @@ -646,9 +647,10 @@ async def run(): @validate_request("doc_id", "name") async def rename(): req = await get_request_json() + uid = current_user.id try: def _rename_sync(): - if not DocumentService.accessible(req["doc_id"], current_user.id): + if not DocumentService.accessible(req["doc_id"], uid): return get_json_result(data=False, message="No authorization.", code=RetCode.AUTHENTICATION_ERROR) e, doc = DocumentService.get_by_id(req["doc_id"]) diff --git a/api/apps/file_app.py b/api/apps/file_app.py index ec535ad55..50cbd185a 100644 --- a/api/apps/file_app.py +++ b/api/apps/file_app.py @@ -247,6 +247,7 @@ def get_all_parent_folders(): async def rm(): req = await get_request_json() file_ids = req["file_ids"] + uid = current_user.id try: def _delete_single_file(file): @@ -285,14 +286,14 @@ async def rm(): return get_data_error_result(message="File or Folder not found!") if not file.tenant_id: return get_data_error_result(message="Tenant not found!") - if not check_file_team_permission(file, current_user.id): + if not check_file_team_permission(file, uid): return get_json_result(data=False, message="No authorization.", code=RetCode.AUTHENTICATION_ERROR) if file.source_type == FileSource.KNOWLEDGEBASE: continue if file.type == FileType.FOLDER.value: - _delete_folder_recursive(file, current_user.id) + _delete_folder_recursive(file, uid) continue _delete_single_file(file) diff --git a/api/apps/kb_app.py b/api/apps/kb_app.py index 7a57ab949..c746af574 100644 --- a/api/apps/kb_app.py +++ b/api/apps/kb_app.py @@ -269,7 +269,8 @@ async def list_kbs(): @validate_request("kb_id") async def rm(): req = await get_request_json() - if not KnowledgebaseService.accessible4deletion(req["kb_id"], current_user.id): + uid = current_user.id + if not KnowledgebaseService.accessible4deletion(req["kb_id"], uid): return get_json_result( data=False, message='No authorization.', @@ -277,7 +278,7 @@ async def rm(): ) try: kbs = KnowledgebaseService.query( - created_by=current_user.id, id=req["kb_id"]) + created_by=uid, id=req["kb_id"]) if not kbs: return get_json_result( data=False, message='Only owner of dataset authorized for this operation.',