add command to enable active/staff

This commit is contained in:
Pierre-antoine Comby 2020-02-25 19:20:39 +01:00
parent e2c1d70eb4
commit a088c6baea
1 changed files with 23 additions and 0 deletions

View File

@ -0,0 +1,23 @@
#!/usr/bin/env python3
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')
def handle(self,*args,**kwargs):
for uname in kwargs["username"]:
user = User.objects.get(username=uname)
user.is_active = True
if kwargs['SUPER']:
user.is_staff = True
user.save()