mirror of https://github.com/etesync/server
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.5 KiB
34 lines
1.5 KiB
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)
|