From e6e7647f3225f2bfc049d730464af2f9594aaa95 Mon Sep 17 00:00:00 2001 From: Les Begnaud Date: Fri, 26 Jul 2019 14:13:15 -0500 Subject: [PATCH] adjusts users and groups startup scripts to accomdate non-uniqueness --- startup_scripts/000_users.py | 4 ++-- startup_scripts/010_groups.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/startup_scripts/000_users.py b/startup_scripts/000_users.py index a2e2929..a1340a1 100644 --- a/startup_scripts/000_users.py +++ b/startup_scripts/000_users.py @@ -29,6 +29,6 @@ with file.open('r') as stream: if user_permissions: user.user_permissions.clear() for permission_codename in user_details.get('permissions', []): - permission = Permission.objects.get(codename=permission_codename) - user.user_permissions.add(permission) + for permission in Permission.objects.filter(codename=permission_codename): + user.user_permissions.add(permission) user.save() diff --git a/startup_scripts/010_groups.py b/startup_scripts/010_groups.py index c296d16..de8705c 100644 --- a/startup_scripts/010_groups.py +++ b/startup_scripts/010_groups.py @@ -29,5 +29,5 @@ with file.open('r') as stream: group.permissions.clear() print("Permissions:", group.permissions.all()) for permission_codename in group_details.get('permissions', []): - permission = Permission.objects.get(codename=permission_codename) - group.permissions.add(permission) + for permission in Permission.objects.filter(codename=permission_codename): + group.permissions.add(permission)