From f779c57f33763698e86f0f99341ad461d91073aa Mon Sep 17 00:00:00 2001 From: Mouse Reeve Date: Tue, 6 Dec 2022 14:25:50 -0800 Subject: [PATCH] Safely get or create permissions for tests/re-running --- bookwyrm/management/commands/initdb.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bookwyrm/management/commands/initdb.py b/bookwyrm/management/commands/initdb.py index 4b43fac7c..fda40bd07 100644 --- a/bookwyrm/management/commands/initdb.py +++ b/bookwyrm/management/commands/initdb.py @@ -10,7 +10,7 @@ def init_groups(): """permission levels""" groups = ["admin", "owner", "moderator", "editor"] for group in groups: - Group.objects.create(name=group) + Group.objects.get_or_create(name=group) def init_permissions(): @@ -18,7 +18,7 @@ def init_permissions(): permissions = [ { "codename": "manage_registration", - "name": "allow or prevent user registrations", + "name": "allow or prevent user registration", "groups": ["admin"], }, { @@ -65,7 +65,7 @@ def init_permissions(): content_type = ContentType.objects.get_for_model(models.User) for permission in permissions: - permission_obj = Permission.objects.create( + permission_obj, _ = Permission.objects.get_or_create( codename=permission["codename"], name=permission["name"], content_type=content_type,