Flask SQLAlchemy对具有给定值的多个列的唯一约束

Question

假设我有两张桌子

订单表

class Order(db.Model):
  id = db.Column(db.Integer, primary_key=True)
  deliveries = db.relationship("Delivery", back_populates="order")

交货表

class Delivery(db.Model):
  id = db.Column(db.Integer, primary_key=True)
  status= db.Column(db.String(20))
  order_id = db.Column(db.Integer, db.ForeignKey('order.id'), nullable=False)
  order = db.relationship("Order", back_populates="deliveries")

一个订单可能有多个交货。交付表的状态字段有以下可能的值["Planned", "Failed", "Delivered"]

问题： 如何确保一个订单没有两个计划交货？

提示： 我知道我可以对多个表实施唯一约束，例如在交付模式中添加下一行

__table_args__ = (db.UniqueConstraint('order_id', 'status', name='_order_status_uc'))

但是通过这样做，我将防止一个订单出现两次失败的交付。

任何帮助将不胜感激，因为我想直接在数据库中强制执行此约束。

Answer 1

更新（假设是 PostgreSQL）

鉴于您使用 PostgreSQL，最直接的解决方案是使用Partial Unique Index：

CREATE UNIQUE INDEX _order_status_uc ON delivery (order_id) WHERE status = 'Planned';

如果你使用 SQLAlchemy 创建索引，下面应该是等价的：

类交付（基础）： # ...

__table_args__ = (
    Index(
        '_order_status_uc',
        'order_id',
        unique=True,
        postgresql_where=(status == 'Planned'),
    ),
)

原答案

假设您使用的 RDBMS 支持这一点，一种实现思路是：

• 在 delivery 表上创建一个计算列，对于“已计划”status 和 NULL 的值为 order_id 否则
• 在此计算列上创建一个UNIQUE Constraint。

这假设以下关于使用的 RDBMS：

1. 计算列不仅受支持，还可以成为索引/约束的一部分。
2. UNIQUE 索引的实现允许多个NULL 值。

请参阅PostgreSQL unique constraint null: Allowing only one Null 文章了解更多背景信息。

Answer 2

您可以使用当前订单ID和计划状态查询您的交付模式，如果有结果，请进行验证。

planned_delivery_exist_boolean = db.session.query(Delivery).filter(Delivery.order_id==order_id).filter(Delivery.status=="Planned").first()

if planned_delivery_exist_boolean:
    # trigger validation error