From 424ceafbb8cda4d714c7675c2f8bd1b387d65620 Mon Sep 17 00:00:00 2001 From: Nico Schottelius Date: Sun, 21 Jun 2020 14:44:55 +0200 Subject: [PATCH] Enhance make-admin to allow making superuser --- uncloud_auth/management/commands/make-admin.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/uncloud_auth/management/commands/make-admin.py b/uncloud_auth/management/commands/make-admin.py index b750bc3..9157439 100644 --- a/uncloud_auth/management/commands/make-admin.py +++ b/uncloud_auth/management/commands/make-admin.py @@ -7,10 +7,15 @@ class Command(BaseCommand): def add_arguments(self, parser): parser.add_argument('username', type=str) + parser.add_argument('--superuser', action='store_true') def handle(self, *args, **options): user = get_user_model().objects.get(username=options['username']) user.is_staff = True + + if options['superuser']: + user.is_superuser = True + user.save() - print("{} is now admin.".format(user.username)) + print(f"{user.username} is now admin (superuser={user.is_superuser})")