在`UniqueConstraint`中创建`CheckConstraint`

时间:2015-06-10 10:21:57

标签: python postgresql sqlalchemy

在创建新记录之前,我想检查一个条件,即native_linux_useris_active的组合是唯一的,但is_active必须是True。可能存在多个native_linux_user is_active=False,但只有一个native_linux_user可以存在is_active=True

我尝试在这样的CheckConstraint内使用UniqueConstraint,但它不起作用。我该如何制作这种约束?

 __table_args__ = (
     UniqueConstraint(
         'native_linux_user',
         CheckConstraint('is_active=True', name='active_check'),
         name='_username_uniqueness'
     ),
 )

1 个答案:

答案 0 :(得分:3)

从SQL端,您可以创建部分UNIQUE索引:

CREATE UNIQUE INDEX idx_unique_native_linux_user_is_active
    ON table (native_linux_user) WHERE is_active=True;

和SQLAlchemy模型以及相应的Index

class Foo(Model):
    id = Column(Integer, primary_key=True)
    native_linux_user = Column(String, nullable=False)
    is_active = Column(Boolean)

    __table_args__ = (
        Index('idx_unique_native_linux_user_is_active', native_linux_user,
            unique=True,
            postgresql_where=(is_active==True),
        ),
    )