配置#
对于自动生成的搜索触发器、索引和搜索向量列,SQLAlchemy-Searchable 提供了许多定制选项。
全局配置选项#
可以通过将以下配置选项传递给 make_searchable()
函数,将它们全局定义
search_trigger_name
定义搜索数据库触发器的名称。默认命名约定为
"{table}_{column}_trigger"
。search_trigger_function_name
定义数据库搜索向量更新函数的名称。默认命名约定为
{table}_{column}_update
。regconfig
这是决定用于搜索的语言配置的 PostgreSQL 文本搜索配置。默认设置是
"pg_catalog.english"
。
以下是利用这些选项的示例
make_searchable(Base.metadata, options={"regconfig": "pg_catalog.finnish"})
更改搜索向量的目录#
在某些情况下,您可能希望将搜索向量的默认语言配置切换为另一种语言。您可以通过提供 regconfig
参数来为 TSVectorType
实现此目的。在以下示例中,我们使用芬兰语代替默认英语
class Article(Base):
__tablename__ = "article"
name = sa.Column(sa.Text(255))
search_vector = TSVectorType("name", regconfig="pg_catalog.finnish")
加权搜索结果#
为了进一步细化搜索结果,可以应用 PostgreSQL 的 术语加权系统(从 A 到 D)。此示例演示了如何优先考虑在文章标题中找到的术语,而不是在内容中找到的术语
class Article(Base):
__tablename__ = "article"
id = sa.Column(sa.Integer, primary_key=True)
title = sa.Column(sa.String(255))
content = sa.Column(sa.Text)
search_vector = sa.Column(
TSVectorType("title", "content", weights={"title": "A", "content": "B"})
)
请记住,在使用加权搜索词时,您需要使用 sort=True
选项进行搜索
query = search(sa.select(Article), "search text", sort=True)
每个类具有多个搜索向量#
在模型需要多个搜索向量的示例中,SQLAlchemy-Searchable 对你而言绰绰有余。以下是如何为 Article
类设置多个搜索向量的
class Article(Base):
__tablename__ = "article"
id = sa.Column(sa.Integer, primary_key=True)
name = sa.Column(sa.String(255))
content = sa.Column(sa.Text)
description = sa.Column(sa.Text)
simple_search_vector = sa.Column(TSVectorType("name"))
fat_search_vector = sa.Column(TSVectorType("name", "content", "description"))
然后可以在执行查询时选择使用哪个搜索向量
query = search(sa.select(Article), "first", vector=Article.fat_search_vector)
组合搜索向量#
有时你可能需要同时从多个表中进行搜索。这可以通过使用组合搜索向量实现。考虑以下模型定义,其中每篇文章具有一个类别
import sqlalchemy as sa
from sqlalchemy.orm import declarative_base
from sqlalchemy_utils.types import TSVectorType
Base = declarative_base()
class Category(Base):
__tablename__ = "category"
id = sa.Column(sa.Integer, primary_key=True)
name = sa.Column(sa.String(255))
search_vector = sa.Column(TSVectorType("name"))
class Article(Base):
__tablename__ = "article"
id = sa.Column(sa.Integer, primary_key=True)
name = sa.Column(sa.String(255))
content = sa.Column(sa.Text)
search_vector = sa.Column(TSVectorType("name", "content"))
category_id = sa.Column(sa.Integer, sa.ForeignKey(Category.id))
category = sa.orm.relationship(Category)
现在考虑一种情况,我们希望找到文章内容或名称或类别名称中包含单词“矩阵”的所有文章。这可以通过以下方式实现
combined_search_vector = Article.search_vector | Category.search_vector
query = search(
sa.select(Article).join(Category), "matrix", vector=combined_search_vector
)
在使用左连接时,此查询会变得稍微复杂一些。然后,你必须考虑到使用 coalesce
函数而 Category.search_vector
可能为 None
的情况
combined_search_vector = Article.search_vector | sa.func.coalesce(
Category.search_vector, ""
)