SQLAlchemy-Searchable 快速入门指南：实现SQLAlchemy全文搜索功能

2025-06-24 14:01:19作者：殷蕙予

项目概述

SQLAlchemy-Searchable 是一个为SQLAlchemy ORM提供全文搜索功能的扩展库。它基于PostgreSQL的全文搜索能力，为开发者提供了简单易用的接口，可以快速为SQLAlchemy模型添加全文搜索功能。

安装步骤

要使用SQLAlchemy-Searchable，首先需要安装它。该库支持Python 3.8及以上版本，包括CPython和PyPy实现。

使用pip安装非常简单：

pip install SQLAlchemy-Searchable

安装完成后，就可以在项目中使用这个强大的全文搜索扩展了。

基础配置

在开始使用前，需要进行一些基础配置。核心配置是通过make_searchable函数完成的，它需要传入SQLAlchemy的declarative_base元数据：

from sqlalchemy.orm import declarative_base
from sqlalchemy_searchable import make_searchable

Base = declarative_base()
make_searchable(Base.metadata)

这一步会为后续的模型添加必要的搜索功能支持。

模型定义

要为模型添加全文搜索功能，需要定义一个特殊的搜索向量列。这里我们以文章(Article)模型为例：

from sqlalchemy import Column, Integer, String, Text
from sqlalchemy_utils.types import TSVectorType

class Article(Base):
    __tablename__ = "article"

    id = Column(Integer, primary_key=True)
    name = Column(String(255))
    content = Column(Text)
    search_vector = Column(TSVectorType("name", "content"))

关键点说明：

TSVectorType是专门用于全文搜索的数据类型
构造函数参数指定了哪些列需要被索引（这里是name和content）
搜索向量列会自动维护这些指定列的搜索索引

数据库初始化

定义好模型后，需要创建数据库表并插入一些测试数据：

from sqlqlchemy import create_engine
from sqlalchemy.orm import configure_mappers, Session

# 创建数据库引擎
engine = create_engine("postgresql://localhost/sqlalchemy_searchable_test")

# 重要：必须先配置映射器
configure_mappers()

# 创建所有表
Base.metadata.create_all(engine)

# 插入测试数据
session = Session(engine)

article1 = Article(name="First article", content="This is the first article")
article2 = Article(name="Second article", content="This is the second article")

session.add(article1)
session.add(article2)
session.commit()

特别注意：configure_mappers()必须在创建表之前调用，这是确保搜索功能正常工作的关键步骤。

执行搜索查询

数据准备就绪后，就可以使用search函数执行全文搜索了：

from sqlalchemy import select
from sqlalchemy_searchable import search

# 搜索包含"first"的记录
query = search(select(Article), "first")
article = session.scalars(query).first()

print(article.name)  # 输出: First article

搜索功能非常灵活，支持复杂的搜索表达式，可以满足各种搜索需求。