create and manage peer groups

This commit is contained in:
Eduardo Silva 2025-01-20 11:41:02 -03:00
parent 797058b29b
commit eefc573c61
9 changed files with 222 additions and 11 deletions

View file

@ -3,6 +3,7 @@ from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User
from .models import UserAcl
from django.core.exceptions import ValidationError
from wireguard.models import PeerGroup
class UserAclForm(UserCreationForm):
@ -44,3 +45,34 @@ class UserAclForm(UserCreationForm):
)
return user
class PeerGroupForm(forms.ModelForm):
class Meta:
model = PeerGroup
fields = ['name', 'peer', 'server_instance']
def __init__(self, *args, **kwargs):
self.user_id = kwargs.pop('user_id', None)
super().__init__(*args, **kwargs)
def clean(self):
cleaned_data = super().clean()
name = cleaned_data.get('name')
peers = cleaned_data.get('peer')
server_instances = cleaned_data.get('server_instance')
if PeerGroup.objects.filter(name=name).exclude(pk=self.instance.pk if self.instance else None).exists():
raise ValidationError("A peer group with that name already exists.")
return cleaned_data
def save(self, commit=True):
peer_group = super().save(commit=False)
if commit:
peer_group.save()
return peer_group