From fb66c429912f1946ee6d126a6332cda6733d842c Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Fri, 12 Oct 2018 14:16:00 -0500
Subject: [PATCH] Add 'emfollow' user to access_managers and em_advocates
---
.../auth/0018_update_emfollow_groups.py | 55 +++++++++++++++++++
1 file changed, 55 insertions(+)
create mode 100644 gracedb/migrations/auth/0018_update_emfollow_groups.py
diff --git a/gracedb/migrations/auth/0018_update_emfollow_groups.py b/gracedb/migrations/auth/0018_update_emfollow_groups.py
new file mode 100644
index 000000000..422ffbda1
--- /dev/null
+++ b/gracedb/migrations/auth/0018_update_emfollow_groups.py
@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.14 on 2018-10-08 21:15:20
+
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+USER = 'emfollow'
+GROUPS = [
+ 'em_advocates',
+ 'access_managers',
+]
+
+
+def add_to_groups(apps, schema_editor):
+ Group = apps.get_model('auth', 'Group')
+ User = apps.get_model('auth', 'User')
+
+ # Get user
+ user = User.objects.get(username=USER)
+
+ # Loop over groups and add user
+ for g in GROUPS:
+ # Get group
+ group = Group.objects.get(name=g)
+
+ # Add user to group
+ group.user_set.add(user)
+
+
+def remove_from_groups(apps, schema_editor):
+ Group = apps.get_model('auth', 'Group')
+ User = apps.get_model('auth', 'User')
+
+ # Get user
+ user = User.objects.get(username=USER)
+
+ # Loop over groups and add user
+ for g in GROUPS:
+ # Get group
+ group = Group.objects.get(name=g)
+
+ # Remove user from group
+ group.user_set.remove(user)
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('auth', '0017_assign_permissions'),
+ ]
+
+ operations = [
+ migrations.RunPython(add_to_groups, remove_from_groups),
+ ]
--
GitLab