mirror of https://github.com/etesync/server
Merge a9c0807a69
into b620d0a39c
commit
e7396fc81d
@ -0,0 +1,5 @@
|
||||
from distutils.util import strtobool
|
||||
|
||||
def argbool(arg):
|
||||
if arg == None: return None
|
||||
return bool(strtobool(arg))
|
@ -0,0 +1,68 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
from ._utils import argbool
|
||||
from myauth.models import User
|
||||
from django.db.utils import IntegrityError
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument( 'username'
|
||||
, type=str
|
||||
, help="New user's login username." )
|
||||
parser.add_argument( '-f'
|
||||
, '--first_name'
|
||||
, '--first'
|
||||
, type=str
|
||||
, default=''
|
||||
, help="New user's first name." )
|
||||
parser.add_argument( '-l'
|
||||
, '--last_name'
|
||||
, '--last'
|
||||
, type=str
|
||||
, default=''
|
||||
, help="New user's last name." )
|
||||
parser.add_argument( '-e'
|
||||
, '--email'
|
||||
, type=str
|
||||
, default=''
|
||||
, help="New user's email address." )
|
||||
parser.add_argument( '-a'
|
||||
, '--is_active'
|
||||
, '--active'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=True
|
||||
, help="Enable login. [YES]" )
|
||||
parser.add_argument( '-s'
|
||||
, '--is_staff'
|
||||
, '--staff'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=False
|
||||
, help="Mark user as staff. [NO]" )
|
||||
parser.add_argument( '-S'
|
||||
, '--is_superuser'
|
||||
, '--superuser'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=False
|
||||
, help="Mark user as superuser. [NO]" )
|
||||
|
||||
def handle(self, *args, **options):
|
||||
try:
|
||||
User.objects.create_user( username = options["username" ]
|
||||
, email = options["email" ]
|
||||
, first_name = options["first_name" ]
|
||||
, last_name = options["last_name" ]
|
||||
, is_active = options["is_active" ]
|
||||
, is_staff = options["is_staff" ]
|
||||
, is_superuser = options["is_superuser"] )
|
||||
except IntegrityError as exception:
|
||||
self.stdout.write(self.style.ERROR(f'Unable to create user "{options["username"]}": ' + str(exception)))
|
||||
exit(1)
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been created.'))
|
||||
exit(0)
|
@ -0,0 +1,33 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
from myauth.models import User
|
||||
from django.db.models.deletion import ProtectedError
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument( 'username'
|
||||
, type=str
|
||||
, help="Login username of the user to be deleted." )
|
||||
parser.add_argument( '--delete-user-data'
|
||||
, action='store_true'
|
||||
, default=False
|
||||
, help="Delete all user's collections!" )
|
||||
|
||||
def handle(self, *args, **options):
|
||||
try:
|
||||
user = User.objects.get(username = options["username"])
|
||||
if options["delete_user_data"]:
|
||||
collections = user.collection_set.all()
|
||||
for collection in collections:
|
||||
collection.delete()
|
||||
user.delete()
|
||||
except User.DoesNotExist as exception:
|
||||
self.stdout.write(self.style.ERROR(f'Unable to delete user "{options["username"]}": ' + str(exception)))
|
||||
exit(1)
|
||||
except ProtectedError as exception:
|
||||
self.stdout.write(self.style.ERROR(f'Unable to delete user "{options["username"]}": ' + str(exception)))
|
||||
self.stdout.write(self.style.NOTICE('Use --delete-user-data to overcome this protection.'))
|
||||
exit(2)
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been deleted.'))
|
||||
exit(0)
|
@ -0,0 +1,9 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
from myauth.models import User
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
def handle(self, *args, **options):
|
||||
for user in User.objects.all():
|
||||
print(user.username)
|
||||
exit(0)
|
@ -0,0 +1,81 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
from ._utils import argbool
|
||||
from myauth.models import User
|
||||
from django.db.utils import IntegrityError
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument( 'username'
|
||||
, type=str
|
||||
, help="User's login username." )
|
||||
parser.add_argument( '-u'
|
||||
, '--new_username'
|
||||
, '--new-username'
|
||||
, type=str
|
||||
, default=None
|
||||
, help="User's new login username." )
|
||||
parser.add_argument( '-f'
|
||||
, '--first_name'
|
||||
, '--first-name'
|
||||
, '--first'
|
||||
, type=str
|
||||
, default=None
|
||||
, help="User's new first name." )
|
||||
parser.add_argument( '-l'
|
||||
, '--last_name'
|
||||
, '--last-name'
|
||||
, '--last'
|
||||
, type=str
|
||||
, default=None
|
||||
, help="User's new last name." )
|
||||
parser.add_argument( '-e'
|
||||
, '--email'
|
||||
, type=str
|
||||
, default=None
|
||||
, help="User's new email address." )
|
||||
parser.add_argument( '-a'
|
||||
, '--is_active'
|
||||
, '--is-active'
|
||||
, '--active'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=None
|
||||
, help="Enable/Disable login." )
|
||||
parser.add_argument( '-s'
|
||||
, '--is_staff'
|
||||
, '--is-staff'
|
||||
, '--staff'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=None
|
||||
, help="Mark/Unmark user as staff." )
|
||||
parser.add_argument( '-S'
|
||||
, '--is_superuser'
|
||||
, '--is-superuser'
|
||||
, '--superuser'
|
||||
, nargs='?'
|
||||
, type=argbool
|
||||
, const=True
|
||||
, default=None
|
||||
, help="Mark/Unmark user as superuser." )
|
||||
|
||||
def handle(self, *args, **options):
|
||||
try:
|
||||
user = User.objects.get(username = options["username"])
|
||||
if options["new_username"] != None: user.username = options["new_username"]
|
||||
if options["email" ] != None: user.email = options["email" ]
|
||||
if options["first_name" ] != None: user.first_name = options["first_name" ]
|
||||
if options["last_name" ] != None: user.last_name = options["last_name" ]
|
||||
if options["is_active" ] != None: user.is_active = options["is_active" ]
|
||||
if options["is_staff" ] != None: user.is_staff = options["is_staff" ]
|
||||
if options["is_superuser"] != None: user.is_superuser = options["is_superuser"]
|
||||
user.save()
|
||||
except (User.DoesNotExist, ValueError) as exception:
|
||||
self.stdout.write(self.style.ERROR(f'Unable to modify user "{options["username"]}": ' + str(exception)))
|
||||
exit(1)
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been modified.'))
|
||||
exit(0)
|
Loading…
Reference in new issue