Make list

Author: Maciej Gol

django_celery_beat/schedulers.py

@@ -264,6 +264,22 @@ def setup_schedule(self):
     def all_as_schedule(self):
         debug('DatabaseScheduler: Fetching database schedule')
         s = {}
+        for model in self.enabled_models():
+            try:
+                s[model.name] = self.Entry(model, app=self.app)
+            except ValueError:
+                pass
+        return s
+
+    def enabled_models(self):
+        """Return list of enabled periodic tasks.
+
+        Allows overriding how the list of periodic tasks is fetched without
+        duplicating the filtering/querying logic.
+        """
+        return list(self.enabled_models_qs())
+
+    def enabled_models_qs(self):
         next_schedule_sync = now() + datetime.timedelta(
             seconds=SCHEDULE_SYNC_MAX_INTERVAL
         )
@@ -278,12 +294,7 @@ def all_as_schedule(self):
         exclude_query = exclude_clock_tasks_query | exclude_cron_tasks_query
 
         # Fetch only the tasks we need to consider
-        for model in self.Model.objects.enabled().exclude(exclude_query):
-            try:
-                s[model.name] = self.Entry(model, app=self.app)
-            except ValueError:
-                pass
-        return s
+        return self.Model.objects.enabled().exclude(exclude_query)
 
     def _get_crontab_exclude_query(self):
         """