如何按外部值对 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

我能想到解决这个问题的唯一方法是创建一个与主模型相关的新排名模型。在每个查询中，我将排名项插入到该模型中，然后能够通过关系执行 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')