我想实现一个登录系统,以通过电子邮件或用户名登录Django Admin。有人知道如何实施吗?我使用自定义用户模型。我知道如何允许用户使用用户名或电子邮件登录网站。但是它在Django Admin界面中不起作用
from django.conf import settings
from django.contrib.auth import get_user_model
class EmailOrUsernameModelBackend(object):
def authenticate(self, username=None, password=None):
if '@' in username:
kwargs = {'email': username}
else:
kwargs = {'username': username}
try:
user = get_user_model().objects.get(**kwargs)
if user.check_password(password):
return user
except User.DoesNotExist:
return None
def get_user(self, username):
try:
return get_user_model().objects.get(pk=username)
except get_user_model().DoesNotExist:
return None
您需要在authenticated
方法中传递请求:
from django.conf import settings
from django.contrib.auth import get_user_model
class EmailOrUsernameModelBackend(object):
def authenticate(self, request, username=None, password=None):
if '@' in username:
kwargs = {'email__iexact': username}
else:
kwargs = {'username': username}
try:
user = get_user_model().objects.get(**kwargs)
if user.check_password(password):
return user
except get_user_model().DoesNotExist:
return None
def get_user(self, username):
try:
return get_user_model().objects.get(pk=username)
except get_user_model().DoesNotExist:
return None
另外,请确保将其放入 AUTHENTICATION_BACKENDS
中:
AUTHENTICATION_BACKENDS = ['path.to.EmailOrUsernameModelBackend']
您可以从UserManager
类创建自定义用户经理子类别,并覆盖get_by_natural_key
方法。
from django.contrib.auth.models import UserManager
class UserAccountManager(UserManager):
def get_by_natural_key(self, username):
return self.get(
models.Q(**{self.model.USERNAME_FIELD: username})
| models.Q(**{self.model.EMAIL_FIELD: username})
)
请注意,这将通过电子邮件或密码而不仅在管理页面上对系统中的所有用户进行身份验证。