Django ORM命令精确/突出匹配在顶部

时间:2017-09-25 05:33:29

标签: python django django-queryset django-orm

我需要根据Django ORM中的匹配长度来排序结果。

我在Suburb字段中有一个name表,其中包含位置详细信息。

我要求用给定的文字搜索表格,并按照完全匹配/最突出的匹配顺序进行搜索

例如:

1)如果搜索字符串是' America'那么结果应该是[美国,南美洲,北美......]    在这种情况下,我们找到了一个完整的匹配,它必须是第一个元素。

2)如果搜索是port,那么结果应该是[' Port Melbourne' ' Portsea',East Airport]

在这种情况下,我们发现端口在分隔符之前是完全匹配。

我知道我可以使用多个查询并加入它们,例如一个用于完全匹配,另一个用于部分匹配,然后将它们与部分匹配时的排除相连接像

search_list=  [x.name for x in Suburb.objects.filter(name=search)] 
# Then
search_list += [x.name for x in Suburb.objects.filter(name__iregex=r"[[:<:]]{0}".format(search)).exclude(name__in=search_list)]

我可以继续这样下去。但想知道我们是否有更好的方法。

有任何线索吗?

提前致谢

2 个答案:

答案 0 :(得分:2)

基于func

postgres的解决方案(应该在mysql中工作,但不能测试):

from django.db.models import Func

class Position(Func):
    function = 'POSITION'
    arg_joiner = ' IN '

    def __init__(self, expression, substring):
        super(Position, self).__init__(substring, expression)


Suburb.objects.filter(
    name__icontains=search).annotate(
    pos=Position('name', search)).order_by('pos')

编辑:根据Tim Graham的修正,在Django docs - Avoiding SQL injection中推荐。

答案 1 :(得分:1)

Django 2.0实现了一个函数

StrIndex(string, substring)

  

返回一个正整数,对应于string中第一次出现的substring的1索引位置,如果找不到substring,则返回0。

示例:

from django.db.models.functions import StrIndex

qs = (
    Suburb.objects
    .filter(name__contains='America')
    .annotate(search_index=StrIndex('name', Value('America')))
)