Django自定义用户验证字段-重写authenticate方法

在登录过程中不一定只需要username和password来进行登录,可能还会需要手机号加密码邮箱加密码等。

一、models.py

from django.db import models
from django.contrib.auth.models import AbstractUser

# Create your models here.


class UserProfile(AbstractUser):
    """
    用户
    """
    name = models.CharField(max_length=30, null=True, blank=True, verbose_name="姓名")
    birthday = models.DateField(null=True, blank=True, verbose_name="出生年月")
    gender = models.CharField(max_length=6, choices=(("male", u"男"), ("female", "女")), default="female", verbose_name="性别")
    mobile = models.CharField(null=True, blank=True, max_length=11, verbose_name="电话")
    email = models.EmailField(max_length=100, null=True, blank=True, verbose_name="邮箱")

继承AbstractUser基类,重写用户表

二、settings.py

AUTH_USER_MODEL = 'users.UserProfile'

使用自定义的用户表

AUTHENTICATION_BACKENDS = {
    'users.views.CustomBackend'
}

自定义验证类

from django.contrib.auth.backends import ModelBackend
from django.db.models import Q
from django.shortcuts import render

# Create your views here.
from .models import UserProfile


class CustomBackend(ModelBackend):
    """
    自定义用户验证
    """
    def authenticate(self, request, username=None, password=None, **kwargs):
        try:
            user = UserProfile.objects.get(Q(username=username) | Q(mobile=username))
            if user.check_password(password):
                return user
        except Exception as e:
            return None

验证类相关代码-使用用户名或者手机号进行登录