在Django中按指定分类ID优先级排序产品QuerySet

嘿，我来帮你搞定这个按指定分类优先级排序产品的需求！根据你给出的模型结构和排序要求，我们可以利用Django ORM里的Case和When来实现数据库层面的高效排序，不用在Python里手动处理数据，还能保持QuerySet的惰性特性。

先回顾下你的模型（方便对照）：

class Category(models.Model):
    name = models.CharField(max_length=100)

class Product(models.Model):
    # 你的其他字段...
    category = models.ForeignKey(Category, related_name="products", on_delete=models.CASCADE)

假设你已经有了指定优先级的分类ID列表：

categories_ids = [3,5,1,4,2]

解决方案：用Case/When构建自定义排序规则 #

我们可以给每个分类ID分配一个对应的排序优先级数字，然后让QuerySet按照这个数字排序。具体代码如下：

1. 先导入需要的ORM工具：

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

2. 构建排序条件：

# 给每个分类ID对应一个排序位置，索引越小优先级越高
ordering = Case(
    *[When(category_id=id, then=pos) for pos, id in enumerate(categories_ids)],
    # 可选：设置不在列表中的分类产品的默认排序位置，这里让它们排在最后
    default=len(categories_ids),
    output_field=IntegerField()
)

3. 获取排序后的产品QuerySet：

sorted_products = Product.objects.all().order_by(ordering)

代码解释 #

• enumerate(categories_ids)会给列表里的每个分类ID分配一个索引值，比如3对应0，5对应1，以此类推，索引越小的分类，对应的产品会排在越前面。
• Case和When组合起来，会为每个产品匹配它所属分类ID对应的索引值，作为排序依据。
• default参数用来处理那些分类ID不在指定列表里的产品，这里设置为len(categories_ids)，意味着这些产品会排在所有指定分类的产品之后；如果想让它们排在最前面，可以把default设为-1。

另一种写法：用annotate显式添加排序字段 #

如果你想更直观地看到排序依据，也可以用annotate给每个产品添加一个sort_order字段，再按这个字段排序，效果是完全一样的：

sorted_products = Product.objects.annotate(
    sort_order=Case(
        *[When(category_id=id, then=pos) for pos, id in enumerate(categories_ids)],
        default=len(categories_ids),
        output_field=IntegerField()
    )
).order_by('sort_order')

这样处理后，sorted_products就会按照你想要的顺序返回：先显示所有分类ID为3的产品，接着是分类5的，然后是分类1的，以此类推。

备注：内容来源于stack exchange，提问作者Ali Joghataee