уникальная sqlalchemy в нескольких столбцах

Допустим, у меня есть класс, который представляет местоположения. Места "принадлежат" клиентам. Местоположения обозначены 10-символьным кодом Unicode. «Код местоположения» должен быть уникальным среди местоположений для конкретного клиента.

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

Так что, если у меня есть два клиента, клиент "123" и клиент "456". Они оба могут иметь местоположение, называемое «основным», но ни одно из них не может иметь двух мест, называемых основным.

Я могу справиться с этим в бизнес-логике, но хочу убедиться, что нет никакого способа легко добавить требование в sqlalchemy. Опция unique = True работает только применительно к определенному полю, и это приведет к тому, что вся таблица будет иметь уникальный код только для всех местоположений.

Выписка из документации из Column:

unique - Когда True, указывает, что этот столбец содержит ограничение уникальности, или, если index также имеет значение True, указывает, что Индекс должен быть создан с флагом уникальности. Чтобы указать несколько столбцов в ограничении / индексе или указать явное имя, явно используйте конструкции UniqueConstraint или Index .

Так как они принадлежат Таблице, а не сопоставленному Классу, они объявляются в определении таблицы или, если используются декларативные, как в __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'),
                     )

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))