Django 注册自定义用户模型并使用管理员权限

在本文中，我们将介绍如何在Django中注册自定义用户模型并使用管理员权限。Django是一个强大的Python Web框架，它提供了可扩展的用户认证系统，允许我们快速创建和管理用户。

创建自定义用户模型

首先，我们需要创建一个自定义用户模型。在Django中，我们可以通过继承AbstractBaseUser类和BaseUserManager类来实现这一目标。

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


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

    def create_superuser(self, email, password):
        user = self.create_user(email=email, password=password)
        user.is_staff = True
        user.is_superuser = True
        user.save(using=self._db)
        return user


class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=30)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['first_name', 'last_name']

    def __str__(self):
        return self.email

在上述代码中，我们定义了一个CustomUserManager类，该类继承自BaseUserManager，并提供了创建用户和创建超级用户的方法。然后，我们定义了一个CustomUser类，该类继承自AbstractBaseUser和PermissionsMixin。我们在CustomUser类中添加了一些必要的字段，如电子邮件、名字等，并指定了USERNAME_FIELD和REQUIRED_FIELDS。

注册自定义用户模型

在我们创建自定义用户模型后，需要将它注册到Django的用户认证系统中。为此，我们需要在settings.py文件中进行相应的配置。

# settings.py

AUTH_USER_MODEL = 'myapp.CustomUser'

上述代码中，我们将AUTH_USER_MODEL设置为我们自定义的用户模型路径，其中myapp是我们自己的应用程序名称。

使用管理员权限管理自定义用户模型

一旦我们注册了自定义用户模型，我们就可以使用管理员权限来管理用户。Django的管理员界面提供了方便的界面来管理用户及其相关信息。

首先，我们需要在admin.py文件中导入相关模块，并将我们的自定义用户模型注册到管理员站点。

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin

from .models import CustomUser


class CustomUserAdmin(UserAdmin):
    list_display = ('email', 'first_name', 'last_name', 'is_staff')
    search_fields = ('email', 'first_name', 'last_name')
    ordering = ('email',)
    filter_horizontal = ()


admin.site.register(CustomUser, CustomUserAdmin)

在上述代码中，我们定义了一个自定义的CustomUserAdmin类，该类继承自Django的UserAdmin类，并设置了在管理员界面中显示的字段、可搜索的字段及排序字段。