From 95d46a5625a9c047d90944517663dddba2d65273 Mon Sep 17 00:00:00 2001 From: Aditya Bharadwaj Date: Fri, 20 Jan 2023 20:48:03 +0000 Subject: [PATCH] Fix add user to group --- journalism_workflow/load_sample_data.py | 6 ++++-- simple_workflow/load_sample_data.py | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/journalism_workflow/load_sample_data.py b/journalism_workflow/load_sample_data.py index 310049f9..d5373fbc 100644 --- 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 index 00d433e4..7a7714be 100644 --- 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)