mozilla / fx-private-relay / 243b9f6a-6560-4235-88c8-835a09fc9bc3 / 1

Source File

0.0

/emails/management/commands/deactivate_user_by_token.py

from django.core.management.base import BaseCommand, CommandError

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.")

1	from django.core.management.base import BaseCommand, CommandError	×
2
3	from ...models import Profile	×
4
5
6	class Command(BaseCommand):	×
7	help = "Removes an API token to effectively block access."	×
8
9	def add_arguments(self, parser):	×
10	parser.add_argument("api_token", nargs=1)	×
11
12	def handle(self, args, *options):	×
13	try:	×
14	profile = Profile.objects.get(api_token=options["api_token"][0])	×
15	profile.user.is_active = False	×
16	profile.user.save()	×
17	self.stdout.write("SUCCESS: deactivated user.")	×
18	except Profile.DoesNotExist:	×
19	self.stdout.write("ERROR: Could not find user with that token.")	×