Django 自定义用户模型、组与权限

在本文中，我们将介绍如何使用 Django 自定义用户模型来管理用户，以及如何使用组和权限来控制用户的访问和操作权限。

Django 自定义用户模型

Django 提供了一个默认的用户模型 django.contrib.auth.models.User，但在某些情况下，我们需要根据具体的应用需求来自定义用户模型。自定义用户模型可以为我们提供更多的灵活性和扩展性。

首先，我们需要在 Django 项目的 settings.py 文件中进行相关配置。将 AUTH_USER_MODEL 属性设置为我们自定义的用户模型的引用，例如：

AUTH_USER_MODEL = 'myapp.MyUser'

接下来，我们需要创建自定义的用户模型，继承自 AbstractBaseUser 和 PermissionsMixin 类。并且我们需要实现一些必需的字段和方法，例如用户名、密码、邮箱等。

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

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

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

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

    objects = MyUserManager()

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

在以上代码中，我们创建了一个 MyUserManager 类来管理用户的创建过程，包括创建普通用户和创建超级用户。MyUser 类继承了 AbstractBaseUser 和 PermissionsMixin 类，并定义了一些必要的字段和方法。

在数据库迁移之后，我们就可以使用自定义的用户模型来处理用户相关的功能了。

组与权限

组与权限是 Django 提供的一种强大的方式来管理用户的访问和操作权限。通过将用户分配到不同的组，并为每个组分配不同的权限，我们可以实现对用户权限的精细控制。

首先，我们可以在 Django 的管理后台创建组，并为每个组分配相应的权限。例如，我们可以创建一个名为 “编辑员” 的组，并为其分配可以编辑文章和评论的权限。

from django.contrib.auth.models import Group, Permission

group = Group.objects.create(name='编辑员')
permission1 = Permission.objects.get(codename='add_article')
permission2 = Permission.objects.get(codename='change_article')
permission3 = Permission.objects.get(codename='delete_article')
group.permissions.add(permission1, permission2, permission3)

然后，我们可以将用户添加到相应的组中。例如，我们可以将一个用户添加到 “编辑员” 组中：

from django.contrib.auth import get_user_model

User = get_user_model()
user = User.objects.get(username='example')
user.groups.add(group)

接下来，我们就可以根据用户的组与权限来进行权限控制了。例如，我们可以使用装饰器 @permission_required 来限制只有具有特定权限的用户才能访问某个视图:

from django.contrib.auth.decorators import permission_required
from django.shortcuts import render

@permission_required('app.change_article')
def edit_article(request, article_id):
    # 编辑文章的逻辑
    return render(request, 'edit_article.html')

在上面的例子中，只有具有 “change_article” 权限的用户才能访问 edit_article 视图。