|
25 | 25 | from knowledge.serializers.common import BatchSerializer |
26 | 26 | from maxkb.conf import PROJECT_DIR |
27 | 27 | from tools.models import Tool |
28 | | -from trigger.models import TriggerTypeChoices, Trigger, TriggerTaskTypeChoices, TriggerTask |
| 28 | +from trigger.models import TriggerTypeChoices, Trigger, TriggerTaskTypeChoices, TriggerTask, TaskRecord |
29 | 29 |
|
30 | 30 |
|
31 | 31 | class BatchActiveSerializer(serializers.Serializer): |
@@ -401,7 +401,7 @@ def batch_delete(self, instance: Dict, with_valid=True): |
401 | 401 | self.is_valid(raise_exception=True) |
402 | 402 | workspace_id = self.data.get("workspace_id") |
403 | 403 | trigger_id_list = instance.get("id_list") |
404 | | - |
| 404 | + TaskRecord.objects.filter(trigger_id__in=trigger_id_list).delete() |
405 | 405 | TriggerTask.objects.filter(trigger_id__in=trigger_id_list).delete() |
406 | 406 | Trigger.objects.filter(workspace_id=workspace_id, id__in=trigger_id_list).delete() |
407 | 407 |
|
@@ -469,6 +469,7 @@ def edit(self, instance: Dict, with_valid=True): |
469 | 469 | def delete(self): |
470 | 470 | self.is_valid(raise_exception=True) |
471 | 471 | trigger_id = self.data.get('trigger_id') |
| 472 | + TaskRecord.objects.filter(trigger_id=trigger_id).delete() |
472 | 473 | TriggerTask.objects.filter(trigger_id=trigger_id).delete() |
473 | 474 | Trigger.objects.filter(id=trigger_id).delete() |
474 | 475 | return True |
|
0 commit comments