71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from django.db import migrations
|
||
|
|
|
||
|
|
|
||
|
|
def create_superadmin_group(apps, schema_editor):
|
||
|
|
"""
|
||
|
|
Migrates the groups to create an admin group with all permissions
|
||
|
|
granted. Replaces the delegate group to get pk=2.
|
||
|
|
|
||
|
|
- Create new delegate group. Move users and permissions to it.
|
||
|
|
- Rename the old delegate group to Admin and remove all permissions.
|
||
|
|
- If a group with the name 'Admin' (probably with pk = 4) exists, move all
|
||
|
|
users from it to the new superadmin group and delete it. If not, check for
|
||
|
|
the staff group and assign all users to the superadmin group.
|
||
|
|
"""
|
||
|
|
Group = apps.get_model('users', 'Group')
|
||
|
|
|
||
|
|
# If no groups exists at all, skip this migration
|
||
|
|
if Group.objects.count() == 0:
|
||
|
|
return
|
||
|
|
|
||
|
|
# Get the new superadmin group (or the old delegates)
|
||
|
|
superadmin, created_superadmin_group = Group.objects.get_or_create(pk=2, defaults={'name': '__temp__'})
|
||
|
|
|
||
|
|
if not created_superadmin_group:
|
||
|
|
new_delegate = Group.objects.create(name='Delegates2')
|
||
|
|
new_delegate.permissions.set(superadmin.permissions.all())
|
||
|
|
superadmin.permissions.set([])
|
||
|
|
|
||
|
|
for user in superadmin.user_set.all():
|
||
|
|
user.groups.add(new_delegate)
|
||
|
|
user.groups.remove(superadmin)
|
||
|
|
|
||
|
|
finished_moving_users = False
|
||
|
|
try:
|
||
|
|
admin = Group.objects.get(name='Admin')
|
||
|
|
for user in admin.user_set.all():
|
||
|
|
user.groups.add(superadmin)
|
||
|
|
user.groups.remove(admin)
|
||
|
|
admin.delete(skip_autoupdate=True)
|
||
|
|
finished_moving_users = True
|
||
|
|
except Group.DoesNotExist:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if not finished_moving_users:
|
||
|
|
try:
|
||
|
|
staff = Group.objects.get(name='Staff')
|
||
|
|
for user in staff.user_set.all():
|
||
|
|
user.groups.add(superadmin)
|
||
|
|
except Group.DoesNotExist:
|
||
|
|
pass
|
||
|
|
|
||
|
|
superadmin.name = 'Admin'
|
||
|
|
superadmin.save(skip_autoupdate=True)
|
||
|
|
if not created_superadmin_group:
|
||
|
|
new_delegate.name = 'Delegates'
|
||
|
|
new_delegate.save(skip_autoupdate=True)
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
('users', '0006_user_email'),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.RunPython(create_superadmin_group),
|
||
|
|
]
|