mozilla / fx-private-relay / b438d20a-f354-46f7-80b7-2ff2f189d5a4

from django.core.management.base import BaseCommand

from ...models import Profile

class Command(BaseCommand):

    help = "Removes an API token to effectively block access."

    def add_arguments(self, parser):

        parser.add_argument("api_token", nargs=1)

    def handle(self, *args, **options):

        try:

            profile = Profile.objects.get(api_token=options["api_token"][0])

            profile.user.is_active = False

            profile.user.save()

            self.stdout.write("SUCCESS: deactivated user.")

        except Profile.DoesNotExist:

            self.stdout.write("ERROR: Could not find user with that token.")