From 3f37fe4826d66eabebd1d8716bddbaef1ba1c48f Mon Sep 17 00:00:00 2001 From: PCoder Date: Thu, 26 Sep 2019 15:10:16 +0530 Subject: [PATCH] Set username for non-superusers only --- otpauth/models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/otpauth/models.py b/otpauth/models.py index 6e1898e..6e000d3 100644 --- a/otpauth/models.py +++ b/otpauth/models.py @@ -21,7 +21,11 @@ class OTPSeed(AbstractUser): """ inject username to ensure it stays unique / is setup at all """ - self.username = "{}@{}".format(self.name, self.realm) + if not self.is_superuser: + self.username = "{}@{}".format(self.name, self.realm) + else: + self.name = self.username + super().save(*args, **kwargs) def __str__(self):