indexing - SQLAlchemy根据继承的表键在mixin中添加索引？

Question

在 SQLAlchemy 中，我想定义一个在继承表中自动创建索引的 mixin。

假设继承表有一个名为“keys”的成员列表，我希望 mixin 在继承表中创建一个对键中列出的列的单个多列索引。但是，在创建表之前，mixin 不知道键是什么！

我该怎么做？

Answer 1

例子：

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.declarative import declared_attr
from sqlalchemy import MetaData, Column, Index, Integer

metadata = MetaData()
Base = declarative_base(metadata=metadata)


class MyMixin(object):

    @declared_attr
    def __table_args__(cls):
        return (Index('test_idx_%s' % cls.__tablename__, *cls.INDEX),)


class MyModel(MyMixin, Base):
    __tablename__ = 'atable'
    INDEX = ('a', 'b',)
    id = Column(Integer, primary_key=True)
    a = Column(Integer)
    b = Column(Integer)
    c = Column(Integer)


if __name__ == '__main__':
    from sqlalchemy import create_engine
    engine = create_engine('sqlite:///', echo=True)
    metadata.bind = engine
    metadata.create_all() 

文档：组合来自多个 Mixin 的表/映射器参数