From 9703eb6538dc8cc3863f1e043838a63cbcc94d8f Mon Sep 17 00:00:00 2001 From: PCoder Date: Thu, 26 Sep 2019 15:26:10 +0530 Subject: [PATCH] Set name, realm and seed for superusers also --- otpauth/models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/otpauth/models.py b/otpauth/models.py index 6e000d3..81ae71f 100644 --- a/otpauth/models.py +++ b/otpauth/models.py @@ -4,6 +4,7 @@ from rest_framework import exceptions from rest_framework import authentication import json import logging +import pyotp logger = logging.getLogger(__name__) @@ -25,6 +26,8 @@ class OTPSeed(AbstractUser): self.username = "{}@{}".format(self.name, self.realm) else: self.name = self.username + self.realm = "ungleich-admin" + self.seed = pyotp.random_base32() super().save(*args, **kwargs)