kpsul/kfet/auth/forms.py

from django import forms
from django.contrib.auth.models import Group, User

from .fields import KFetPermissionsField


class GroupForm(forms.ModelForm):
    permissions = KFetPermissionsField()

    def clean_name(self):
        name = self.cleaned_data["name"]
        return "K-Fêt %s" % name

    def clean_permissions(self):
        kfet_perms = self.cleaned_data["permissions"]
        # TODO: With Django >=1.11, the QuerySet method 'difference' can be
        # used.
        # other_groups = self.instance.permissions.difference(
        #     self.fields['permissions'].queryset
        # )
        if self.instance.pk is None:
            return kfet_perms
        other_perms = self.instance.permissions.exclude(
            pk__in=[p.pk for p in self.fields["permissions"].queryset]
        )
        return list(kfet_perms) + list(other_perms)

    class Meta:
        model = Group
        fields = ["name", "permissions"]


class UserGroupForm(forms.ModelForm):
    groups = forms.ModelMultipleChoiceField(
        Group.objects.filter(name__icontains="K-Fêt"),
        label="Statut équipe",
        required=False,
    )

    def clean_groups(self):
        kfet_groups = self.cleaned_data.get("groups")
        if self.instance.pk is None:
            return kfet_groups
        other_groups = self.instance.groups.exclude(name__icontains="K-Fêt")
        return list(kfet_groups) + list(other_groups)

    class Meta:
        model = User
        fields = ["groups"]