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

【问题标题】：How to use PostgreSQL extensions in SQLAlchemy (specifically Flask-SQLAlchemy)?如何在 SQLAlchemy（特别是 Flask-SQLAlchemy）中使用 PostgreSQL 扩展？
【发布时间】：2022-01-01 04:13:40
【问题描述】：

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

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

【问题讨论】：

你试过psycopg吗？
那将是我可以依赖的原始 SQL 方法。

标签： python postgresql sqlalchemy flask-sqlalchemy

【解决方案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()

【讨论】：

啊，太漂亮了，非常感谢