9cf2f9f298
This will remove imports from __future__ that are no longer needed. https://docs.python.org/3.5/library/2to3.html#2to3fixer-future
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
|
|
|
|
import re
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.core.management.base import BaseCommand
|
|
from six import text_type
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
help = """
|
|
This command will set is_staff to true for one or more users.
|
|
Lookup by username or email address, assumes usernames
|
|
do not look like email addresses.
|
|
"""
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('users',
|
|
nargs='+',
|
|
help='Users to set or unset (with the --unset flag) as superusers')
|
|
parser.add_argument('--unset',
|
|
action='store_true',
|
|
dest='unset',
|
|
default=False,
|
|
help='Set is_staff to False instead of True')
|
|
|
|
def handle(self, *args, **options):
|
|
for user in options['users']:
|
|
try:
|
|
if re.match(r'[^@]+@[^@]+\.[^@]+', user):
|
|
v = User.objects.get(email=user)
|
|
else:
|
|
v = User.objects.get(username=user)
|
|
|
|
if options['unset']:
|
|
v.is_staff = False
|
|
else:
|
|
v.is_staff = True
|
|
|
|
v.save()
|
|
print('Modified {} sucessfully.'.format(user))
|
|
|
|
except Exception as err: # pylint: disable=broad-except
|
|
print("Error modifying user with identifier {}: {}: {}".format(user, type(err).__name__,
|
|
text_type(err)))
|
|
|
|
print('Complete!')
|