python - 清除 Django Admin 或表单中预期的 ExclusionConstraint IntegrityError

Question

文档展示了一种接受 PostgreSQL 提供的数据完整性约束的方法，例如重叠范围的ExclusionConstraint。 您可以从文档 here 阅读建议的解决方案。 我想要一个预订系统，确保一个“东西”(这里是培训师/老师)不能在重叠的时间段内被预订两次。我将使用文档中的第二个示例，其中重叠标准源自现有字段:

from django.contrib.postgres.constraints import ExclusionConstraint
from django.contrib.postgres.fields import (
    DateTimeRangeField,
    RangeBoundary,
    RangeOperators,
)
from django.db import models
from django.db.models import Func, Q


class TsTzRange(Func):
    function = 'TSTZRANGE'
    output_field = DateTimeRangeField()


class Reservation(models.Model):
    trainer = models.ForeignKey('Trainer', on_delete=models.CASCADE)
    start = models.DateTimeField()
    end = models.DateTimeField()
    cancelled = models.BooleanField(default=False)

    class Meta:
        constraints = [
            ExclusionConstraint(
                name='exclude_overlapping_reservations',
                expressions=(
                    (TsTzRange('start', 'end', RangeBoundary()), RangeOperators.OVERLAPS),
                    ('trainer', RangeOperators.EQUAL),
                ),
                condition=Q(cancelled=False),
            ),
        ]

所以，这对我来说效果很好，当尝试保存无效范围时，我会得到预期的 IntegrityError:

IntegrityError at /admin/trainer/trainingevent/add/

conflicting key value violates exclusion constraint "exclude_overlapping_reservations"
DETAIL:  Key (tstzrange(start, "end", '[)'::text), trainer_id)=(["2020-12-19 16:20:00+00","2020-12-19 16:55:00+00"), 1) conflicts with existing key (tstzrange(start, "end", '[)'::text), trainer_id)=(["2020-12-19 16:15:00+00","2020-12-19 16:45:00+00"), 1).

这引出了我的问题: 如何验证字段，或者更确切地说，创建适当的 clean() 方法来验证输入而不重复功能？ 从我目前的角度来看，最好的办法是要求 PostgreSQL 以某种方式进行检查 - 或者以某种方式将模型保存在 try catch block 中。

因此，在更一般的情况下，问题应该等于“如何清除 Django 中的 IntegrityError”。

遗憾的是，我在文档中或其他任何地方都找不到任何有关此内容的内容，因此任何提示都值得赞赏，并提前致谢。

Answer 1

所以，我想不可能重用约束来查询重叠事件，因为在 Django 尝试将对象保存到数据库时，它已经必须被清理，至少从我能发现的情况来看是这样。 无论如何，我当然会让约束保持事件状态，因为这是对我的数据完整性的有力保证，但为了获得我正在寻找的功能，我在模型中添加了一个自定义管理器。

如下所示，它符合“预订房间”模式(只需将 Trainer 替换为 Room)，但您需要根据您的需要对其进行修改。

from django.db import models


class TrainingEventManager(models.Manager):
    def check_availability(self, trainer, start, end):
        return not (
            super()
            .get_queryset()
            .filter(trainer=trainer)
            .filter(start__gte=start)
            .filter(end__lte=end)
            .exists()
        )

查询给定的日期时间范围(start__gte 和 end__lte)，检查它是否有条目(exists())并返回 bool 值相反，因为我想知道可用性而不是“is_reserved”答案。