authentik/passbook/core/forms/groups.py

"""passbook Core Group forms"""
from django import forms
from django.contrib.admin.widgets import FilteredSelectMultiple

from passbook.core.models import Group, User


class GroupForm(forms.ModelForm):
    """Group Form"""

    members = forms.ModelMultipleChoiceField(
        User.objects.all(), required=False, widget=FilteredSelectMultiple('users', False))

    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'])
        return instance

    class Meta:

        model = Group
        fields = ['name', 'parent', 'members', 'attributes']
        widgets = {
            'name': forms.TextInput(),
        }