在PostgreSQL上使用SQLAlchemy创建全文搜索索引

发布时间：2020-12-13 16:06:28 所属栏目：百科来源：网络整理

导读：我需要使用SQLAlchemy在 Python中创建PostgreSQL全文搜索索引.这是我想要的SQL： CREATE TABLE person ( id INTEGER PRIMARY KEY,name TEXT );CREATE INDEX person_idx ON person USING GIN (to_tsvector('simple',name)); 现在,在使用ORM时,如何使用SQLAlch

我需要使用SQLAlchemy在 Python中创建PostgreSQL全文搜索索引.这是我想要的SQL：

CREATE TABLE person ( id INTEGER PRIMARY KEY,name TEXT );
CREATE INDEX person_idx ON person USING GIN (to_tsvector('simple',name));

现在,在使用ORM时,如何使用SQLAlchemy执行第二部分：

class Person(db.Model):
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String)

解决方法

您可以使用__table_args__中的索引创建索引.我还使用一个函数来创建ts_vector,以便在需要多个字段时使其更加整洁和可重用.如下所示：

from sqlalchemy.dialects import postgresql

def create_tsvector(*args):
    exp = args[0]
    for e in args[1:]:
        exp += ' ' + e
    return func.to_tsvector('english',exp)

class Person(db.Model):
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String)

    __ts_vector__ = create_tsvector(
        cast(func.coalesce(name,''),postgresql.TEXT)
    )

    __table_args__ = (
        Index(
            'idx_person_fts',__ts_vector__,postgresql_using='gin'
        )
    )

更新：
使用索引的示例查询：

query = Person.__ts_vector__.match(expressions,postgresql_regconfig='english')
people = query.all()

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!