sqlalchemy 全文检索

SQLAlchemy提供了全文检索的功能，可以通过使用全文检索插件来实现。常见的全文检索插件有以下几种：

1. PostgreSQL的tsvector和tsquery

PostgreSQL提供了tsvector和tsquery两个数据类型来支持全文检索。tsvector可以将文本转化为一种特殊的向量格式，而tsquery则是一个查询语言，可以用来匹配tsvector向量。在SQLAlchemy中，使用全文检索需要安装psycopg2和postgresql-contrib两个包。

安装完这两个包后，需要在数据库中创建一个全文检索索引。下面是一个创建索引的示例代码：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql.expression import text

engine = create_engine('postgresql://user:password@localhost/dbname')

Base = declarative_base()
Session = sessionmaker(bind=engine)
session = Session()

class MyTable(Base):
    __tablename__ = 'mytable'
    id = Column(Integer, primary_key=True)
    text = Column(String)
    
    @classmethod
    def search(cls, query):
        ts_query = text("to_tsquery(:query)").bindparams(query=query)
        return session.query(cls).filter(cls.text.match(ts_query)).all()

Base.metadata.create_all(engine)

# 创建全文检索索引
session.execute(text("CREATE INDEX mytable_text_idx ON mytable USING gin(to_tsvector('english', text))"))
session.commit()

上面的代码中，search方法使用了text函数将查询语句转化为SQLAlchemy的文本表达式。然后使用match方法将查询语句与全文检索索引进行匹配。

2. SQLite的FTS5

SQLite的FTS5是一种全文检索插件，可以通过在表中创建虚拟表来实现全文检索功能。在SQLAlchemy中，需要使用sqlite3的全文检索扩展包来支持FTS5插件。

下面是一个使用FTS5插件实现全文检索的示例代码：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql.expression import text

engine = create_engine('sqlite:///example.db')

Base = declarative_base()
Session = sessionmaker(bind=engine)
session = Session()

class MyTable(Base):
    __tablename__ = 'mytable'
    id = Column(Integer, primary_key=True)
    text = Column(String)
    
    @classmethod
    def search(cls, query):
        fts_query = text("SELECT * FROM mytable WHERE text MATCH :query").bindparams(query=query)
        return session.execute(fts_query).fetchall()

Base.metadata.create_all(engine)

# 创建虚拟表
session.execute(text("CREATE VIRTUAL TABLE mytable_fts5 USING fts5(text)"))
session.commit()

# 将数据导入虚拟表
session.execute(text("INSERT INTO mytable_fts5(docid, text) SELECT id, text FROM mytable"))
session.commit()

上面的代码中，首先需要安装sqlite3的全文检索扩展包。然后使用CREATE VIRTUAL TABLE语句创建虚拟表，并将数据导入虚拟表中。最后使用MATCH语句进行全文检索。