2012-07-12 120 views
0

在SQLAlchemy中,我想定义一个mixin,它会自动在继承表中创建一个索引。SQLAlchemy基于继承的表键在mixin中添加索引?

假设继承表有一个名为'keys'的成员列表,我希望mixin在继承表中创建一个多列索引,而不是在键中列出的列中。但是,在创建表之前,mixin不知道密钥是什么。

我该怎么做?

回答

2

实施例:

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

文档:Combining Table/Mapper Arguments from Multiple Mixins