Allow scheduling notification rules
This commit is contained in:
39
core/management/commands/scheduling.py
Normal file
39
core/management/commands/scheduling.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
|
||||
from core.util import logs
|
||||
from core.models import NotificationRule
|
||||
from core.db.storage import db
|
||||
import aioschedule as schedule
|
||||
import asyncio
|
||||
from time import sleep
|
||||
|
||||
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]
|
||||
|
||||
def run_schedule(interval_seconds):
|
||||
print("Running schedule", interval_seconds)
|
||||
matching_rules = NotificationRule.objects.filter(
|
||||
enabled=True, interval=interval_seconds
|
||||
)
|
||||
|
||||
class Command(BaseCommand):
|
||||
def handle(self, *args, **options):
|
||||
for interval in INTERVALS:
|
||||
schedule.every(interval).seconds.do(run_schedule, interval_seconds=interval)
|
||||
|
||||
loop = asyncio.get_event_loop()
|
||||
|
||||
while True:
|
||||
loop.run_until_complete(schedule.run_pending())
|
||||
sleep(10)
|
||||
Reference in New Issue
Block a user