Django多模型下实现用户名仅在公司内唯一的方案咨询

方案一：PostgreSQL部分表达式唯一索引（数据库层面兜底） #

这是最可靠的实现方式，直接从数据库层面约束数据唯一性，能避免代码逻辑漏洞导致的重复数据。PostgreSQL支持表达式索引和部分索引，刚好适配你的场景：

实现步骤 #

1. 生成空迁移文件
   执行命令创建一个空的迁移文件：

python manage.py makemigrations --empty your_app_name

2. 添加索引创建逻辑
   打开生成的迁移文件，在operations中加入RunSQL操作，创建跨表的唯一索引：

from django.db import migrations, models

class Migration(migrations.Migration):
    dependencies = [
        ('your_app_name', '00xx_your_previous_migration'),  # 替换为你的上一个迁移文件名
    ]

    operations = [
        migrations.RunSQL(
            sql="""
            CREATE UNIQUE INDEX unique_employee_username_per_company ON your_app_name_employee (
                (SELECT username FROM your_app_name_account WHERE your_app_name_account.id = your_app_name_employee.account_id),
                company_id
            );
            """,
            reverse_sql="""
            DROP INDEX IF EXISTS unique_employee_username_per_company;
            """
        )
    ]

这个索引会确保：同一个company_id下，关联的Account的username唯一，且仅作用于有Employee记录的账号（即员工类型账号），企业类型账号无Employee记录，不受此约束。

3. 执行迁移

python manage.py migrate

方案二：代码层面校验（提前拦截错误） #

数据库约束会抛出IntegrityError，为了给用户更友好的提示，需要在代码层面添加前置校验，比如在模型的clean方法中处理：

1. 在Employee模型中添加校验 #

from django.core.exceptions import ValidationError
from django.db import models

class Employee(models.Model):
    account = models.OneToOneField(Account, on_delete=models.CASCADE)
    company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name='employees')

    def clean(self):
        super().clean()
        # 排除当前记录（适配更新场景），检查同公司下是否有重复用户名
        duplicate = Employee.objects.filter(
            company=self.company,
            account__username=self.account.username
        ).exclude(pk=self.pk).exists()
        if duplicate:
            raise ValidationError(f"用户名 '{self.account.username}' 在本公司已存在，请更换")

    def save(self, *args, **kwargs):
        self.full_clean()  # 保存前强制触发校验
        super().save(*args, **kwargs)

2. 在Account模型中补充校验（针对员工类型账号） #

如果直接创建员工类型的Account并关联Employee，可在Account的clean方法中补充校验：

class Account(auth_models.AbstractBaseUser):
    username = models.CharField(max_length=150)
    ACCOUNT_TYPE_CHOICES = [
        ('employee', '员工'),
        ('company', '企业'),
    ]
    account_type = models.CharField(max_length=20, choices=ACCOUNT_TYPE_CHOICES)

    def clean(self):
        super().clean()
        if self.account_type == 'employee' and hasattr(self, 'employee') and self.employee.company:
            duplicate = Employee.objects.filter(
                company=self.employee.company,
                account__username=self.username
            ).exclude(account=self).exists()
            if duplicate:
                raise ValidationError(f"用户名 '{self.username}' 在所属公司已存在")

注意事项 #

• 数据库约束是最后一道防线，必须保留，避免因批量操作、直接ORM调用等跳过代码校验的场景导致重复数据。
• 使用Django表单或DRF序列化器时，确保调用is_valid()方法，会自动触发模型的clean校验逻辑。
• 迁移时注意替换代码中的your_app_name为实际应用名称，以及迁移依赖的上一个迁移文件编号。

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