如何修改Django基础Group模型功能？实现组名按公司唯一

问题背景 #

我们有一个Django应用，架构情况如下：

• 存在Company模型，关联多个自定义用户模型（继承AbstractUser），每个用户仅归属一个Company
• 已完成用户模型扩展，添加company字段，实现用户名按公司维度唯一（不同公司允许存在同名用户）
• 现在需要对Django基础Group模型实现相同逻辑：允许不同公司存在同名组，但同一公司内组名必须唯一

此前尝试继承默认Group扩展出CustomGroup模型，添加company字段（与基础Group为一对一关联），功能可正常运行，但无法修改原Group的全局名称唯一约束，实现company与name的联合唯一。现有CustomGroup代码如下：

class CustomGroup(Group):
    company = models.ForeignKey(Company, on_delete=models.CASCADE, blank=True, null=True)
    hidden = models.BooleanField(default=False)

    class Meta:
        indexes = [models.Index(fields=['company'])]

实现思路 #

方案1：重写Group模型（推荐，彻底解决约束问题） #

Django支持通过配置替换默认的Group模型，从根源上实现联合唯一约束，步骤如下：

1. 创建完全自定义的Group模型，不继承默认Group，但保留权限关联逻辑：

from django.db import models
from django.contrib.auth.models import Permission

class CustomGroup(models.Model):
    name = models.CharField(max_length=150)
    company = models.ForeignKey(Company, on_delete=models.CASCADE, blank=True, null=True)
    permissions = models.ManyToManyField(
        Permission,
        blank=True,
        related_name='custom_groups'  # 避免与默认Group的关联名称冲突
    )
    hidden = models.BooleanField(default=False)

    class Meta:
        unique_together = ('name', 'company')  # 核心：实现公司内组名唯一
        indexes = [models.Index(fields=['company', 'name'])]

    def __str__(self):
        return f"{self.name} ({self.company.name if self.company else '全局'})"

2. 在项目settings.py中配置替换默认Group：

AUTH_GROUP_MODEL = '你的应用名.CustomGroup'

3. 注意：此方案更适合项目初期实施；若已有生产数据，需编写数据迁移脚本将原Group数据迁移至新模型。

方案2：信号+自定义验证拦截（兼容现有数据） #

若无法替换默认Group模型，可通过模型验证和信号拦截，在应用层保证联合唯一：

1. 在CustomGroup的clean方法中添加验证逻辑：

from django.core.exceptions import ValidationError

class CustomGroup(Group):
    company = models.ForeignKey(Company, on_delete=models.CASCADE, blank=True, null=True)
    hidden = models.BooleanField(default=False)

    class Meta:
        indexes = [models.Index(fields=['company'])]

    def clean(self):
        super().clean()
        # 验证同一公司内组名唯一
        if self.company:
            exists = CustomGroup.objects.filter(
                name=self.name,
                company=self.company
            ).exclude(pk=self.pk).exists()
            if exists:
                raise ValidationError(f"公司{self.company.name}已存在名为{self.name}的组")
        else:
            # 无公司的组保持全局唯一（可根据需求调整逻辑）
            exists = CustomGroup.objects.filter(name=self.name, company__isnull=True).exclude(pk=self.pk).exists()
            if exists:
                raise ValidationError(f"已存在名为{self.name}的全局组")

2. 注册信号拦截Group模型的保存操作（避免直接操作Group时绕过验证）：

from django.db.models.signals import pre_save
from django.dispatch import receiver
from django.contrib.auth.models import Group
from django.core.exceptions import ValidationError

@receiver(pre_save, sender=Group)
def check_group_name_uniqueness(sender, instance, **kwargs):
    # 关联对应的CustomGroup实例
    try:
        custom_group = CustomGroup.objects.get(pk=instance.pk)
        company = custom_group.company
    except CustomGroup.DoesNotExist:
        company = None

    # 执行联合唯一验证
    if company:
        exists = Group.objects.filter(
            name=instance.name,
            customgroup__company=company
        ).exclude(pk=instance.pk).exists()
        if exists:
            raise ValidationError(f"公司{company.name}已存在名为{instance.name}的组")
    else:
        exists = Group.objects.filter(name=instance.name, customgroup__isnull=True).exclude(pk=instance.pk).exists()
        if exists:
            raise ValidationError(f"已存在名为{instance.name}的全局组")

3. 注意：此方案能覆盖大部分业务场景，但无法阻止直接操作数据库绕过验证，适合已有数据无法迁移的情况。

方案3：数据库层面添加联合唯一约束 #

在数据库层添加联合唯一索引，从底层保证数据一致性，配合应用层验证使用：

1. 生成并执行迁移文件，添加联合唯一约束：

# 迁移文件示例
from django.db import migrations, models

class Migration(migrations.Migration):
    dependencies = [
        ('你的应用名', '上一次迁移的编号'),
    ]

    operations = [
        migrations.AddConstraint(
            model_name='customgroup',
            constraint=models.UniqueConstraint(fields=['name', 'company'], name='unique_group_per_company'),
        ),
    ]

2. 执行迁移后，数据库会直接拦截违反约束的插入/更新操作，配合方案2的验证逻辑，可在应用层给出友好提示。

内容的提问来源于stack exchange，提问作者4bs3nt