sqlalchemy - 如何声明包含多列主键的表类？

Question

主键的列必须按特定顺序排列。

我从文档中看到了一些代码:

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer)

    __mapper_args__ = {
        'primary_key':[id]
    }

但是就是不行(我用的是mysql，不会生成id主键)。任何可能的解决方案？

Answer 1

如果列的声明顺序与它们应该在主键中的顺序相同:

class User(Base):
    field1 = Column(Integer, primary_key=True)
    field2 = Column(Integer, primary_key=True)

否则在__table_args__中声明:

class User(Base):
    field1 = Column(Integer)
    field2 = Column(Integer)
    __table_args__ = (
        PrimaryKeyConstraint(field2, field1),
        {},
    )