def __init__(self, user, *args, **kwargs):
super(ResetTwoFactorAuthForm, self).__init__(*args, **kwargs)
if user:
try:
self.token = UserAuthToken.objects.get(user=user)
self.fields["type"].initial = self.token.type
except UserAuthToken.DoesNotExist:
self.token = UserAuthToken(user=user)
else:
self.token = None
def save(self):
if not self.user:
return None
try:
token = UserAuthToken.objects.get(user=self.user)
except UserAuthToken.DoesNotExist:
token = UserAuthToken(user=self.user)
token.encrypted_seed = encrypt_value(random_seed(30))
token.save()
return token
def save(self):
try:
token = UserAuthToken.objects.get(user=self.user)
except UserAuthToken.DoesNotExist:
token = UserAuthToken(user=self.user)
base36_with_checksum = self.cleaned_data["key"]
seed = util.key_to_seed(base36_with_checksum)
token.type = UserAuthToken.TYPE_HOTP
token.reset_seed(seed)
# Start at the second code
token.counter = 1
token.save()
