1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from django.core.management.base import BaseCommand, CommandError
from django.db import transaction
from django.contrib.auth import get_user_model
class Command(BaseCommand):
help = "Promotes the named users to an admin superuser."
def add_arguments(self, parser):
super(Command, self).add_arguments(parser)
parser.add_argument(
"usernames",
metavar="usernames",
nargs="*",
help="Usernames to promote to admin superuser.",
)
@transaction.atomic()
def handle(self, **kwargs):
verbosity = kwargs["verbosity"]
User = get_user_model()
for username in kwargs["usernames"]:
try:
user = User.objects.get(username=username)
except User.DoesNotExist:
raise CommandError("User with username {username} does not exist".format(
username=username,
))
else:
user.is_staff = True
user.is_superuser = True
user.save()
if verbosity >= 1:
self.stdout.write("Promoted {user} to admin superuser".format(
user=user,
))
|
