Fix add user to group

b12io · Jan 20, 2023 · 95d46a5 · 95d46a5
1 parent 654b001
commit 95d46a5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/journalism_workflow/load_sample_data.py b/journalism_workflow/load_sample_data.py
@@ -20,8 +20,10 @@ def load(workflow_version):
         user.set_password(password)
         user.save()
         worker, _ = Worker.objects.update_or_create(user=user)
-        project_admins = Group.objects.get_or_create(name='project_admins')
-        project_admins.user_set.add(user)
+        project_admins, created = Group.objects.get_or_create(
+            name='project_admins')
+        if created:
+            user.groups.add(project_admins)
 
         # Grant the worker the desired certifications.
         for certification_slug, certification_role in certifications:

diff --git a/simple_workflow/load_sample_data.py b/simple_workflow/load_sample_data.py
@@ -21,5 +21,7 @@ def load(workflow_version):
     user.save()
     if created:
         Worker.objects.create(user=user)
-    project_admins = Group.objects.get_or_create(name='project_admins')
-    project_admins.user_set.add(user)
+    project_admins, created = Group.objects.get_or_create(
+        name='project_admins')
+    if created:
+        user.groups.add(project_admins)