如何访问与修改HayStack的SearchQuerySet？自定义排序遇赋值难题

你碰到的这个情况很常见——SearchQuerySet并不是普通的Python列表，它是HayStack提供的懒加载查询集合，设计上就不支持元素赋值操作，毕竟它背后还关联着搜索引擎的查询逻辑。不过要实现你想要的自定义排序（基于查询词和标题的Levenshtein距离），其实很容易解决，下面是具体步骤：

1. 先把SearchQuerySet转换成普通列表 #

因为SearchQuerySet是可迭代对象，我们可以直接用list()把它转换成普通的Python列表，这样就能自由操作元素或者进行自定义排序了：

results_list = list(results)

2. 实现Levenshtein距离计算 #

首先确保你安装了python-Levenshtein库（如果没装，执行pip install python-Levenshtein），然后导入距离计算函数。如果不想依赖第三方库，也可以自己实现一个简单的Levenshtein距离算法，不过用成熟库更高效。

3. 用自定义规则排序 #

使用Python内置的sorted()函数，通过key参数指定排序依据：我们为每个结果计算查询词q和它的title字段的Levenshtein距离，距离越小说明匹配度越高，所以按升序排列即可。

修改后的完整示例代码 #

from Levenshtein import distance as levenshtein_distance

def sqs_query(self, sqs):
    q = self.request.QUERY_PARAMS.get('q', '')
    results = sqs.filter(content=q)
    # 将懒加载的SearchQuerySet转为可操作的普通列表
    results_list = list(results)
    # 基于Levenshtein距离自定义排序
    sorted_results = sorted(
        results_list,
        key=lambda result: levenshtein_distance(q, result.title)
    )
    return sorted_results

def get_queryset(self):
    sqs = self.sqs_query(self.sqs_models(SearchQuerySet()))
    return sqs

注意事项 #

• 这种方式需要把所有查询结果加载到内存中，如果你的结果集非常大（比如上万条），可能会有性能问题。这种情况下可以考虑在HayStack的索引中添加自定义评分逻辑，或者在搜索引擎层面做排序优化，但中小规模的数据用这个方法完全没问题。
• 如果你需要调整排序顺序（比如距离大的排前面），可以在key的结果前加负号，或者给sorted()传reverse=True参数。

内容的提问来源于stack exchange，提问作者Vinicius Morais