diff --git a/gracedb/migrations/auth/0016_create_access_and_superevent_groups.py b/gracedb/migrations/auth/0016_create_access_and_superevent_groups.py new file mode 100644 index 0000000000000000000000000000000000000000..461952d763f01fc5d220c6924ba73b4f2a14c251 --- /dev/null +++ b/gracedb/migrations/auth/0016_create_access_and_superevent_groups.py @@ -0,0 +1,43 @@ +# -*- coding: utf-8 -*- +# Generated by Django 1.11.14 on 2018-08-02 18:33 +# Create new groups for managing external access to superevents +# and for allowing non-Test superevent creation and update. We +# also add users to those groups + +from __future__ import unicode_literals + +from django.db import migrations + +GROUPS = { + 'access_managers': [], + 'superevent_managers': ['emfollow'], +} + +def add_groups(apps, schema_editor): + Group = apps.get_model('auth', 'Group') + User = apps.get_model('auth', 'User') + + for group_name, usernames in GROUPS.iteritems(): + g, _ = Group.objects.get_or_create(name=group_name) + users = User.objects.filter(username__in=usernames) + g.user_set.add(*users) + + +def remove_groups(apps, schema_editor): + Group = apps.get_model('auth', 'Group') + User = apps.get_model('auth', 'User') + + for group_name in GROUPS: + g = Group.objects.get(name=group_name) + g.delete() + + +class Migration(migrations.Migration): + + dependencies = [ + ('auth', '0015_update_emfollow_accounts'), + ] + + operations = [ + migrations.RunPython(add_groups, remove_groups), + ]