add command to enable active/staff
This commit is contained in:
parent
e2c1d70eb4
commit
a088c6baea
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue