Django自定义用户认证示例详解

Django自定义用户认证示例详解

Django提供了默认的用户认证系统，但有时我们需要根据自己的需求进行自定义。下面是一个完整的Django自定义用户认证的攻略。

使用Django内置的auth模块进行用户认证

首先我们来回顾一下Django内置的用户认证系统。在settings.py中配置'django.contrib.auth'后，我们可以使用内置的User模型和auth模块来实现用户认证。下面是一个使用Django内置认证系统的示例：

# views.py

from django.contrib.auth import authenticate, login
from django.http import HttpResponse

def login_view(request):
    if request.method == 'POST':
        username = request.POST['username']
        password = request.POST['password']
        user = authenticate(request, username=username, password=password)
        if user is not None:
            login(request, user)
            return HttpResponse('登录成功')
        else:
            return HttpResponse('用户名或密码错误')
    else:
        return render(request, 'login.html')

在这个示例中，我们首先引入了authenticate和login方法。authenticate方法利用username和password验证用户的身份是否合法，并返回对应的用户实例，若验证失败则返回None；login方法将指定用户标记为已登录。

使用Django内置的用户认证系统可以快速实现用户认证，但有时我们需要更灵活的自定义功能。

自定义用户模型

我们可以通过继承AbstractBaseUser和PermissionsMixin类来自定义用户模型。下面是一个自定义用户模型的示例：

# models.py

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from django.db import models

class UserManager(BaseUserManager):
    def create_user(self, email, password=None,  **extra_fields):
        if not email:
            raise ValueError('The Email field must be set')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self.db)
        return user

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        return self.create_user(email, password, **extra_fields)

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    name = models.CharField(max_length=30, null=True, blank=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)

    USERNAME_FIELD = 'email'

    objects = UserManager()

    def __str__(self):
        return self.email

通过继承AbstractBaseUser和PermissionsMixin类，我们自定义了用户模型，并在UserManager中实现了用户的创建和超级用户的创建方法。上面的示例中，我们新添加了email和name字段，并将email字段设置为用户名。

创建自定义认证后端

为了使用自定义用户模型，我们需要创建自定义的认证后端。下面是一个自定义认证后端的示例：

# backends.py

from django.contrib.auth.backends import BaseBackend
from django.core.exceptions import ObjectDoesNotExist
from .models import User

class EmailBackend(BaseBackend):
    def authenticate(self, request, username=None, password=None, **kwargs):
        try:
            user = User.objects.get(email=username)
            if user.check_password(password):
                return user
        except ObjectDoesNotExist:
            pass

    def get_user(self, user_id):
        try:
            return User.objects.get(pk=user_id)
        except User.DoesNotExist:
            return None

上面的示例中，我们继承了BaseBackend类，并重写了authenticate和get_user方法。在authenticate方法中，我们根据传入的username来查找用户。在此例中，我们使用的是email作为用户名，所以在查找用户时需要使用User.objects.get(email=username)。authenticate方法返回User对象验证成功，否则返回None。get_user方法返回对应ID的用户，如果用户不存在则返回None。

完成以上步骤后，我们需要将自定义认证后端加入Django认证后端的列表中，也就是在settings.py中添加AUTHENTICATION_BACKENDS配置：

AUTHENTICATION_BACKENDS = [
    'myapp.backends.EmailBackend',
    'django.contrib.auth.backends.ModelBackend',
]

在这个配置中，我们先使用自定义认证后端进行认证，如果认证失败则使用Django默认的认证后端。注意自定义的认证后端需要放在默认的认证后端前面。

这样，我们就完成了Django自定义用户认证的全部流程。