Создать индекс полного текста с SQLAlchemy на PostgreSQL

Question

Мне нужно создать индекс полного текста 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) 

Answer 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' 
     ) 
    ) 

---

Update: Пример запроса с использованием индекса:

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