Update PeriodicTasks on bulk task enable/disable in admin. (#492)

Author: bartdag

djcelery/admin.py

@@ -23,6 +23,7 @@
 from .models import (
     TaskState, WorkerState,
     PeriodicTask, IntervalSchedule, CrontabSchedule,
+    PeriodicTasks
 )
 from .humanize import naturaldate
 from .utils import is_database_scheduler, make_aware
@@ -350,13 +351,20 @@ class PeriodicTaskAdmin(admin.ModelAdmin):
     actions = ['enable_tasks',
                'disable_tasks']
 
+    def update_periodic_tasks(self):
+        dummy_periodic_task = PeriodicTask()
+        dummy_periodic_task.no_changes = False
+        PeriodicTasks.changed(dummy_periodic_task)
+
     @action(_('Enable selected periodic tasks'))
     def enable_tasks(self, request, queryset):
         queryset.update(enabled=True)
+        self.update_periodic_tasks()
 
     @action(_('Disable selected periodic tasks'))
     def disable_tasks(self, request, queryset):
         queryset.update(enabled=False)
+        self.update_periodic_tasks()
 
     def changelist_view(self, request, extra_context=None):
         extra_context = extra_context or {}

djcelery/tests/test_admin.py

@@ -4,7 +4,9 @@
 from django.test import RequestFactory, TestCase
 
 from djcelery.admin import PeriodicTaskAdmin
-from djcelery.models import PeriodicTask, IntervalSchedule, PERIOD_CHOICES
+from djcelery.models import (
+    PeriodicTask, IntervalSchedule, PERIOD_CHOICES, PeriodicTasks
+)
 
 
 class MockRequest(object):
@@ -52,8 +54,11 @@ def test_enable_tasks_should_enable_disabled_periodic_tasks(self):
         PeriodicTask.objects.create(name='Killer Queen', task='killer_queen',
                                     interval=self.interval, enabled=False),
         queryset = PeriodicTask.objects.filter(pk=1)
+        last_update = PeriodicTasks.objects.get(ident=1).last_update
         self.pt_admin.enable_tasks(request, queryset)
+        new_last_update = PeriodicTasks.objects.get(ident=1).last_update
         self.assertTrue(PeriodicTask.objects.get(pk=1).enabled)
+        self.assertNotEqual(last_update, new_last_update)
 
     def test_disable_tasks_should_disable_enabled_periodic_tasks(self):
         """