From f7313bfcc157ff3680b03a73d0bb3dfb4b8b960e Mon Sep 17 00:00:00 2001
From: wangruidong <940853815@qq.com>
Date: Thu, 21 Nov 2024 18:42:09 +0800
Subject: [PATCH] perf: Audits job api disable periodic task

---
 apps/audits/api.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/apps/audits/api.py b/apps/audits/api.py
index d586f8e3b..0c7278476 100644
--- a/apps/audits/api.py
+++ b/apps/audits/api.py
@@ -7,6 +7,7 @@ from django.db.models import F, Value, CharField, Q
 from django.db.models.functions import Cast
 from django.http import HttpResponse, FileResponse
 from django.utils.encoding import escape_uri_path
+from django_celery_beat.models import PeriodicTask
 from rest_framework import generics
 from rest_framework import status
 from rest_framework import viewsets
@@ -74,6 +75,21 @@ class JobsAuditViewSet(OrgModelViewSet):
         queryset = queryset.exclude(type=Types.upload_file).filter(instant=False)
         return queryset
 
+    def perform_update(self, serializer):
+        job = self.get_object()
+        is_periodic = serializer.validated_data.get('is_periodic')
+        if job.is_periodic != is_periodic:
+            job.is_periodic = is_periodic
+            job.save()
+        name, task, args, kwargs = job.get_register_task()
+        task_obj = PeriodicTask.objects.filter(name=name).first()
+        if task_obj:
+            is_periodic = job.is_periodic
+            if task_obj.enabled != is_periodic:
+                task_obj.enabled = is_periodic
+                task_obj.save()
+        return super().perform_update(serializer)
+
 
 class FTPLogViewSet(OrgModelViewSet):
     model = FTPLog