Django中如何限制外键仅关联CustomUser的教师类型用户？

背景与现有模型 #

我有一个名为CustomUser的模型，包含教师和学生两种用户类型，通过regd_as字段存储用户类型，模型代码如下：

class CustomUser(AbstractBaseUser, PermissionsMixin):
    
    REGD_AS_CHOICES = (
        ('TCH', 'TEACHER'),
        ('STU', 'STUDENT'),
    )
    
    id = models.UUIDField(primary_key = True, editable = False, default = uuid.uuid4)
    
    email = models.EmailField(unique=True)
    
    date_joined = models.DateTimeField(auto_now_add = True)
    
    regd_as = models.CharField(max_length= 10, choices= REGD_AS_CHOICES, null=True)
    
    is_staff = models.BooleanField(default=False)
    is_active = models.Boolea

现在需要创建一个仅为教师用户生成推荐码的ReferalCode模型，模型代码如下：

class ReferalCode(models.Model):
    
    id = models.UUIDField(primary_key=True, editable=False, default=uuid.uuid4)
    
    teacher_user = models.ForeignKey(CustomUser, on_delete=models.SET_NULL, related_name='teacher_user_referal', null=True)
    
    referal_code = models.CharField(max_length=100, unique=True,null=True, blank=True)

核心问题 #

能否直接将teacher_user字段的外键仅关联CustomUser模型中的教师类型用户？目前我通过信号实现仅存储教师用户到ReferalCode模型中。

回答 #

不能直接通过Django外键字段本身实现仅关联教师类型的CustomUser，因为外键是基于整个模型的，Django ORM不支持直接对外键添加模型内的字段筛选约束（比如只允许regd_as='TCH'的记录）。

不过可以通过以下几种方式强化约束，替代或补充信号的实现：

• 数据库层面添加校验约束：
  在ReferalCode模型的Meta类中添加CheckConstraint，从数据库层面限制关联用户必须是教师：

  class ReferalCode(models.Model):
      # 保留现有字段定义...
      
      class Meta:
          constraints = [
              models.CheckConstraint(
                  check=models.Q(teacher_user__regd_as='TCH'),
                  name='referal_code_teacher_only'
              )
          ]
  

  这种方式能阻止绕过Django业务逻辑的操作（比如直接通过数据库插入非法数据），保障数据完整性。

• 模型层重写save方法校验：
  在ReferalCode的save方法中添加校验逻辑，若关联的不是教师用户则抛出异常：

  def save(self, *args, **kwargs):
      if self.teacher_user and self.teacher_user.regd_as != 'TCH':
          raise ValueError("推荐码仅能关联教师用户")
      super().save(*args, **kwargs)
  

• Admin后台限制可选用户：
  如果使用Django Admin，可在ReferalCodeAdmin类中重写formfield_for_foreignkey，仅展示教师用户选项：

  from django.contrib import admin
  from .models import ReferalCode, CustomUser
  
  class ReferalCodeAdmin(admin.ModelAdmin):
      def formfield_for_foreignkey(self, db_field, request, **kwargs):
          if db_field.name == 'teacher_user':
              kwargs['queryset'] = CustomUser.objects.filter(regd_as='TCH')
          return super().formfield_for_foreignkey(db_field, request, **kwargs)
  
  admin.site.register(ReferalCode, ReferalCodeAdmin)
  

信号的方式属于事后触发校验，结合上述方法可以从数据库、模型、后台多个层面确保只有教师用户能关联到ReferalCode，让约束更严谨。

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