sqlalchemy在多个列中是唯一的

时间:2012-04-07 23:55:16

标签: python sqlalchemy

假设我有一个代表地点的班级。地点“属于”客户。位置由unicode 10字符代码标识。 “位置代码”在特定客户的位置中应该是唯一的。

The two below fields in combination should be unique
customer_id = Column(Integer,ForeignKey('customers.customer_id')
location_code = Column(Unicode(10))

所以,如果我有两个客户,客户“123”和客户“456”。它们都可以有一个名为“main”的位置,但它们都不能有两个名为main的位置。

我可以在业务逻辑中处理这个问题,但我想确保无法在sqlalchemy中轻松添加需求。 unique = True选项似乎仅在应用于特定字段时才起作用,它会导致整个表只有一个唯一的代码用于所有位置。

2 个答案:

答案 0 :(得分:229)

摘自Column的{​​{3}}:

  

唯一 - 如果为True,则表示此列包含唯一   约束,或者如果 index 也为True,则表示索引   应该使用唯一标志创建。在中指定多个列   约束/索引或指定显式名称,使用   <{3}}或documentation明确构建。

由于它们属于表而不属于映射类,因此可以在表定义中声明它们,或者如__table_args__中那样使用声明性:

# version1: table definition
mytable = Table('mytable', meta,
    # ...
    Column('customer_id', Integer, ForeignKey('customers.customer_id')),
    Column('location_code', Unicode(10)),

    UniqueConstraint('customer_id', 'location_code', name='uix_1')
    )
# or the index, which will ensure uniqueness as well
Index('myindex', mytable.c.customer_id, mytable.c.location_code, unique=True)


# version2: declarative
class Location(Base):
    __tablename__ = 'locations'
    id = Column(Integer, primary_key = True)
    customer_id = Column(Integer, ForeignKey('customers.customer_id'), nullable=False)
    location_code = Column(Unicode(10), nullable=False)
    __table_args__ = (UniqueConstraint('customer_id', 'location_code', name='_customer_location_uc'),
                     )

答案 1 :(得分:15)

from flask_sqlalchemy import SQLAlchemy
db = SQLAlchemy()

class Location(Base):
      __table_args__ = (
        # this can be db.PrimaryKeyConstraint if you want it to be a primary key
        db.UniqueConstraint('customer_id', 'location_code'),
      )
      customer_id = Column(Integer,ForeignKey('customers.customer_id')
      location_code = Column(Unicode(10))