Add scheduling models and fields for peer management

This commit is contained in:
Eduardo Silva
2026-01-28 11:57:54 -03:00
parent 35e2899b3c
commit d3033d3936
11 changed files with 182 additions and 1 deletions

0
scheduler/__init__.py Normal file
View File

1
scheduler/admin.py Normal file
View File

@@ -0,0 +1 @@
# Register your models here.

6
scheduler/apps.py Normal file
View File

@@ -0,0 +1,6 @@
from django.apps import AppConfig
class SchedulerConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'scheduler'

View File

@@ -0,0 +1,66 @@
# Generated by Django 5.2.9 on 2026-01-28 14:57
import uuid
import django.db.models.deletion
from django.db import migrations, models
class Migration(migrations.Migration):
initial = True
dependencies = [
('wireguard', '0030_peer_enabled_by_schedule_peer_suspend_reason_and_more'),
]
operations = [
migrations.CreateModel(
name='ScheduleProfile',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('name', models.CharField(max_length=100)),
('created', models.DateTimeField(auto_now_add=True)),
('updated', models.DateTimeField(auto_now=True)),
('uuid', models.UUIDField(default=uuid.uuid4, editable=False)),
],
),
migrations.CreateModel(
name='PeerScheduling',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('next_scheduled_enable_at', models.DateTimeField(blank=True, null=True)),
('next_scheduled_disable_at', models.DateTimeField(blank=True, null=True)),
('schedule_last_calculated_at', models.DateTimeField(blank=True, null=True)),
('next_manual_suspend_at', models.DateTimeField(blank=True, null=True)),
('next_manual_unsuspend_at', models.DateTimeField(blank=True, null=True)),
('manual_suspend_reason', models.TextField(blank=True, null=True)),
('created', models.DateTimeField(auto_now_add=True)),
('updated', models.DateTimeField(auto_now=True)),
('uuid', models.UUIDField(default=uuid.uuid4, editable=False)),
('peer', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='schedule', to='wireguard.peer')),
('profile', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='scheduler.scheduleprofile')),
],
options={
'indexes': [models.Index(fields=['next_scheduled_enable_at'], name='scheduler_p_next_sc_ef2e87_idx'), models.Index(fields=['next_scheduled_disable_at'], name='scheduler_p_next_sc_203dba_idx'), models.Index(fields=['next_manual_suspend_at'], name='scheduler_p_next_ma_31b89c_idx'), models.Index(fields=['next_manual_unsuspend_at'], name='scheduler_p_next_ma_b980b8_idx')],
},
),
migrations.CreateModel(
name='ScheduleSlot',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('start_weekday', models.PositiveSmallIntegerField(choices=[(0, 'Monday'), (1, 'Tuesday'), (2, 'Wednesday'), (3, 'Thursday'), (4, 'Friday'), (5, 'Saturday'), (6, 'Sunday')])),
('end_weekday', models.PositiveSmallIntegerField(choices=[(0, 'Monday'), (1, 'Tuesday'), (2, 'Wednesday'), (3, 'Thursday'), (4, 'Friday'), (5, 'Saturday'), (6, 'Sunday')])),
('start_time', models.TimeField()),
('end_time', models.TimeField()),
('created', models.DateTimeField(auto_now_add=True)),
('updated', models.DateTimeField(auto_now=True)),
('uuid', models.UUIDField(default=uuid.uuid4, editable=False)),
('profile', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='slots', to='scheduler.scheduleprofile')),
],
options={
'ordering': ('start_weekday', 'start_time'),
'constraints': [models.UniqueConstraint(fields=('profile', 'start_weekday', 'end_weekday', 'start_time', 'end_time'), name='uniq_slot_per_profile')],
},
),
]

View File

69
scheduler/models.py Normal file
View File

@@ -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"]),
]

1
scheduler/tests.py Normal file
View File

@@ -0,0 +1 @@
# Create your tests here.

1
scheduler/views.py Normal file
View File

@@ -0,0 +1 @@
# Create your views here.