Add natural key support to the PeriodicTask models. Needed for dumpdata/loaddata to work properly. (#552)

Signed-off-by: Roberto Rosario <[email protected]>

Author: siloraptor

djcelery/managers.py

@@ -129,6 +129,9 @@ class PeriodicTaskManager(ExtendedManager):
     def enabled(self):
         return self.filter(enabled=True)
 
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
 
 class TaskManager(ResultManager):
     """Manager for :class:`celery.models.Task` models."""

djcelery/models.py

@@ -295,6 +295,9 @@ def schedule(self):
         if self.crontab:
             return self.crontab.schedule
 
+    def natural_key(self):
+        return (self.name,)
+
 
 signals.pre_delete.connect(PeriodicTasks.changed, sender=PeriodicTask)
 signals.pre_save.connect(PeriodicTasks.changed, sender=PeriodicTask)