This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/core/forms/groups.py

33 lines
967 B
Python
Raw Normal View History

2019-03-08 14:49:45 +00:00
"""passbook Core Group forms"""
from django import forms
from passbook.core.models import Group, User
from django.contrib.admin.widgets import FilteredSelectMultiple
2019-03-08 14:49:45 +00:00
class GroupForm(forms.ModelForm):
"""Group Form"""
members = forms.ModelMultipleChoiceField(
User.objects.all(), required=False, widget=FilteredSelectMultiple('users', False))
2019-03-08 14:49:45 +00:00
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
if self.instance.pk:
self.initial['members'] = self.instance.user_set.values_list('pk', flat=True)
def save(self, *args, **kwargs):
instance = super().save(*args, **kwargs)
if instance.pk:
instance.user_set.clear()
instance.user_set.add(*self.cleaned_data['members'])
2019-03-08 14:49:45 +00:00
return instance
class Meta:
model = Group
fields = ['name', 'parent', 'members', 'tags']
widgets = {
'name': forms.TextInput(),
}