diff --git a/safesite/migrations/0376_troublefollowtask_usable.py b/safesite/migrations/0376_troublefollowtask_usable.py new file mode 100644 index 00000000..47362184 --- /dev/null +++ b/safesite/migrations/0376_troublefollowtask_usable.py @@ -0,0 +1,18 @@ +# Generated by Django 2.2.8 on 2021-01-27 21:23 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('safesite', '0375_auto_20210125_2120'), + ] + + operations = [ + migrations.AddField( + model_name='troublefollowtask', + name='usable', + field=models.IntegerField(default=1), + ), + ] diff --git a/safesite/models.py b/safesite/models.py index 0ca4b54a..d3aa3dcd 100644 --- a/safesite/models.py +++ b/safesite/models.py @@ -1103,6 +1103,7 @@ class TroubleFollowTask(models.Model): taskexpire = models.DateTimeField('任务过期时间', null=True, blank=True) desc = models.TextField('跟踪描述', null=True, blank=True) tasknote = models.TextField('任务执行说明', null=True, blank=True) + usable = models.IntegerField(default=1) # 1是可用,2是变更计划,0是不可用 taskset = models.ForeignKey(GridTaskSet, null=True, blank=True, verbose_name='由哪个设置派发', on_delete=models.SET_NULL) class RiskActTask(models.Model): # 风险点排查执行记录 diff --git a/safesite/tasks.py b/safesite/tasks.py index 3301cc8b..e05a8b12 100644 --- a/safesite/tasks.py +++ b/safesite/tasks.py @@ -312,6 +312,7 @@ def sendGridtask(obj): def sendGridtask2(**kwargs): obj = GridTaskSet.objects.get(id=kwargs['gridtaskset']) if obj.trouble: + TroubleFollowTask.objects.filter(taskset=obj, usable=1).update(usable=0) nowtime = datetime.datetime.now() if obj.expire: newm = nowtime + datetime.timedelta(hours=int(obj.expire)) @@ -338,6 +339,11 @@ def sendGridtask2(**kwargs): def expireRiskacttask(): RiskActTask.objects.filter(taskexpire__lte = datetime.datetime.now(), usable=1).update(usable=0) + +@shared_task +def expireTroublefollowtask(): + TroubleFollowTask.objects.filter(taskexpire__lte = datetime.datetime.now(), usable=1).update(usable=0) + @shared_task def closeExamtest(): # 自动关闭到期的考试 diff --git a/safesite/templates/troublefollow.html b/safesite/templates/troublefollow.html index 119d2fc1..7315aa00 100644 --- a/safesite/templates/troublefollow.html +++ b/safesite/templates/troublefollow.html @@ -24,6 +24,15 @@