mirror of
https://github.com/eduardogsilva/wireguard_webadmin.git
synced 2025-04-19 08:55:12 +00:00
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from django import forms
|
|
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):
|
|
user_level = forms.ChoiceField(choices=UserAcl.user_level.field.choices, required=True, label="User Level")
|
|
|
|
class Meta(UserCreationForm.Meta):
|
|
model = User
|
|
fields = UserCreationForm.Meta.fields + ('user_level',)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.user_id = kwargs.pop('user_id', None)
|
|
super().__init__(*args, **kwargs)
|
|
if self.instance and self.instance.pk:
|
|
self.fields['password1'].required = False
|
|
self.fields['password2'].required = False
|
|
self.fields['username'].widget.attrs['readonly'] = True
|
|
|
|
def clean_username(self):
|
|
username = self.cleaned_data.get('username')
|
|
if User.objects.filter(username=username).exclude(pk=self.user_id).exists():
|
|
raise ValidationError("A user with that username already exists.")
|
|
return username
|
|
|
|
def save(self, commit=True):
|
|
user = super().save(commit=False)
|
|
new_password = self.cleaned_data.get("password1")
|
|
|
|
if new_password:
|
|
user.set_password(new_password)
|
|
user.save()
|
|
else:
|
|
if not user.id:
|
|
user.save()
|
|
|
|
if commit:
|
|
user_acl, created = UserAcl.objects.update_or_create(
|
|
user=user,
|
|
defaults={'user_level': self.cleaned_data.get('user_level')}
|
|
)
|
|
|
|
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
|
|
|