From aae9bbc6f0ac07fce760d9b7e150a2a5cd2b9760 Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Thu, 27 Jun 2019 15:23:19 -0500
Subject: [PATCH] Bugfix migration for populating raven_users group

---
 .../migrations/auth/0022_populate_raven_users_group.py   | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/gracedb/migrations/auth/0022_populate_raven_users_group.py b/gracedb/migrations/auth/0022_populate_raven_users_group.py
index 44a8e6935..441f66766 100644
--- a/gracedb/migrations/auth/0022_populate_raven_users_group.py
+++ b/gracedb/migrations/auth/0022_populate_raven_users_group.py
@@ -17,11 +17,12 @@ def add_users(apps, schema_editor):
     # Get group
     pg = Group.objects.get(name=GROUP_NAME)
 
-    # Get users
-    users = User.objects.filter(username__in=USERS)
+    # Users might not exist yet since they are populated from the LDAP
+    for username in USERS:
+        user, _ = User.objects.get_or_create(username=username)
 
-    # Add users
-    pg.user_set.add(*users)
+        # Add user to group
+        pg.user_set.add(user)
 
 
 def remove_users(apps, schema_editor):
-- 
GitLab