如何在数据库层面对Django QuerySet实现自定义顺序排序

完全可以在数据库层面完成自定义优先级排序，无需将全量数据加载到Python内存做运算，以下是可直接落地的ORM方案，跨主流数据库通用。

通用方案：使用Case/When条件表达式（全数据库兼容，推荐） #

Django ORM原生支持Case/When条件表达式，会将自定义排序规则翻译成数据库兼容的CASE WHEN语句，排序逻辑完全在数据库侧执行，兼容PostgreSQL、MySQL、SQLite等所有Django支持的数据库。
实现代码如下：

from django.db.models import Case, IntegerField, When

# 按照HIGH -> MEDIUM -> LOW的优先级排序，同优先级按id升序排列
result = Todo.objects.annotate(
    sort_weight=Case(
        When(priority=Todo.Priority.HIGH, then=1),
        When(priority=Todo.Priority.MEDIUM, then=2),
        When(priority=Todo.Priority.LOW, then=3),
        output_field=IntegerField()
    )
).order_by('sort_weight', 'id')

以上逻辑和Python sorted实现的排序规则完全一致，针对测试数据的返回顺序为：

• todo2（优先级HIGH）
• todo5（优先级HIGH）
• todo1（优先级MEDIUM）
• todo4（优先级MEDIUM）
• todo3（优先级LOW）

PostgreSQL专属简化方案 #

如果业务用PostgreSQL作为数据库，可以用Django内置的PostgreSQL扩展函数ArrayPosition简化写法，性能和Case/When方案基本一致：

from django.contrib.postgres.functions import ArrayPosition
from django.db.models import IntegerField

# 数组元素顺序对应排序优先级，越靠前权重越高
priority_sort_order = [Todo.Priority.HIGH, Todo.Priority.MEDIUM, Todo.Priority.LOW]
result = Todo.objects.annotate(
    sort_weight=ArrayPosition(
        priority_sort_order,
        'priority',
        output_field=IntegerField()
    )
).order_by('sort_weight', 'id')

最佳实践：封装为QuerySet复用 #

如果该排序规则是业务通用逻辑，可以将其封装为模型自定义QuerySet方法，避免重复编码：

class TodoQuerySet(models.QuerySet):
    def order_by_priority(self):
        return self.annotate(
            sort_weight=Case(
                When(priority=Todo.Priority.HIGH, then=1),
                When(priority=Todo.Priority.MEDIUM, then=2),
                When(priority=Todo.Priority.LOW, then=3),
                output_field=IntegerField()
            )
        ).order_by('sort_weight', 'id')

class Todo(models.Model):
    class Priority(models.IntegerChoices):
        HIGH = 1, "High"
        LOW = 2, "Low"
        MEDIUM = 3, "Medium"

    title = models.CharField(max_length=255)
    priority = models.PositiveSmallIntegerField(choices=Priority.choices, db_index=True)

    # 替换默认objects管理器
    objects = TodoQuerySet.as_manager()

封装后支持链式调用，过滤、分页等操作均可正常下推到数据库执行：

# 直接获取按优先级排序的过滤结果
result = Todo.objects.filter(title__contains="待办").order_by_priority()

以上方案相比全量拉取数据到内存排序的实现，数据量较大时性能提升非常明显，符合数据库层运算优先的优化原则。

内容的提问来源于stack exchange，提问作者Johnny Metz