Django按特定顺序从id的数组中获取QuerySet

时间:2011-02-06 23:12:03

标签: django arrays

快速为你准备一个:

我有一个id列表,我想用它来返回一个QuerySet(或者如果需要的话,还有数组),但是我想保持这个顺序。

由于

5 个答案:

答案 0 :(得分:126)

自Django 1.8以来,您可以这样做:

from django.db.models import Case, When

pk_list = [10, 2, 1]
preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(pk_list)])
queryset = MyModel.objects.filter(pk__in=pk_list).order_by(preserved)

答案 1 :(得分:56)

我认为您不能在数据库级别强制执行该特定订单,因此您需要在python中执行此操作。

id_list = [1, 5, 7]
objects = Foo.objects.filter(id__in=id_list)

objects = dict([(obj.id, obj) for obj in objects])
sorted_objects = [objects[id] for id in id_list]

这将构建一个以id为键的对象字典,因此在构建排序列表时可以轻松检索它们。

答案 2 :(得分:25)

如果你想使用in_bulk这样做,你实际上需要合并上面的两个答案:

id_list = [1, 5, 7]
objects = Foo.objects.in_bulk(id_list)
sorted_objects = [objects[id] for id in id_list]

否则结果将是字典而不是特定的有序列表。

答案 3 :(得分:21)

这是在数据库级别执行此操作的方法。复制粘贴:blog.mathieu-leplatre.info  :

<强>的MySQL

SELECT *
FROM theme
ORDER BY FIELD(`id`, 10, 2, 1);

与Django相同:

pk_list = [10, 2, 1]
ordering = 'FIELD(`id`, %s)' % ','.join(str(id) for id in pk_list)
queryset = Theme.objects.filter(pk__in=[pk_list]).extra(
           select={'ordering': ordering}, order_by=('ordering',))

PostgreSQL

SELECT *
FROM theme
ORDER BY
  CASE
    WHEN id=10 THEN 0
    WHEN id=2 THEN 1
    WHEN id=1 THEN 2
  END;

与Django相同:

pk_list = [10, 2, 1]
clauses = ' '.join(['WHEN id=%s THEN %s' % (pk, i) for i, pk in enumerate(pk_list)])
ordering = 'CASE %s END' % clauses
queryset = Theme.objects.filter(pk__in=pk_list).extra(
           select={'ordering': ordering}, order_by=('ordering',))

答案 4 :(得分:9)

OneToMany