suhh-知识库查询添加拥有编辑账号信息

api/controllers/console/datasets/datasets.py

@@ -222,7 +222,8 @@ class DatasetApi(Resource):
         if data.get("permission") == "partial_members":
             part_users_list = DatasetPermissionService.get_dataset_partial_member_list(dataset_id_str)
             data.update({"partial_member_list": part_users_list})
-
+        edit_user_id_list = DatasetPermissionService.get_dataset_edit_user_ids(dataset_id_str)
+        data.update({"edit_user_ids": edit_user_id_list})
         return data, 200
 
     @setup_required

api/services/dataset_service.py

@@ -2749,6 +2749,20 @@ class DatasetCollectionBindingService:
 
 class DatasetPermissionService:
     @classmethod
+    def get_dataset_edit_user_ids(cls, dataset_id):
+        edit_user_ids_query = (
+            db.session.query(
+                DatasetPermissionAll.account_id,
+            )
+            .filter(DatasetPermissionAll.dataset_id == dataset_id, DatasetPermissionAll.has_edit_permission == True)
+            .all()
+        )
+        edit_user_ids = []
+        for edit_user_id in edit_user_ids_query:
+            edit_user_ids.append(edit_user_id.account_id)
+        return edit_user_ids
+
+    @classmethod
     def get_dataset_partial_member_list(cls, dataset_id):
         user_list_query = (
             db.session.query(