nk20-scripts/management/commands/make_su.py

23 lines
826 B
Python

# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later
from django.core.management.base import BaseCommand
from django.contrib.auth.models import User
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument('username', nargs='+', type=str)
parser.add_argument('-S', "--SUPER", action='store_true', help='make superuser')
parser.add_argument('-s', "--STAFF", action='store_true', help='make staff')
def handle(self, *args, **kwargs):
for uname in kwargs["username"]:
user = User.objects.get(username=uname)
user.is_active = True
if kwargs['STAFF']:
user.is_staff = True
if kwargs['SUPER']:
user.is_superuser = True
user.save()