在 Django 查询集顶部返回完全匹配

Question

我有一个名为“用户”的 django 模型，它存储了一些关于人的基本信息，即名字和姓氏。我目前在我的 django 模型中进行了简单搜索，如果用户输入 first 名称，django 查询集将返回前 10 个匹配项，按 last 名称排序。

例如，目前，如果您搜索“Sam”，您可能会得到以下结果：

1. 山姆·雅培
2. 塞缪尔·贝克
3. 萨米·罗杰斯
4. 山姆·西蒙斯

代码很简单：

User.objects.filter(Q(first__istartswith=token)).order_by('last')

但是，我想更改此设置，以便首先返回任何 精确 名字匹配，然后是其余结果。因此，如果有人输入“Sam”，结果应该是：

1. 山姆·雅培
2. 山姆·西蒙斯
3. 塞缪尔·贝克
4. 萨米·罗杰斯

（确切的名字首先匹配，按姓氏排序，然后是其余匹配，按姓氏排序）。

我想把它变成 2 个查询集，然后只是组合列表，但我想知道是否可以在 1 个查询中做到这一点，最好坚持使用基本的 django 查询集 API（而不是编写一次性查询）。有谁知道这样做的方法吗？

提前致谢。

Answer 1

我认为仅使用一个查询（至少使用 Django ORM）是不可能做到这一点的。

所以您的 2 个查询应该如下所示：

limit = 10
q1 = User.objects.filter(first__iexact=token).order_by('last')[:limit]
limit -= len(q1)
if limit:
    q2 = User.objects.exclude(pk__in=q1).filter(first__istartswith=token).order_by('last')[:limit]
else:
    q2 = []
users = list(q1) + list(q2)

另一种方法是在 python 中过滤查询，但您必须获取所有结果，而不仅仅是最后 10 个：

query = User.objects.filter(first__istartswith=token).order_by('last')
exacts = [user for user in query if user.first == token]
others = [user for user in query if user.first != token]
users = exacts + others

Answer 2

# Get exact matches first
qs1 = User.objects.filter(first__iexact=token).order_by('last')

# get secondary results second
qs2 = User.objects.filter(first__istartswith=token).exclude(qs1).order_by('last')

result = itertools.chain(qs1, qs2)

还可以看看这个问题，它比您可能需要的更深入：How to combine 2 or more querysets in a Django view?

Answer 3

这并不漂亮，我个人建议使用搜索引擎，例如django-haystack，但是，如果您知道您使用的是什么数据库，您可以使用QuerySet.extra 添加一个字段来对记录进行排序：

extra = {'is_exact': "%s.first LIKE '%s'" % (User._meta.db_table, token)}
User.objects.filter(Q(first__istartswith=token)).extra(select=extra).order_by('-is_exact', 'last')

Answer 4

我认为您实际上可能需要全文搜索来完成此操作。查看djang-sphinx。

如果您的示例与您的完整用例一样复杂，您可以只处理用户代码中的排序和排序。

Answer 5

您可以按多个属性排序。

User.objects.filter(Q(first__istartswith=token)).order_by('first', 'last')

因此，您首先按first 排序，以便您的对象根据完全匹配和后续匹配获得过滤器。然后你在last 上再次订购，按照姓氏排序。