add group administration
This commit is contained in:
30
passbook/core/forms/groups.py
Normal file
30
passbook/core/forms/groups.py
Normal file
@ -0,0 +1,30 @@
|
||||
"""passbook Core Group forms"""
|
||||
from django import forms
|
||||
|
||||
from passbook.core.models import Group, User
|
||||
|
||||
|
||||
class GroupForm(forms.ModelForm):
|
||||
"""Group Form"""
|
||||
|
||||
members = forms.ModelMultipleChoiceField(User.objects.all(), required=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['users'])
|
||||
return instance
|
||||
|
||||
class Meta:
|
||||
|
||||
model = Group
|
||||
fields = ['name', 'parent', 'members', 'tags']
|
||||
widgets = {
|
||||
'name': forms.TextInput(),
|
||||
}
|
Reference in New Issue
Block a user