feat: dpm checkwork增加字段

apps/dpm/migrations/0002_auto_20240612_1042.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.2.12 on 2024-06-12 02:42
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('dpm', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='checkwork',
+            name='usable',
+            field=models.BooleanField(default=True, verbose_name='是否可用'),
+        ),
+        migrations.AlterField(
+            model_name='checkwork',
+            name='time_end',
+            field=models.DateTimeField(blank=True, null=True, verbose_name='关闭时间'),
+        ),
+    ]

apps/dpm/tasks.py

@@ -4,19 +4,23 @@ from celery import shared_task
 from apps.dpm.models import CheckTaskSet, CheckWork
 from django.utils import timezone
 from datetime import timedelta
+from django.db import transaction
 
 
 @shared_task(base=CustomTask)
 def dispath_checkwork_task(checktaskset: str):
     cts = CheckTaskSet.objects.get(id=checktaskset)
+    with transaction.atomic():
+        CheckWork.objects.filter(checktaskset=cts).update(usable=False)
         cw = CheckWork()
         cw.type = 20
         cw.checktaskset = cts
-    cw.name = '风险点检查(自动派发)'
         now = timezone.now()
         cw.time_start = now
-    expire_hour = cts.expire if cts.expire else 24
+        local_time = timezone.localtime(now)
-    cw.time_end = now + timedelta(hours=expire_hour)
+        cw.name = f'风险点排查_{local_time.strftime('%Y%m%d%H%M%S')}'
+        if cts.expire:
+            cw.time_end = now + timedelta(hours=cts.expire)
         cw.user_duty = cts.user_duty
         cw.riskpoint = cts.riskpoint
         cw.note = cts.note