django - 如何按外部值对 Django QuerySet 进行排序？

Question

我有一个由 (id, rank) 对组成的字典。我想对 id 执行 Django 查询，以便生成的查询集按排名（降序）排序。

获取查询集很容易：

rankings = {...}
result = MyModel.objects.filter(id__in=rankings.keys())

似乎答案应该涉及某种注释，我可以将其用作 order_by 的一部分，但我不知道如何到达那里。

编辑：我没有提到我需要结果是一个 QuerySet，因为这是一个美味的 API 管道的一部分。

Answer 1

我对 Django > 1.10 和 PostgreSQL > 9.5 的解决方案

from django.db.models import Func, Value, IntegerField, CharField
from django.contrib.postgres.fields import ArrayField


class ArrayPosition(Func):
    function = 'array_position'

    def __init__(self, items, *expressions, **extra):
        if isinstance(items[0], int):
            base_field = IntegerField()
        else:
            base_field = CharField(max_length=max(len(i) for i in items))
        first_arg = Value(list(items), output_field=ArrayField(base_field))
        expressions = (first_arg, ) + expressions
        super().__init__(*expressions, **extra)


pk_list = [234,12,23]
queryset = SomeModel.objects.filter(pk__in=pk_list, ...)\
    .annotate(ordering=ArrayPosition(pk_list, F('pk'), output_field=IntegerField()))\
    .order_by('ordering')

Answer 2

像这样的东西？

rankings = { 1 : 2, 2: 1, ... } # i.e. { 'id' : 'ranking', ... }
objects = list(MyModel.objects.filter(id__in=rankings.keys()))
objects.sort(key=lambda obj: rankings[obj.id])

Answer 3

我能想出解决这个问题的唯一方法是创建一个与主模型相关的新排名模型。在每个查询中，我将排名项目插入该模型，然后能够通过关系执行 order_by。（使用注释将排名添加到学校记录中。）

class UserSchoolRanking(models.Model):
    user = models.ForeignKey(User)
    school = models.ForeignKey(School)
    rank = models.IntegerField()

bulk_user_school_rank = [UserSchoolRank(user=user, school_id=k, rank=v)
                                         for k, v in rankings.iteritems()]
UserSchoolRank.objects.bulk_create(bulk_user_school_rank)

schools = School.objects.filter(userschoolrank__user=user)\
                .annotate(rank=Min('userschoolrank__rank'))\
                .order_by('-userschoolrank__rank')