如何在Django查询Product时获取关联ImageAlbum及对应Image数据

核心思路 #

默认的Django ORM查询对于外键/OneToOne关联只会存储关联的主键值，第一次实际访问关联对象时才会触发单独的数据库查询。要实现查询Product时同步带出关联的ImageAlbum、以及Album下的所有Image对象，只需要配合Django内置的 select_related 和 prefetch_related 两个查询优化API，在初始查询阶段就把多层关联数据一次性加载到内存，全程最多3次SQL查询就能拿到全部需要的内容，不会触发逐对象的N+1查询。

基础实现代码 #

直接在查询集链式调用即可：
Product.objects.select_related('images').prefetch_related('images__images').all()

参数说明 #

• select_related('images')：因为Product和ImageAlbum是正向OneToOne关联，该方法会在查询Product主表时直接通过SQL JOIN关联ImageAlbum表，查询返回的product.images直接就是完整的ImageAlbum实例，不需要二次查库获取主键对应的相册数据
• prefetch_related('images__images')：跨两层关联，从ImageAlbum反向取所有关联的Image对象（对应Image模型里的album外键反向related_name为images），该方法会在拿到所有关联的相册ID后，单独执行一次IN查询把对应相册的所有Image对象一次性取回到内存，挂到每个相册实例的关联管理器上，后续访问时无需重复查库

自定义查询封装（可选） #

如果不想每次写查询都手动加预加载参数，可以通过自定义模型Manager把逻辑封装，后续直接调用即可：

from django.db import models

class ProductQuerySet(models.QuerySet):
    def with_full_media(self):
        return self.select_related('images').prefetch_related('images__images')

class ProductManager(models.Manager):
    def get_queryset(self):
        return ProductQuerySet(self.model, using=self._db)
    
    def with_full_media(self):
        return self.get_queryset().with_full_media()

# 替换原Product模型的objects属性
class Product(models.Model):
    # 原有字段保持不变
    title = models.CharField(max_length=300)
    price = models.IntegerField()
    description = models.TextField(max_length=2000, help_text="This is the description of the product")
    images = models.OneToOneField(ImageAlbum, related_name='model', on_delete=models.CASCADE)
    
    # 替换默认Manager
    objects = ProductManager()

封装后直接调用Product.objects.with_full_media().all()即可拿到带全量关联数据的查询集。

灵活预取说明 #

如果不需要把相册下的所有Image都拉回来，只想预取特定筛选条件的Image（比如只取默认图、只取缩略图），可以用Prefetch对象自定义预取规则，减少不必要的数据传输：

from django.db.models import Prefetch

# 示例：只预取每个相册下标记为默认的Image
products = Product.objects.select_related('images').prefetch_related(
    Prefetch(
        'images__images',
        queryset=Image.objects.filter(default=True),
        to_attr='filtered_default_imgs'
    )
).all()

注意：这种自定义预取的结果会存到指定的to_attr属性里，不会覆盖默认的images关联管理器，访问时直接读取该属性即可获取筛选后的结果。

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