Skip to content

Commit d955947

Browse files
committed
fix: Trigger cannot be deleted
1 parent e97c36c commit d955947

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

apps/trigger/serializers/trigger.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from knowledge.serializers.common import BatchSerializer
2626
from maxkb.conf import PROJECT_DIR
2727
from tools.models import Tool
28-
from trigger.models import TriggerTypeChoices, Trigger, TriggerTaskTypeChoices, TriggerTask
28+
from trigger.models import TriggerTypeChoices, Trigger, TriggerTaskTypeChoices, TriggerTask, TaskRecord
2929

3030

3131
class BatchActiveSerializer(serializers.Serializer):
@@ -401,7 +401,7 @@ def batch_delete(self, instance: Dict, with_valid=True):
401401
self.is_valid(raise_exception=True)
402402
workspace_id = self.data.get("workspace_id")
403403
trigger_id_list = instance.get("id_list")
404-
404+
TaskRecord.objects.filter(trigger_id__in=trigger_id_list).delete()
405405
TriggerTask.objects.filter(trigger_id__in=trigger_id_list).delete()
406406
Trigger.objects.filter(workspace_id=workspace_id, id__in=trigger_id_list).delete()
407407

@@ -469,6 +469,7 @@ def edit(self, instance: Dict, with_valid=True):
469469
def delete(self):
470470
self.is_valid(raise_exception=True)
471471
trigger_id = self.data.get('trigger_id')
472+
TaskRecord.objects.filter(trigger_id=trigger_id).delete()
472473
TriggerTask.objects.filter(trigger_id=trigger_id).delete()
473474
Trigger.objects.filter(id=trigger_id).delete()
474475
return True

0 commit comments

Comments
 (0)