diff --git a/apps/ops/mixin.py b/apps/ops/mixin.py index fd6f9cf27..166c5db75 100644 --- a/apps/ops/mixin.py +++ b/apps/ops/mixin.py @@ -95,10 +95,8 @@ class PeriodTaskModelMixin(models.Model): @property def schedule(self): from django_celery_beat.models import PeriodicTask - try: - return PeriodicTask.objects.get(name=str(self)) - except PeriodicTask.DoesNotExist: - return None + name = self.get_register_task()[0] + return PeriodicTask.objects.filter(name=name).first() class Meta: abstract = True