feat: Batch operate

Author: zhanweizhang7

apps/trigger/api/trigger.py

@@ -11,7 +11,8 @@
 from rest_framework import serializers
 
 from common.mixins.api_mixin import APIMixin
-from trigger.serializers.trigger import TriggerCreateRequest, TriggerResponse
+from knowledge.serializers.common import BatchSerializer
+from trigger.serializers.trigger import TriggerCreateRequest, TriggerResponse, BatchActiveSerializer
 
 
 class TriggerCreateAPI(APIMixin):
@@ -35,6 +36,27 @@ def get_request():
     def get_response():
         return TriggerResponse
 
+class TriggerBatchDeleteAPI(APIMixin):
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="workspace_id",
+                description="工作空间id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            )
+        ]
+    @staticmethod
+    def get_request():
+        return BatchSerializer
+
+class TriggerBatchActiveAPI(APIMixin):
+    @staticmethod
+    def get_request():
+        return BatchActiveSerializer
+
 
 class TriggerOperateAPI(APIMixin):
     @staticmethod

apps/trigger/serializers/trigger.py

@@ -20,10 +20,15 @@
 from common.db.search import page_search
 from common.exception.app_exception import AppApiException
 from common.field.common import ObjectField
+from knowledge.serializers.common import BatchSerializer
 from tools.models import Tool
 from trigger.models import TriggerTypeChoices, Trigger, TriggerTaskTypeChoices, TriggerTask
 
 
