Django: __in查询不会在查询集中保持顺序

Question

我有一系列按特定顺序排列的ID

>>> album_ids = [24, 15, 25, 19, 11, 26, 27, 28]
>>> albums = Album.objects.filter( id__in=album_ids, published= True )
>>> [album.id for album in albums]
[25, 24, 27, 28, 26, 11, 15, 19]

我需要按照album_ids中id的顺序，在queryset中获取相应的专辑。请问有什么方法可以保持这个顺序吗？

- Ahsan

---

这个回答解决了你的问题吗？Django按特定顺序从ID数组获取QuerySet - adnanmuttaleb

Answer 1

假设ID列表不太大，您可以将QS转换为Python中的列表并进行排序：

album_list = list(albums)
album_list.sort(key=lambda album: album_ids.index(album.id))

- Daniel Roseman

---

我有一个大型数据库，ID 可能达到数百万。 - Ahsan

仅供参考，我认为ID可以达到数百万在大多数应用程序中并不相关，因为我们很少需要加载这么多的信息。相反，您可以使用django-pagination仅加载显示所需的数量，并提供更好的用户体验。 - r_31415

这是一个不错的解决方案，但如何保持QuerySet继续在其上工作呢？ - Jose Luis de la Rosa

我仍在努力理解这个问题，但如果你需要保留QuerySet，比如将其传递给Admin中的ModelForm，那么这篇文章似乎有答案：http://blog.mathieu-leplatre.info/django-create-a-queryset-from-a-list-preserving-order.html。在我的情况下，我需要根据content_objects对一组genericrelations进行排序。 - Hylidan

如果无法检索到对象（即该对象不再存在），则操作失败。 - madprops

Answer 2

你不能通过Django ORM来完成这个任务。 但是自己实现这个功能非常简单:

album_ids = [24, 15, 25, 19, 11, 26, 27, 28]
albums = Album.objects.filter(published=True).in_bulk(album_ids) # this gives us a dict by ID
sorted_albums = [albums[id] for id in albums_ids if id in albums]

- alTus

Answer 3

对于Django版本>=1.8，请使用以下代码：

from django.db.models import Case, When

field_list = [8, 3, 6, 4]
preserved = Case(*[When(field=field, then=position) for position, field in enumerate(field_list)])
queryset = MyModel.objects.filter(field__in=field_list).order_by(preserved)

以下是数据库级别的PostgreSQL查询表示：

这里是数据库级别的PostgreSQL查询表示：

SELECT *
FROM MyModel
ORDER BY
  CASE
    WHEN id=8 THEN 0
    WHEN id=3 THEN 1
    WHEN id=6 THEN 2
    WHEN id=4 THEN 3
  END;

- Rashid Mahmood

---

这是一个非常优美的解决方案，值得更多的关注。 - Hammad

只有这个东西在规模上对我有效。 - Saša Kalaba

Answer 4

你可以通过 Django 的 ORM 使用 extra QuerySet 修改器来完成。

>>> album_ids = [24, 15, 25, 19, 11, 26, 27, 28]
>>> albums = Album.objects.filter( id__in=album_ids, published= True
             ).extra(select={'manual': 'FIELD(id,%s)' % ','.join(map(str, album_ids))},
                     order_by=['manual'])

- Harish

---

请注意，如果此解决方案用于字符串数据，则不安全防止 SQL 注入。 - Barney Szabolcs

Answer 5

使用 @Soitje 的解决方案：https://dev59.com/kG445IYBdhLWcg3wUoxe#37648265

def filter__in_preserve(queryset: QuerySet, field: str, values: list) -> QuerySet:
    """
    .filter(field__in=values), preserves order.
    """
    # (There are not going to be missing cases, so default=len(values) is unnecessary)
    preserved = Case(*[When(**{field: val}, then=pos) for pos, val in enumerate(values)])
    return queryset.filter(**{f'{field}__in': values}).order_by(preserved)


album_ids = [24, 15, 25, 19, 11, 26, 27, 28]
albums =filter__in_preserve(album.objects, 'id', album_ids).all()

请注意，您需要确保album_ids是唯一的。

备注：

1.)该解决方案应安全地适用于任何其他字段，而不会冒任何SQL注入攻击的风险。

2.)Case(Django doc)生成类似于https://stackoverflow.com/a/33753187/1031191的SQL查询。

order by case id 
          when 24 then 0
          when 15 then 1
          ...
          else 8 
end

- Barney Szabolcs

Answer 6

如果您使用MySQL并希望通过使用字符串列来保留顺序。

words = ['I', 'am', 'a', 'human']
ordering = 'FIELD(`word`, %s)' % ','.join(str('%s') for word in words)
queryset = ModelObejectWord.objects.filter(word__in=tuple(words)).extra(
                            select={'ordering': ordering}, select_params=words, order_by=('ordering',))

- lechat