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

Question

表的 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 中的索引语句。不是世界末日，但很容易忘记一步。我非常关注自动化，所以如果我能让这一切在应用程序设置期间发生，那么奖金！

Answer 1

索引的创建非常简单。对于具有 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 事件。有关详细信息，请参阅自定义 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 数据类型：值得一个单独的问题。