arno/etesync-server
1
0
Fork 0
mirror of https://github.com/etesync/server synced 2025-02-21 12:52:09 +00:00
Code Issues Releases Wiki Activity

Improve users-delete interface

Browse Source
This commit is contained in:
Martin Michalec 2021-06-04 00:14:02 +02:00
parent 211fb12d43
commit 8803d749db
1 changed files with 18 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
django_etebase/admin-cli/management/commands/users-delete.py
View File

@ -4,22 +4,27 @@ from myauth.models import User
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument( '-y'
, '--yes'
parser.add_argument( 'usernames'
, default=False
, type=str
, nargs='*'
, default=[]
, help="Delete ALL users!" )
parser.add_argument( '-a'
, '--all'
, action='store_true'
, default=False
, help="Allow deletion of all users!" )
, help="Delete ALL users!" )
def handle(self, *args, **options):
if options["yes"] != True:
print('Do you really want to delete all users? [y/N]: ', end='')
if input()[0] not in ('y', 'Y', 'yes', 'YES', 'Yes'):
self.stdout.write(self.style.SUCCESS(f'No users have been deleted.'))
exit(0)
try:
for user in User.objects.all():
user.delete()
self.stdout.write(self.style.SUCCESS(f'All users have been deleted.'))
if options["all"]:
for user in User.objects.all():
user.delete()
self.stdout.write(self.style.SUCCESS(f'All users have been deleted.'))
else:
for username in options["usernames"]:
User.objects.get(username=username).delete()
self.stdout.write(self.style.SUCCESS(f'Users have been deleted.'))
except User.DoesNotExist as exception:
self.stdout.write(self.style.ERROR(f'Unable to delete all users: ' + str(exception)))
self.stdout.write(self.style.ERROR(f'Unable to delete users: ' + str(exception)))