Allow scheduling notification rules
parent
9ee9c7abde
commit
2a1e6b3292
@ -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)
|
@ -0,0 +1,18 @@
|
||||
# Generated by Django 4.1.3 on 2023-01-14 14:54
|
||||
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('core', '0016_notificationrule_interval_notificationrule_window'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AlterField(
|
||||
model_name='notificationrule',
|
||||
name='interval',
|
||||
field=models.IntegerField(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')], default=0),
|
||||
),
|
||||
]
|
Loading…
Reference in New Issue