根据另一个字段的值对 Django 模型添加约束

Question

我有一个带有 1 个主键和 3 个字段的简单模型（简化）：

• 及格分数
• 最高分
• 最大尝试次数

此模型是通过继承 django.db.models 创建的。这是最小的可重现代码：

from django.db import models
class QuestionSet(models.Model):
    passingscore = models.PositiveSmallIntegerField("passing score")
    maxscore = models.PositiveSmallIntegerField("max score")
    maxattempt = models.PositiveSmallIntegerField("max attempt")

我想添加一个约束，使passingscore 在数据库中永远不会大于maxscore。

根据this thread on StackOverflow，我使用了跨越多个字段的约束，例如unique_together。但显然这是一个不同的用例。

我也简单考虑过直接写PostgreSQL代码加约束：

ALTER TABLE tableB ADD CONSTRAINT score_policy_1 CHECK (maxscore >= passingscore) 

但这违背了使用 ORM 的目的，违反了“松散耦合”的理念，使得在不同的数据库后端之间迁移变得困难。

如果这是可能的，请指出在 Django 中编写此约束的更惯用的方式。

Answer 1

这应该适用于您的情况。

from django.db import models
class QuestionSet(models.Model):
    passingscore = models.PositiveSmallIntegerField("passing score")
    maxscore = models.PositiveSmallIntegerField("max score")
    maxattempt = models.PositiveSmallIntegerField("max attempt")
    
    class Meta:
        constraints = [
            models.CheckConstraint(
                name="%(app_label)s_%(class)s_passingscore_lte_maxscore",
                check=models.Q(passingscore__lte=models.F("maxscore")),
            )
        ]

Answer 2

您可以在视图层或数据库层中执行此操作。

如果你想在DB层做，你可以覆盖模型的save方法，并检查那里的值：

def save(self, *args, **kwargs):

    if self.passingscore > self.maxscore:
        raise ValueError("passingscore can't be greater than maxscore")

    super().save(*args, **kwargs)

---

如果你想处理数据库约束，你可以通过继承BaseConstraint创建你自己的约束类，并将你的sql放在constraint_sql方法中。