+class BatchActiveSerializer(serializers.Serializer):
+    id_list = serializers.ListField(required=True, child=serializers.UUIDField(required=True), label=_('id list'))
+    is_active = serializers.BooleanField(required=True, label=_("is_active"))
+
 class InputField(serializers.Serializer):
     source = serializers.CharField(required=True, label=_("source"), validators=[
         validators.RegexValidator(regex=re.compile("^custom|reference$"),
@@ -65,7 +70,7 @@ def validate_user_input_field_list(self, value):
 class ToolTaskParameterSerializer(serializers.Serializer):
     input_field_list = serializers.JSONField(required=False)
 
-    def validate_input_field_list(self,value):
+    def validate_input_field_list(self, value):
         if not value:
             return value
         if not isinstance(value, dict):
@@ -315,6 +320,36 @@ def to_trigger_task_model(trigger_id: str, task_data: dict):
             meta=task_data.get('meta', {})
         )
 
+    class Batch(serializers.Serializer):
+        workspace_id = serializers.CharField(required=True, label=_('workspace id'))
+        user_id = serializers.UUIDField(required=True, label=_("User ID"))
+
+        def is_valid(self, *, raise_exception=False):
+            super().is_valid(raise_exception=True)
+
+        @transaction.atomic
+        def batch_delete(self, instance: Dict, with_valid=True):
+            if with_valid:
+                BatchSerializer(data=instance).is_valid(model=Trigger, raise_exception=True)
+                self.is_valid(raise_exception=True)
+            workspace_id = self.data.get("workspace_id")
+            trigger_id_list = instance.get("id_list")
+
+            TriggerTask.objects.filter(trigger_id__in=trigger_id_list).delete()
+            Trigger.objects.filter(workspace_id=workspace_id, id__in=trigger_id_list).delete()
+
+            return True
+
+        def batch_switch(self, instance: Dict, with_valid=True):
+            if with_valid:
+                BatchSerializer(data=instance).is_valid(model=Trigger, raise_exception=True)
+                self.is_valid(raise_exception=True)
+            workspace_id = self.data.get("workspace_id")
+            trigger_id_list = instance.get("id_list")
+            is_active = instance.get("is_active")
+            Trigger.objects.filter(workspace_id=workspace_id, id__in=trigger_id_list).update(is_active=is_active)
+
+            return True
 
 class TriggerOperateSerializer(serializers.Serializer):
     trigger_id = serializers.UUIDField(required=True, label=_('trigger id'))
@@ -367,6 +402,7 @@ def edit(self, instance: Dict, with_valid=True):
     def delete(self):
         self.is_valid(raise_exception=True)
         trigger_id = self.data.get('trigger_id')
+        TriggerTask.objects.filter(trigger_id=trigger_id).delete()
         Trigger.objects.filter(id=trigger_id).delete()
         return True
 

apps/trigger/urls.py

@@ -15,7 +15,9 @@
 
 urlpatterns = [
     path('workspace/<str:workspace_id>/trigger', views.TriggerView.as_view(), name='trigger'),
-    path('workspace/<str:workspace_id>/trigger/<str:trigger_id>', views.TriggerView.Operate.as_view(), name='trigger'),
+    path('workspace/<str:workspace_id>/trigger/<str:trigger_id>', views.TriggerView.Operate.as_view(), name='trigger operate'),
+    path('workspace/<str:workspace_id>/trigger/batch_delete', views.TriggerView.BatchDelete.as_view(), name='delete batch'),
+    path('workspace/<str:workspace_id>/trigger/batch_activate', views.TriggerView.BatchActivate.as_view(), name='activate batch'),
     path('workspace/<str:workspace_id>/trigger/<int:current_page>/<int:page_size>', views.TriggerView.Page.as_view(),
          name='trigger_page'),
     path('workspace/<str:workspace_id>/task', views.TriggerTaskView.as_view(), name='task'),

apps/trigger/views/trigger.py

@@ -8,19 +8,16 @@
 """
 from django.utils.translation import gettext_lazy as _
 from drf_spectacular.utils import extend_schema
-from rest_framework import serializers
 from rest_framework.request import Request
 from rest_framework.views import APIView
 
 from application.api.application_api import ApplicationCreateAPI
 from common import result
 from common.auth import TokenAuth
-from trigger.serializers.trigger import TriggerSerializer, TriggerQuerySerializer, TriggerOperateSerializer
-from common.auth.authentication import has_permissions, get_is_permissions
-from common.constants.permission_constants import PermissionConstants, RoleConstants, ViewPermission, CompareConstants
-from common.log.log import log
-from tools.api.tool import GetInternalToolAPI
-from trigger.api.trigger import TriggerCreateAPI, TriggerOperateAPI, TriggerEditAPI
+from trigger.serializers.trigger import TriggerQuerySerializer, TriggerOperateSerializer
+
+from trigger.api.trigger import TriggerCreateAPI, TriggerOperateAPI, TriggerEditAPI, TriggerBatchDeleteAPI, \
+    TriggerBatchActiveAPI
 from trigger.serializers.trigger import TriggerSerializer
 
 
@@ -102,7 +99,41 @@ def delete(self, request: Request, workspace_id: str, trigger_id: str):
                 data={'trigger_id': trigger_id, 'workspace_id': workspace_id, 'user_id': request.user.id}
             ).delete())
 
+    class BatchDelete(APIView):
+        authentication_classes = [TokenAuth]
+
+        @extend_schema(
+            methods=['PUT'],
+            description=_('Delete trigger in batches'),
+            summary=_('Delete trigger in batches'),
+            operation_id=_('Delete trigger in batches'),  # type: ignore
+            parameters=TriggerBatchDeleteAPI.get_parameters(),
+            request=TriggerBatchDeleteAPI.get_request(),
+            responses=result.DefaultResultSerializer,
+            tags=[_('Trigger')]  # type: ignore
+        )
+        def put(self, request: Request, workspace_id: str):
+            return result.success(TriggerSerializer.Batch(
+                data={'workspace_id': workspace_id, 'user_id': request.user.id}
+            ).batch_delete(request.data))
 
+    class BatchActivate(APIView):
+        authentication_classes = [TokenAuth]
+
+        @extend_schema(
+            methods=['PUT'],
+            description=_('Activate trigger in batches'),
+            summary=_('Activate trigger in batches'),
+            operation_id=_('Activate trigger in batches'),  # type: ignore
+            parameters=TriggerBatchDeleteAPI.get_parameters(),
+            request=TriggerBatchActiveAPI.get_request(),
+            responses=result.DefaultResultSerializer,
+            tags=[_('Trigger')]  # type: ignore
+        )
+        def put(self, request: Request, workspace_id: str):
+            return result.success(TriggerSerializer.Batch(
+                data={'workspace_id': workspace_id, 'user_id': request.user.id}
+            ).batch_switch(request.data))
 
     class Page(APIView):
         authentication_classes = [TokenAuth]