在Django Prefetch的queryset中切片报错，如何只预取前5条？

你碰到的这个错误，是因为给Prefetch传的queryset已经被切片过了——切片后的queryset没法再让Django添加后续的过滤条件（预取时需要自动关联主对象的外键来匹配数据）。要实现预取时只缓存每个主对象的前5条关联数据，有两种可行办法：

方案一：使用窗口函数（推荐，数据库支持时用） #

通过Window和RowNumber函数给每条关联数据标注行号，再过滤出行号≤5的记录，这样既能保留queryset的可修改性，又能限制每个主对象的关联数据数量。

示例代码：

from django.db.models import Prefetch, Window, F
from django.db.models.functions import RowNumber

# 构造带行号过滤的关联查询集
prefetch_queryset = SomeModel.objects.filter(some_field=some_value).annotate(
    # 按主模型的外键字段分组，按id排序生成行号
    row_num=Window(
        expression=RowNumber(),
        partition_by=F('your_foreign_key_field'),  # 替换为关联主模型的外键字段（比如parent_id）
        order_by=F('id').asc()
    )
).filter(row_num__lte=5)

# 创建Prefetch对象并使用
prefetch_obj = Prefetch('related_field', queryset=prefetch_queryset)
some_queryset.prefetch_related(prefetch_obj)

注意：这个方案需要数据库支持窗口函数，比如PostgreSQL、MySQL 8.0+、SQLite 3.25+这些版本。

方案二：自定义Prefetch子类（兼容旧版数据库） #

如果你的数据库不支持窗口函数，可以自定义Prefetch子类，先预取所有符合条件的关联数据，再在内存里给每个主对象的关联列表切片：

from django.db.models import Prefetch

class LimitedPrefetch(Prefetch):
    def __init__(self, lookup, queryset, limit, to_attr=None):
        self.limit = limit
        super().__init__(lookup, queryset, to_attr=to_attr)

    def get_prefetch_queryset(self, instances, queryset=None):
        # 执行默认的预取逻辑
        prefetch_qs, rel_obj_attr, instance_attr, single = super().get_prefetch_queryset(instances, queryset)
        # 对每个实例的关联数据切片取前N条
        if not single:
            for instance in instances:
                related_objs = getattr(instance, instance_attr)
                setattr(instance, instance_attr, related_objs[:self.limit])
        return prefetch_qs, rel_obj_attr, instance_attr, single

# 使用自定义的LimitedPrefetch
prefetch_obj = LimitedPrefetch(
    'related_field',
    queryset=SomeModel.objects.filter(some_field=some_value).order_by('id'),
    limit=5
)
some_queryset.prefetch_related(prefetch_obj)

这个办法适合旧版数据库，但如果关联数据量很大，会占用较多内存，需要根据实际场景选择。

内容的提问来源于stack exchange，提问作者y.dai