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

Question

在我用 Flask 制作的网站中，我试图通过使用 pg_trgm PostgreSQL 扩展来“升级”我的搜索栏以匹配单词，即使拼写错误也是如此。但是，我无法弄清楚如何将扩展程序的自定义函数与 SQLAlchemy 一起使用。

我是否必须使用原始 SQL 才能正确执行查询，或者有没有办法更干净地执行此操作？

Answer 1

假设扩展配置正确，这些函数应该像任何其他数据库函数一样可以访问。例如，下面是使用 SQLAlchemy 核心调用 similarity 函数的方式：

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

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

How to use PostgreSQL extensions in SQLAlchemy (specifically Flask-SQLAlchemy)?

python

postgresql

sqlalchemy

flask-sqlalchemy