def form_valid(self, form):
for device in devices_for_user(self.request.user):
device.delete()
resp = HttpResponseRedirect(reverse('edit_profile'))
resp.delete_cookie('S2FA', domain=settings.SESSION_COOKIE_DOMAIN)
return resp
def test_can_disable_two_factor_auth(self):
totp = TOTPDevice(user=self.admin, name="", confirmed=1)
totp.save()
devices = devices_for_user(self.admin)
i = 0
for device in devices_for_user(self.admin):
if device:
i+=1
assert i > 0
resp = self.client.delete(reverse('two-factor-auth-view', args=[str(self.admin.username)]))
assert resp.status_code == 200
i = 0
for device in devices_for_user(self.admin):
if device:
i+=1
assert i == 0
def handle(self, *args, **options):
for username in args:
try:
user = User.objects.get_by_natural_key(username)
except User.DoesNotExist:
raise CommandError('User "%s" does not exist' % username)
for device in devices_for_user(user):
device.delete()
def delete(self, request, email):
if not email:
error_msg = "email can not be empty"
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
try:
user = User.objects.get(email=email)
except User.DoesNotExist:
error_msg = "User %s not found" % email
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
devices = devices_for_user(user)
if devices:
for device in devices:
device.delete()
return Response({'success': True}, status=status.HTTP_200_OK)
def device_choices(user):
return list((d.persistent_id, d.name) for d in devices_for_user(user))
def device_choices(user):
return list((d.persistent_id, d.name) for d in devices_for_user(user))
def tearDown(self):
try:
for device in devices_for_user(self.admin):
device.delete()
except:
pass