uncloud/uncloud_auth/management/commands/make-admin.py

22 lines
655 B
Python
Raw Normal View History

2020-05-07 10:45:06 +00:00
from django.core.management.base import BaseCommand
from django.contrib.auth import get_user_model
import sys
class Command(BaseCommand):
help = 'Give Admin rights to existing user'
def add_arguments(self, parser):
parser.add_argument('username', type=str)
parser.add_argument('--superuser', action='store_true')
2020-05-07 10:45:06 +00:00
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
2020-05-07 10:45:06 +00:00
user.save()
print(f"{user.username} is now admin (superuser={user.is_superuser})")