如何基于ForeignKey的值为Django模型字段设置默认值

你遇到的错误是因为在模型定义阶段，owner是ForeignKey字段对象，而非具体的Account实例，此时无法访问units属性——只有当创建CustomExercise实例时，owner才会指向一个实际的Account对象。

以下是几种可行的实现方案：

方案一：重写save方法（推荐） #

直接在CustomExercise模型中重写save方法，在新建对象时自动从关联的Account中获取units作为默认值：

class CustomExercise(models.Model):
    # 复用Account的单位选项，避免重复代码
    UNIT_CHOICES = Account.UNIT_CHOICES

    uuid = models.UUIDField(default=uuid.uuid4, unique=True, primary_key=True, editable=False)
    created = models.DateTimeField(auto_now_add=True)

    owner = models.ForeignKey(Account, on_delete=models.CASCADE, null=False, blank=False)
    # 移除错误的default参数
    preferred_units = models.CharField(max_length=255, choices=UNIT_CHOICES, null=False, blank=False)
    name = models.CharField(max_length=255, null=False, blank=False)

    measure_time = models.BooleanField(default=False)
    measure_distance = models.BooleanField(default=False)
    measure_weight = models.BooleanField(default=False)
    measure_reps = models.BooleanField(default=False)

    def save(self, *args, **kwargs):
        # 仅在新建对象（无主键）且未手动设置preferred_units时生效
        if not self.pk and not self.preferred_units:
            self.preferred_units = self.owner.units
        # 调用父类的save方法完成保存
        super().save(*args, **kwargs)

    def __str__(self):
        return f'{self.owner}:{self.name}'

方案二：使用Django信号 #

如果需要在多个模型中复用类似逻辑，可以用pre_save信号在保存前自动设置默认值：

from django.db.models.signals import pre_save
from django.dispatch import receiver

@receiver(pre_save, sender=CustomExercise)
def set_preferred_units_default(sender, instance, **kwargs):
    # 新建对象且未设置preferred_units时，从关联Account取值
    if not instance.pk and not instance.preferred_units:
        instance.preferred_units = instance.owner.units

注意：信号需要在Django启动时加载，通常放在models.py或单独的signals.py文件中（若用单独文件，需在apps.py的ready方法中导入）。

方案三：创建实例时手动设置 #

如果只在特定视图中创建CustomExercise，可以在创建时直接传入关联Account的units：

# 假设已获取当前用户的Account实例account
exercise = CustomExercise.objects.create(
    owner=account,
    name="深蹲",
    preferred_units=account.units,
    measure_weight=True,
    # 其他字段...
)

这种方式适合逻辑简单、创建场景单一的情况，但需要在所有创建CustomExercise的地方都手动处理。

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