add migration to create 'bestuur' group with correct event permissions. Add default admin user. Add form to add users to a group
This commit is contained in:
parent
4306889349
commit
19d791153c
2 changed files with 119 additions and 1 deletions
|
@ -1,3 +1,4 @@
|
|||
from django.contrib import admin
|
||||
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
|
||||
from django.forms import ModelForm, TextInput
|
||||
|
||||
|
@ -21,6 +22,66 @@ class UserMetaForm(ModelForm):
|
|||
model = CustomUser
|
||||
fields = ['real_name', 'student_number']
|
||||
widgets = {
|
||||
'real_name': TextInput(attrs={'placeholder': 'Robbe Van Herck'}),
|
||||
'real_name': TextInput(attrs={'placeholder': 'Robbe Van Herck'}),
|
||||
'student_number': TextInput(attrs={'placeholder': '001700000'})
|
||||
}
|
||||
|
||||
|
||||
# Add user to groups in django admin
|
||||
|
||||
from django import forms
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.contrib.admin.widgets import FilteredSelectMultiple
|
||||
from django.contrib.auth.models import Group
|
||||
|
||||
User = get_user_model()
|
||||
|
||||
|
||||
# Create ModelForm based on the Group model.
|
||||
class GroupAdminForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Group
|
||||
exclude = []
|
||||
|
||||
# Add the users field.
|
||||
users = forms.ModelMultipleChoiceField(
|
||||
queryset=User.objects.all(),
|
||||
required=False,
|
||||
# Use the pretty 'filter_horizontal widget'.
|
||||
widget=FilteredSelectMultiple('users', False)
|
||||
)
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
# Do the normal form initialisation.
|
||||
super(GroupAdminForm, self).__init__(*args, **kwargs)
|
||||
# If it is an existing group (saved objects have a pk).
|
||||
if self.instance.pk:
|
||||
# Populate the users field with the current Group users.
|
||||
self.fields['users'].initial = self.instance.user_set.all()
|
||||
|
||||
def save_m2m(self):
|
||||
# Add the users to the Group.
|
||||
self.instance.user_set.set(self.cleaned_data['users'])
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
# Default save
|
||||
instance = super(GroupAdminForm, self).save()
|
||||
# Save many-to-many data
|
||||
self.save_m2m()
|
||||
return instance
|
||||
|
||||
|
||||
# Unregister the original Group admin.
|
||||
admin.site.unregister(Group)
|
||||
|
||||
|
||||
# Create a new Group admin.
|
||||
class GroupAdmin(admin.ModelAdmin):
|
||||
# Use our custom form.
|
||||
form = GroupAdminForm
|
||||
# Filter permissions horizontal as well.
|
||||
filter_horizontal = ['permissions']
|
||||
|
||||
|
||||
# Register the new Group ModelAdmin.
|
||||
admin.site.register(Group, GroupAdmin)
|
||||
|
|
57
users/migrations/0002_auto_20200724_2340.py
Normal file
57
users/migrations/0002_auto_20200724_2340.py
Normal file
|
@ -0,0 +1,57 @@
|
|||
# Generated by Django 3.0.8 on 2020-07-24 21:40
|
||||
import logging
|
||||
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.core.management.sql import emit_post_migrate_signal
|
||||
from django.db import migrations
|
||||
from django.utils import timezone
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def create_superuser(apps, schema_editor):
|
||||
superuser = get_user_model()(
|
||||
is_superuser=True,
|
||||
is_staff=True,
|
||||
username="admin", # os.environ['ADMIN_USERNAME'],
|
||||
last_login=timezone.now(),
|
||||
)
|
||||
# superuser.set_password(os.environ['ADMIN_PASSWORD'])
|
||||
superuser.set_password('admin')
|
||||
superuser.save()
|
||||
|
||||
|
||||
kers_group_permissions = {
|
||||
"Bestuur": [
|
||||
"add_event",
|
||||
"delete_event",
|
||||
"change_event",
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
def add_group_permissions(apps, schema_editor):
|
||||
db_alias = schema_editor.connection.alias
|
||||
emit_post_migrate_signal(2, False, db_alias)
|
||||
|
||||
Group = apps.get_model('auth', 'Group')
|
||||
Permission = apps.get_model('auth', 'Permission')
|
||||
|
||||
for group in kers_group_permissions:
|
||||
role, created = Group.objects.get_or_create(name=group)
|
||||
logger.info(f'{group} Group created')
|
||||
for perm in kers_group_permissions[group]:
|
||||
role.permissions.add(Permission.objects.get(codename=perm))
|
||||
logger.info(f'Permitting {group} to {perm}')
|
||||
role.save()
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
dependencies = [
|
||||
('users', '0001_initial'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RunPython(create_superuser),
|
||||
migrations.RunPython(add_group_permissions),
|
||||
]
|
Loading…
Reference in a new issue