From dc9ac7895cf2a127a960b9dc13057942f935b1d8 Mon Sep 17 00:00:00 2001 From: Mahmood Nasr Date: Sun, 24 Mar 2019 15:06:11 +0300 Subject: [PATCH] User Model USERNAME_FIELD setted --- mfa/Email.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mfa/Email.py b/mfa/Email.py index c5d9694..3b17bae 100644 --- a/mfa/Email.py +++ b/mfa/Email.py @@ -9,7 +9,9 @@ from .Common import send def sendEmail(request,username,secret): from django.contrib.auth import get_user_model User = get_user_model() - user=User.objects.get(username=username) + key = getattr(User, 'USERNAME_FIELD', 'username') + kwargs = {key: username} + user = User.objects.get(**kwargs) res=render_to_response("mfa_email_token_template.html",{"request":request,"user":user,'otp':secret}) return send([user.email],"OTP", res.content)