Add scheduling models and fields for peer management

scheduler/models.py

@@ -0,0 +1,69 @@
+import uuid
+
+from django.db import models
+from django.utils.translation import gettext_lazy as _
+
+from wireguard.models import Peer
+
+WEEK_DAYS = [
+    (0, _("Monday")),
+    (1, _("Tuesday")),
+    (2, _("Wednesday")),
+    (3, _("Thursday")),
+    (4, _("Friday")),
+    (5, _("Saturday")),
+    (6, _("Sunday")),
+]
+
+class ScheduleProfile(models.Model):
+    name = models.CharField(max_length=100)
+
+    created = models.DateTimeField(auto_now_add=True)
+    updated = models.DateTimeField(auto_now=True)
+    uuid = models.UUIDField(editable=False, default=uuid.uuid4)
+
+
+class ScheduleSlot(models.Model):
+    profile = models.ForeignKey(ScheduleProfile, on_delete=models.CASCADE, related_name="slots")
+    start_weekday = models.PositiveSmallIntegerField(choices=WEEK_DAYS)
+    end_weekday = models.PositiveSmallIntegerField(choices=WEEK_DAYS)
+    start_time = models.TimeField()
+    end_time = models.TimeField()
+
+    created = models.DateTimeField(auto_now_add=True)
+    updated = models.DateTimeField(auto_now=True)
+    uuid = models.UUIDField(editable=False, default=uuid.uuid4)
+
+    class Meta:
+        ordering = ("start_weekday", "start_time")
+        constraints = [
+            models.UniqueConstraint(
+                fields=["profile", "start_weekday", "end_weekday", "start_time", "end_time"],
+                name="uniq_slot_per_profile"
+            ),
+        ]
+
+
+class PeerScheduling(models.Model):
+    peer = models.OneToOneField(Peer, on_delete=models.CASCADE, related_name="schedule")
+    profile = models.ForeignKey(ScheduleProfile, on_delete=models.SET_NULL, null=True, blank=True)
+
+    next_scheduled_enable_at = models.DateTimeField(null=True, blank=True)
+    next_scheduled_disable_at = models.DateTimeField(null=True, blank=True)
+    schedule_last_calculated_at = models.DateTimeField(null=True, blank=True)
+
+    next_manual_suspend_at = models.DateTimeField(null=True, blank=True)
+    next_manual_unsuspend_at = models.DateTimeField(null=True, blank=True)
+    manual_suspend_reason = models.TextField(null=True, blank=True)
+
+    created = models.DateTimeField(auto_now_add=True)
+    updated = models.DateTimeField(auto_now=True)
+    uuid = models.UUIDField(editable=False, default=uuid.uuid4)
+
+    class Meta:
+        indexes = [
+            models.Index(fields=["next_scheduled_enable_at"]),
+            models.Index(fields=["next_scheduled_disable_at"]),
+            models.Index(fields=["next_manual_suspend_at"]),
+            models.Index(fields=["next_manual_unsuspend_at"]),
+        ]