Implement running scheduled rules and check aggregations

core/management/commands/scheduling.py

@@ -1,25 +1,18 @@
 import asyncio
 
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from asgiref.sync import sync_to_async
 from django.core.management.base import BaseCommand
 
-# from core.db.storage import db
-# from core.models import NotificationRule
+from core.db.storage import db
+from core.lib.parsing import QueryError
+from core.lib.rules import NotificationRuleData
+from core.models import NotificationRule
 from core.util import logs
 
 log = logs.get_logger("scheduling")
 
-# INTERVAL_CHOICES = (
-#     (0, "On demand"),
-#     (60, "Every minute"),
-#     (900, "Every 15 minutes"),
-#     (1800, "Every 30 minutes"),
-#     (3600, "Every hour"),
-#     (14400, "Every 4 hours"),
-#     (86400, "Every day"),
-# )
-
-INTERVALS = [60, 900, 1800, 3600, 14400, 86400]
+INTERVALS = [5, 60, 900, 1800, 3600, 14400, 86400]
 
 
 async def job(interval_seconds):
@@ -27,10 +20,17 @@ async def job(interval_seconds):
     Run all schedules matching the given interval.
     :param interval_seconds: The interval to run.
     """
-    print("Running schedule", interval_seconds)
-    # matching_rules = NotificationRule.objects.filter(
-    #     enabled=True, interval=interval_seconds
-    # )
+    matching_rules = await sync_to_async(list)(
+        NotificationRule.objects.filter(enabled=True, interval=interval_seconds)
+    )
+    for rule in matching_rules:
+        log.debug(f"Running rule {rule}")
+        try:
+            rule = NotificationRuleData(rule.user, rule, db=db)
+            await rule.run_schedule()
+            # results = await db.schedule_query_results(rule.user, rule)
+        except QueryError as e:
+            log.error(f"Error running rule {rule}: {e}")
 
 
 class Command(BaseCommand):