Django自定义组权限：实现角色自动关联权限的需求

你可以通过两种方式实现这个需求：重写用户模型的save方法，或者利用Django的信号机制，下面是具体实现：

方法一：重写用户模型的save方法 #

在你的Users模型中重写save方法，当用户的角色被设置或修改时，自动同步角色关联的组到用户的组列表中：

from django.contrib.auth.models import AbstractUser, Group
from django.db import models

class Role(models.Model):
    title = models.CharField(max_length=32)
    groups = models.ManyToManyField(Group)

    def __str__(self) -> str:
        return self.title

class Users(AbstractUser):
    role = models.ForeignKey(
        Role, on_delete=models.CASCADE, related_name='roles', blank=True, null=True)

    def __str__(self):
        return self.username

    def save(self, *args, **kwargs):
        # 先保存用户实例，确保role字段已更新
        super().save(*args, **kwargs)
        
        if self.role:
            # 清除用户现有所有组，避免旧权限残留
            self.groups.clear()
            # 添加角色关联的所有组
            self.groups.add(*self.role.groups.all())
        else:
            # 如果角色被清空，也清空用户的组
            self.groups.clear()

方法二：使用Django的post_save信号 #

如果不想修改用户模型的save方法，可以用信号监听用户模型的保存事件，触发组同步逻辑：

# 在你的models.py或者单独的signals.py文件中
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.contrib.auth.models import Group
from .models import Users, Role

@receiver(post_save, sender=Users)
def sync_user_role_groups(sender, instance, created, **kwargs):
    if instance.role:
        # 清除旧组并添加角色关联的新组，set方法会自动覆盖现有组
        instance.groups.set(instance.role.groups.all())
    else:
        # 角色为空时清空用户组
        instance.groups.clear()

如果把信号放在单独的signals.py文件中，需要在应用的apps.py中注册信号：

# apps.py
from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.signals

注意事项 #

• 两种方法都会在用户角色设置/修改时自动同步组权限，无需手动操作
• 使用clear()或set()方法可以避免用户保留旧角色的权限，确保权限完全匹配当前角色
• 如果需要保留用户自身额外添加的组权限（而非仅继承角色的组），可以去掉clear()或set()，改用add()，但这样可能导致权限叠加，需根据业务需求选择

内容的提问来源于stack exchange，提问作者Apollo Vostok