Django 中怎么自定义用户模块

在 Django 中，用户模块是通过 Django 自带的 auth 应用提供的，但是在实际开发中，我们可能需要对用户模块进行自定义以满足特定的需求，这篇文章将讲解如何在 Django 中自定义用户模块。

## 概述

Django 自带的 auth 应用提供了默认的用户模块，包含了常用的用户认证、授权和用户管理等功能。但是在实际开发中，这些功能可能并不能满足我们的需求，特别是当我们需要增加一些自定义的字段时，就需要对默认的用户模块进行自定义。

在 Django 中，我们可以通过继承 AbstractBaseUser 和 BaseUserManager 类来自定义用户模块，并在 settings.py 文件中修改 AUTH_USER_MODEL 设置，将自定义的用户模块替换掉默认的用户模块。下面我们就来一步一步实现自定义用户模块。

## 创建自定义用户模块

首先，在你的 Django 项目的某个目录下创建一个新的名为 accounts 的应用：

$ python manage.py startapp accounts

接着，在 accounts 应用的 models.py 文件中定义自定义用户模块：

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

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=None, **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):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    date_joined = models.DateTimeField(auto_now_add=True)

    objects = UserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    def __str__(self):
        return self.email

    def get_full_name(self):
        return f'{self.first_name} {self.last_name}'

    def get_short_name(self):
        return self.first_name

    class Meta:
        verbose_name = 'user'
        verbose_name_plural = 'users'

上面的代码中，我们继承了 AbstractBaseUser 和 BaseUserManager 类，并在 User 模型中添加了自定义的字段 email、first_name、last_name 等。同时在 UserManager 类中，我们重写了 create_user 方法和 create_superuser 方法，以便创建自定义模型的用户和超级用户，并在最后将 UserManager 类中添加到 User 模型的 objects 属性中，以便在创建用户时可以使用 UserManager 中的方法。

另外，在自定义模型中，我们需要指定 USERNAME_FIELD 属性，用于指定用户名字段（即在登录时需要输入的字段），而且在 REQUIRED_FIELDS 属性中，我们也可以添加其他必填字段。

## 修改 AUTH_USER_MODEL

接下来，我们需要修改 settings.py 文件，将 AUTH_USER_MODEL 设置为我们刚才创建的自定义用户模型：

AUTH_USER_MODEL = 'accounts.User'

这样，我们就完成了自定义用户模型的创建和设置。

## 迁移数据库

最后，我们需要对数据库进行迁移，以便在数据库中添加新的自定义用户模型：

$ python manage.py makemigrations accounts
$ python manage.py migrate accounts

## 使用自定义用户模块

现在，我们已经完成了自定义用户模块的创建和设置，并迁移了数据库以添加新的模型。下面我们可以使用自定义模型来创建用户并进行认证和授权等操作了。

在 views.py 文件中，我们可以通过 Django 自带的 auth 应用的 authenticate 和 login 函数，来实现用户认证和登录：

from django.shortcuts import render, redirect
from django.contrib.auth import authenticate, login, logout

def login_view(request):
    if request.method == 'POST':
        email = request.POST.get('email')
        password = request.POST.get('password')
        user = authenticate(request, email=email, password=password)
        if user is not None:
            login(request, user)
            return redirect('home')
        else:
            error_msg = 'Invalid email or password'
            return render(request, 'accounts/login.html', {'error_msg': error_msg})
    else:
        return render(request, 'accounts/login.html')

def logout_view(request):
    logout(request)
    return redirect('login')

在上面的代码中，我们从 POST 请求中获取用户输入的 email 和 password，然后调用 authenticate 函数来验证用户的认证信息。如果认证成功，则调用 login 函数实现用户登录，并重定向到指定的页面；反之，返回登录页面并提示错误消息。

另外，在 templates 目录下我们需要创建一个名为 login.html 的模板文件，用于展示登录表单，代码如下：

{% extends 'base.html' %}

{% block content %}
    <h1>Login</h1>
    {% if error_msg %}
        <p>{{ error_msg }}</p>
    {% endif %}
    <form method="post">
        {% csrf_token %}
        <div>
            <label for="email">Email:</label>
            <input type="text" name="email" required>
        </div>
        <div>
            <label for="password">Password:</label>
            <input type="password" name="password" required>
        </div>
        <button type="submit">Login</button>
    </form>
{% endblock %}

我们可以通过类似上面的方式，在 Django 中自定义用户模块，并实现用户认证和管理等功能，以满足不同的需求。