def handle(self, *args, **options):
if len(args) != 1:
raise CommandError('Provide a username as first argument')
try:
user = User.objects.get(username=args[0])
except User.DoesNotExist:
raise CommandError('User with username "%s" not found' % args[0])
try:
token = user.token
except Token.DoesNotExist:
token = Token(user=user)
token.seed = generate_seed()
token.save()
print 'Updated seed for %s to: %s (hex)' % (user.username, token.seed)
print ''
print 'QR Code for Google Authenticator can be found here:'
print ''
print get_qr_url(user.username, token.seed)
print ''
def handle(self, *args, **options):
if len(args) != 1:
raise CommandError('Provide a username as first argument')
try:
user = User.objects.get(username=args[0])
except User.DoesNotExist:
raise CommandError('User with username "%s" not found' % args[0])
try:
token = user.token
except Token.DoesNotExist:
token = Token(user=user)
token.seed = generate_seed()
token.save()
print 'Updated seed for %s to: %s (hex)' % (user.username, token.seed)
print ''
print 'QR Code for Google Authenticator can be found here:'
print ''
print get_qr_url(user.username, token.seed)
print ''
def done(self, form_list, **kwargs):
form_data = [f.cleaned_data for f in form_list]
token = Token(seed=self.get_token_seed(), user=self.request.user)
token.method = form_data[1]['method']
if token.method == 'sms':
token.phone = form_data[2]['phone']
elif token.method == 'call':
token.phone = form_data[2]['phone']
token.save()
return HttpResponseRedirect(settings.LOGIN_REDIRECT_URL)
