Создайте индекс полнотекстового поиска с помощью SQLAlchemy на PostgreSQL

Мне нужно создать индекс полнотекстового поиска PostgreSQL на Python с помощью SQLAlchemy. Вот что мне нужно в SQL: 

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

Теперь, как я делаю вторую часть с SQLAlchemy при использовании ORM: 

class Person(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String)
1 2
python python-3.x postgresql sqlalchemy flask-sqlalchemy

1 ответ:

Вы можете создать индекс, используя Index в __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()
7