SQLAlchemy声明:定义触发器和索引(Postgres 9)

时间:2012-01-19 16:35:37

标签: postgresql sqlalchemy turbogears

表的SQLAlchemy类中是否有一种方法可以为该表定义/创建触发器和索引?

例如,如果我有一个像...这样的基本表格。

class Customer(DeclarativeBase):
    __tablename__ = 'customers'
    customer_id = Column(Integer, primary_key=True,autoincrement=True)
    customer_code = Column(Unicode(15),unique=True)
    customer_name = Column(Unicode(100))
    search_vector = Column(tsvector) ## *Not sure how do this yet either in sqlalchemy*.

我现在想要创建一个触发器来更新“search_vector”

CREATE TRIGGER customers_search_vector_update BEFORE INSERT OR UPDATE
ON customers
FOR EACH ROW EXECUTE PROCEDURE
tsvector_update_trigger(search_vector,'pg_catalog.english',customer_code,customer_name);

然后我想将该字段添加为索引......

create index customers_search_vector_indx ON customers USING gin(search_vector);

现在我从我的应用程序执行任何类型的数据库重新生成后,我必须为tsvector列添加列,触发器定义,然后从psql执行索引语句。不是世界末日,而是容易忘记一步。我全都是关于自动化的,所以如果我能在应用程序设置期间实现这一切,那么奖金!

1 个答案:

答案 0 :(得分:37)

指标可以直接创建。对于具有index=True参数的单列,如下所示:

customer_code = Column(Unicode(15),unique=True,index=True)

但是如果您想要更多地控制名称和选项,请使用显式Index()构造:

Index('customers_search_vector_indx', Customer.__table__.c.search_vector, postgresql_using='gin')

触发器也可以创建,但那些仍然需要基于SQL并且挂钩到DDL事件。有关详细信息,请参阅Customizing DDL,但代码可能与此类似:

from sqlalchemy import event, DDL
trig_ddl = DDL("""
    CREATE TRIGGER customers_search_vector_update BEFORE INSERT OR UPDATE
    ON customers
    FOR EACH ROW EXECUTE PROCEDURE
    tsvector_update_trigger(search_vector,'pg_catalog.english',customer_code,customer_name);
""")
tbl = Customer.__table__
event.listen(tbl, 'after_create', trig_ddl.execute_if(dialect='postgresql'))

旁注:我不知道如何配置tsvector数据类型:值得单独提问。