如何在 SQLAlchemy（特別是 Flask-SQLAlchemy）中使用 PostgreSQL 擴展？

Question

在我使用 Flask 制作的網站中，我試圖通過使用 pg_trgm PostgreSQL 擴展來“升級”我的搜索欄，以匹配單詞，即使拼寫錯誤也是如此。 但是，我無法弄清楚如何將擴展的自定義功能與 SQLAlchemy 一起使用。

我是否必須使用原始 SQL 來正確執行查詢，或者有沒有辦法更干凈地做到這一點？

Answer 1

假設擴展配置正確，這些功能應該可以像任何其他數據庫 function 一樣訪問。 例如，以下是使用 SQLAlchemy 內核調用similarity function 的方式：

import sqlalchemy as sa

engine = sa.create_engine('postgresql+psycopg2:///test', echo=True, future=True)

metadata = sa.MetaData()

tbl = sa.Table('t70546926', metadata,
               sa.Column('c1', sa.String, primary_key=True),
               sa.Column('c2', sa.String))
tbl.drop(engine, checkfirst=True)
tbl.create(engine)

ins = sa.insert(tbl).values(c1='Alice', c2='Alison')

with engine.begin() as conn:
    conn.execute(ins)

query = sa.select(sa.func.similarity(tbl.c.c1, tbl.c.c2).label('similarity_result'))
with engine.connect() as conn:
    rows = conn.execute(query)
    for row in rows:
        print(row.similarity_result)

對於 Flask-SQLAlchemy，你可以這樣做：

result = dbsession.query(sa.func.similarity(val1, val2)).all()