在Postgres上使用sqlalchemy创建部分唯一索引

Question

SQLAlchemy支持在postgresql中创建部分索引.

是否可以通过SQLAlchemy 创建部分唯一索引？

想象一下表/模型如下:

class ScheduledPayment(Base):
     invoice_id = Column(Integer)
     is_canceled = Column(Boolean, default=False)

我想要一个唯一的索引,其中给定发票只能有一个"活动"的ScheduledPayment.

我可以在postgres中手动创建:

CREATE UNIQUE INDEX only_one_active_invoice on scheduled_payment 
     (invoice_id, is_canceled) where not is_canceled;

我想知道如何使用SQLAlchemy 0.9将其添加到我的SQLAlchemy模型中.

Answer 1

class ScheduledPayment(Base):
    id = Column(Integer, primary_key=True)
    invoice_id = Column(Integer)
    is_canceled = Column(Boolean, default=False)

    __table_args__ = (
        Index('only_one_active_invoice', invoice_id, is_canceled,
              unique=True,
              postgresql_where=(~is_canceled)),
    )

Answer 2

如果有人通过寻找设置部分唯一约束来停止,其中列可以选择NULL,这是如何:

__table_args__ = (
    db.Index(
        'uk_providers_name_category',
        'name', 'category',
        unique=True,
        postgresql_where=(user_id.is_(None))),
    db.Index(
        'uk_providers_name_category_user_id',
        'name', 'category', 'user_id',
        unique=True,
        postgresql_where=(user_id.isnot(None))),
)

where user_id列是可以的NULL,我希望在所有三列中强制执行唯一约束(name, category, user_id),NULL只是其中一个允许的值user_id.