SQLAlchemy：可以声明性地将列声明为主键吗？

Question

我有一个基类mixin：

class MyColumns(object):
    id = Column(Integer)
    foo = Column(Integer)
    bar = Column(Integer)

class MyMainTable(MyColumns, Base):
    __tablename__ = 'main_table'
    pk_id = PrimaryKeyConstraint("id")

我希望能够将id声明为MyMainTable上的PK。 我无法在MyColumns其声明为PK，因为我需要在另一个表中使用MyColumns ，其中id不是PK（出于审计目的而做）。 当我运行上面的代码时，我得到

sqlalchemy.exc.ArgumentError: Mapper Mapper|MyMainTable|main_table could not assemble any primary key columns for mapped table 'main_table'

有什么办法可以这样添加PK声明吗？

Answer 1

我找到了此处记录的解决方案： http : //docs.sqlalchemy.org/en/rel_0_9/core/constraints.html#setting-up-constraints-when-using-the-declarative-orm-extension

您需要将约束添加到__table_args__ ：

class MyColumns(object):
    id = Column(Integer)
    foo = Column(Integer)
    bar = Column(Integer)

class MyMainTable(MyColumns, Base):
    __tablename__ = 'main_table'
    __table_args__ = (
        PrimaryKeyConstraint("id", name="pk_id"),
    )