Django 创建用户

在 Django 中，我们可以使用内置的 User 模型来创建和管理用户。但是，在大多数情况下，我们需要根据自己的需求来自定义用户模型。在这篇文章中，我们将介绍如何创建和管理用户。

创建用户模型

首先，我们需要创建一个用户模型。我们可以从 Django 的 AbstractBaseUser 和 PermissionsMixin 类中继承，来创建一个自定义的用户模型。

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

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    username = models.CharField(max_length=30, unique=True)
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    date_joined = models.DateTimeField(auto_now_add=True)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username']

    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.username

在上面的代码中，我们定义了一个 User 模型，该模型继承自 AbstractBaseUser 和 PermissionsMixin，并包含以下字段：

• email：电子邮件地址，用于唯一标识用户。
• username：用户名，也可以用来唯一标识用户。
• first_name：名字，可选。
• last_name：姓氏，可选。
• is_staff：标识用户是否可以访问后台管理界面。
• is_active：标识用户是否处于活跃状态。
• date_joined：用户注册时间。

我们还定义了一个 USERNAME_FIELD，来指定用于登录的字段，以及一个 REQUIRED_FIELDS，用于创建用户时必填的字段。

创建用户管理器

接下来，我们需要创建一个用户管理器，用于处理用户的创建、更新和删除等操作。我们可以从 Django 的 BaseUserManager 类中继承，来创建一个自定义的用户管理器。

from django.contrib.auth.models import BaseUserManager

class UserManager(BaseUserManager):
    use_in_migrations = True

    def _create_user(self, email, password, **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_user(self, email, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

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

        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')
        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')

        return self._create_user(email, password, **extra_fields)

在上面的代码中，我们定义了一个 UserManager 类，该类继承自 BaseUserManager，并包含以下方法：

• _create_user：用于创建普通用户。
• create_user：用于创建普通用户。
• create_superuser：用于创建超级用户。

我们使用 _create_user 方法来创建用户，并在 create_user 和 create_superuser 中设置默认值。同时，在 create_superuser 中，我们检查了超级用户的权限是否正确赋值。

替换用户模型

现在，我们已经定义了用户模型和用户管理器，我们需要告诉 Django 使用我们自定义的用户模型。我们可以在 settings.py 文件中添加以下配置：

AUTH_USER_MODEL = 'app_name.User'

将 app_name 替换为你的应用名称，即可使用自定义的用户模型。

创建用户

现在，我们可以使用用户管理器来创建用户了。我们可以在视图函数中调用 create_user 方法来创建普通用户，或者调用 create_superuser 方法来创建超级用户。

from django.contrib.auth import get_user_model
from django.views import View
from django.http import HttpResponse

User = get_user_model()

class CreateUserView(View):
    def post(self, request):
        email = request.POST.get('email')
        password = request.POST.get('password')
        User.objects.create_user(email=email, password=password)
        return HttpResponse('User created.')

在上面的代码中，我们定义了一个视图类 CreateUserView，该视图类继承自 Django 的基础视图类 View。在视图函数中，我们从请求中获取了用户提交的电子邮件地址和密码，并使用用户管理器 User.objects 来创建用户。

结论

本文介绍了如何创建和管理用户，我们定义了用户模型和用户管理器，并替换了 Django 的默认用户模型。使用自定义的用户模型，可以让我们更好地满足项目的需求，同时也能提高我们的开发效率。