Django：如何通过外键关联获取商品图片URL？

背景与需求 #

我正在开发一个网站，需要将原本包含图片字段的Product类调整为拆分后的结构：把图片字段移到单独的ProductImage模型中，通过外键关联。现在需要修改原有的imageURL属性，实现通过外键关联获取图片URL，替代之前直接访问Product自身图片字段的逻辑。

现有代码结构 #

models.py #

class Product(models.Model):
    name    = models.CharField(max_length=60)
    price   = models.DecimalField(max_digits=10, decimal_places=2)
    
    # 数字商品无需配送
    digital = models.BooleanField(default=False, null=True, blank=False)
    # 图片字段已移至ProductImage
    # image   = models.ImageField(null=True, blank=True)

    @property
    def imageURL(self):
        try:
            url = self.image.url
        # 获取占位图（当图片加载失败时）
        except: 
            url = '/images/placeholder.png'
        return url
    

class ProductImage(models.Model):
    
    product     = models.ForeignKey(Product, related_name="images", on_delete=models.SET_DEFAULT, default=None, null=True)
    image1      = models.ImageField(null=True, blank=True)
    image2      = models.ImageField(null=True, blank=True)
    image3      = models.ImageField(null=True, blank=True)
    image4      = models.ImageField(null=True, blank=True)
    image5      = models.ImageField(null=True, blank=True)
    image6      = models.ImageField(null=True, blank=True)

view.html #

{% for product in products %}    
    <img class="thumbnail" src="{{ product.imageURL }}">
{% endfor %}

views.py #

def store_view(request):
    # 其他代码
    products = Product.objects.all()
    context.update({"products":products})
    return render(request, "view.html", context)

尝试的修改与错误 #

我尝试修改Product类的imageURL属性，直接通过外键关联获取图片：

class Product(models.Model):
    # 其他属性不变
    
    @property
    def imageURL(self):
        try:
            url = self.images.image1.url
        # 获取占位图（当图片加载失败时）
        except:
            url = '/images/placeholder.png'
        return url

但触发错误：AttributeError: 'RelatedManager' object has no attribute 'image1'

解决方案 #

1. 修正imageURL属性逻辑 #

错误原因是self.images是RelatedManager对象，代表当前商品关联的所有ProductImage实例集合，不能直接访问image1字段。需要先获取具体的ProductImage实例，再访问图片字段：

class Product(models.Model):
    # 其他属性不变
    
    @property
    def imageURL(self):
        placeholder = '/images/placeholder.png'
        try:
            # 获取当前商品关联的第一个ProductImage实例
            product_image = self.images.first()
            if not product_image:
                return placeholder
            # 优先取image1的URL，若为空则返回占位图
            if product_image.image1:
                return product_image.image1.url
            return placeholder
        except:
            return placeholder

2. 优化：自动获取第一个非空图片 #

如果希望自动从image1到image6中取第一个非空的图片，可以修改逻辑：

class Product(models.Model):
    # 其他属性不变
    
    @property
    def imageURL(self):
        placeholder = '/images/placeholder.png'
        try:
            product_image = self.images.first()
            if not product_image:
                return placeholder
            # 遍历所有图片字段，返回第一个非空的图片URL
            for field_name in ['image1', 'image2', 'image3', 'image4', 'image5', 'image6']:
                image = getattr(product_image, field_name)
                if image:
                    return image.url
            # 所有图片字段都为空时返回占位图
            return placeholder
        except:
            return placeholder

3. 优化数据库查询性能 #

为避免循环查询商品时产生N+1数据库请求，在视图中使用prefetch_related预取关联的ProductImage数据：

def store_view(request):
    # 其他代码
    # 预取关联的ProductImage数据，减少数据库查询次数
    products = Product.objects.prefetch_related('images').all()
    context.update({"products":products})
    return render(request, "view.html", context)

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