Django Rest Framework自定义排序：实现Enum字段colorCode按指定顺序排列

该需求完全可以实现，Django提供了两种常用的自定义排序方案，其中Python内存层面排序的功能和Java的Comparator逻辑一致。

方案1：数据库层面自定义排序（推荐，性能更高） #

适合数据量较大的场景，直接通过ORM在SQL查询阶段完成排序，不需要把全量数据加载到内存。
首先导入依赖的ORM函数：

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

之后构造自定义排序规则查询即可：

# 给每个颜色指定排序权重，权重越小排序越靠前
color_order = Case(
    When(colorCode='RED', then=Value(1)),
    When(colorCode='YELLOW', then=Value(2)),
    When(colorCode='GREEN', then=Value(3)),
    output_field=IntegerField()
)
# 按自定义权重排序
queryset = MyModel.objects.annotate(custom_order=color_order).order_by('custom_order')

如果需要将该规则设为模型的默认排序，可直接修改Model的Meta配置：

class Meta:
    managed = True
    db_table = 'MyModel'
    ordering = [
        Case(
            When(colorCode='RED', then=Value(1)),
            When(colorCode='YELLOW', then=Value(2)),
            When(colorCode='GREEN', then=Value(3)),
            output_field=IntegerField()
        )
    ]

方案2：Python内存层面排序（灵活度更高，等价于Java Comparator） #

适合数据量较小的场景，直接在Python层面对查询集进行自定义排序，写法更简洁：

# 定义排序映射规则，值越小排序越靠前
sort_key_map = {'RED': 0, 'YELLOW': 1, 'GREEN': 2}
# 对查询集自定义排序
queryset = MyModel.objects.all()
sorted_queryset = sorted(queryset, key=lambda item: sort_key_map[item.colorCode])

如果存在colorCode不在预设列表中的情况，可以在映射规则中添加默认值避免报错，比如sort_key_map.get(item.colorCode, 999)，不在规则内的内容会默认排在最后。

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