按列表中的值对查询集进行排序

时间:2011-05-11 15:06:13

标签: python django

是否可以按查询中提供的元素列表对django查询集进行排序? 例如,如果我这样做

m.objects.filter(id__in=[3,1,8])

我不认为查询集的顺序是id 3的元素,id 1的元素和id 8的元素。

由于

8 个答案:

答案 0 :(得分:5)

没有。除了一些人为的机制之外,没有办法做你想做的事情:

qs = m.objects.filter(id__in=[3,1,8])
qs_sorted = list()
for id in [3,1,8]:
    qs_sorted.append(qs.get(id=id))

你也需要在那里抛出一些异常处理,以防其中一个指定的id没有被实际返回。

否定的是,这将有效地否定查询集的延迟加载。事实上,它很可能会生成多个数据库查询(虽然没有测试过)。此外,您最终得到的是普通的python列表而不是查询集,因此不再需要进行过滤。

答案 1 :(得分:3)

请参阅Ordering by the order of values in a SQL IN() clause - 您需要下拉到raw queryset并且它是特定于数据库的。

如果您正在使用MySQL,那么您应该能够

Model.objects.raw('select * from myapp_model where id in (3,1,8) order by field(id,3,1,8)')

答案 2 :(得分:3)

由于Django> = 1.11中存在CaseWhen,因此您可以以更符合orm的方式进行操作,同时保留查询集的所有优势:

from django.db import models

order = ['b', 'a', 'z', 'x', 'c']
whens = []
for sort_index, value in enumerate(order):
    whens.append(models.When(my_field=value, then=sort_index))

qs = MyModel.objects.annotate(_sort_index=models.Case(*whens, output_field=models.IntegerField()))
qs.order_by('_sort_index')

这将生成如下内容:

from django.db import models

order = ['b', 'a', 'z', 'x', 'c']
qs = MyModel.objects.annotate(
    _sort_index=models.Case(
        models.When(my_field='b', then=0),
        models.When(my_field='a', then=1),
        models.When(my_field='z', then=2),
        models.When(my_field='x', then=3),
        models.When(my_field='c', then=4),
        output_field=models.IntegerField()
    )
).order_by('_sort_index')

我建议仅将其与少量列表项一起使用,因为它会炸毁数据库查询。

https://docs.djangoproject.com/ko/1.11/ref/models/conditional-expressions/#case

答案 3 :(得分:2)

您的大多数排序选项都假设您正在排序的内容中有订单。如果您可以定义顺序,则可以使用order_by和模型上的字段,或sorted使用python。如果你想重新排序(看似没有逻辑),你仍然可以使用sorted

for id in [3,1,8]:
    sorted(qs, key=lambda x: x.id == id)

答案 4 :(得分:1)

也许你可以使用order-by作为zeekay提到的。

如果您的QuerSet可以转换为列表,那么您可以使用类似的东西

sorted(queryset, key=lambda x: x.id) 

答案 5 :(得分:1)

谢谢大家的回答。我想出了另一个解决方案。它不是最好的,但只需要对数据库进行一次查询。 我有一个列表,其中包含我想要的ID。然后我进行查询并用数据填写字典,最后我遍历l以有序的方式拾取对象。

l =[7,1,4]
p_dict = {}
p = Poi.objects.filter(id__in=l)
for i in p:
    p_dict[i.id] = i

使用这个我会使用大量内存吗?但我获得了对数据库的访问。

@ a-lee我会研究你的解决方案并尝试将其转换为postgresql。

答案 6 :(得分:0)

嗯,我不知道最近是否刚刚更改了这个问题(至少从8月开始就是这种情况),但您可以使用OP的语法进行查询。 ..

答案 7 :(得分:0)

_list = [3,1,2]
sorted(queryset, key=lambda x: _list.index(x.id))