我需要用Python和SQLAlchemy创建一个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__中的Index创建索引.此外，如果需要多个字段，我还使用一个函数来创建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:
people = Person.query.filter(Person.__ts_vector__.match(expressions, postgresql_regconfig='english')).all()