Sqlalchemy __table_args__的声明性继承

时间:2014-11-30 08:16:52

标签: python sqlalchemy flask-sqlalchemy

假设我有这样的表

class Base(object):
    id = Column(Integer, primary_key=True)
    date_updated = Column(Date, nullable=True)
    rating = Column(Integer, nullable=False, default=0)
    status = Column(SmallInteger, nullable=False, default=0)

    @declared_attr
    def __tablename__(cls):
        return "table_%s" % cls.LANG

    @declared_attr
    def __table_args__(cls):
        return (
            Index('ix__%s__rating' % cls.__tablename__, 'rating'),
            Index(
                'ix__%s__status' % cls.__tablename__,
                'status',
                postgresql_where=column('status') == 0,
            ),
            Index(
                'ix__%s__date_updated' % cls.__tablename__,
                'date_updated',
                postgresql_where=column('date_updated') != None,
            ),
        )

   class TableEn(Base, db.Model):
       LANG = 'en'


   class TableUk(Base, db.Model):
       LANG = 'uk'

我已经找到了如何在父status中创建date_updated__table_args__这样的(部分)索引的方法。

但我需要创建 desc 排序索引rating,例如func.desc(rating),但我不知道该怎么做。

所有变体都不适合我(变体和它的错误):

  • Index('ix__%s__rating' % cls.__tablename__, 'rating desc')

    KeyError:'rating desc'

  • Index('ix__%s__rating' % cls.__tablename__, cls.rating.desc())

    sqlalchemy.exc.ArgumentError:无法将未命名的列添加到列集合

  • Index('ix__%s__rating' % cls.__tablename__, desc('rating'))

    在db

    中创建模式时

    sqlalchemy.exc.ProgrammingError :( ProgrammingError)语法错误在“)”附近 第1行:创建索引ix__table_en__rating on table_en()

当然我可以使用直接SQL手动创建该索引,但我确信解决方案存在于某处。

提前致谢!

1 个答案:

答案 0 :(得分:1)

这里是做你想做的简化代码:

class BaseModel(Base):

    __abstract__ = True

    id = Column(Integer, primary_key=True)
    rating = Column(Integer, nullable=False, default=0)

    @declared_attr
    def __tablename__(cls):
        return "table_%s" % cls.LANG

    @classmethod
    def build_indexes(cls):
        Index('ix__%s__rating' % cls.__tablename__, cls.__table__.c.rating.desc())


@event.listens_for(BaseModel, 'instrument_class', propagate=True)
def receive_mapper_configured(mapper, class_):
    class_.build_indexes()


class TableEn(BaseModel):
   LANG = 'en'


class TableUk(BaseModel):
   LANG = 'uk'

完整代码here

这里有2个问题。首先,您应该针对表列调用desc,而不是映射器对象属性。但是,如果你这样做,你就会遇到问题,因为__table__属性是在从__table__args__获取信息后创建的。因此,您需要在创建__table__字段后创建索引。一种方法是通过sqlalchemy活动。