Django商品变体模型：如何获取唯一颜色查询集

问题描述 #

我正在用Django搭建电商网站，需求是商品可以有不同颜色和尺寸，同时要跟踪各尺寸的库存。我已经创建了两个ProductVariation实例：

• product='test'、size='small'、color='red'、quantity=5
• product='test'、size='large'、color='red'、quantity=10
  数据库里已经生成这两个对象。现在渲染商品详情页时，想查询得到仅包含红色的唯一颜色查询集。

我见过一段代码，但看不懂：

Color.objects.filter(productvariant__product=some_product).distinct()

我的模型代码如下：

import uuid
from django.db import models

class Product(models.Model):
    """
    Product model to allow creating product records and linking with category 
    """
    product_uuid = models.UUIDField(
        default=uuid.uuid4, editable=False, unique=True)
    product_name = models.CharField(max_length=255)
    sub_category = models.ForeignKey(
        'SubCategory', on_delete=models.CASCADE, related_name='products')
    product_description = models.TextField()
    is_featured = models.BooleanField(default=False)
    product_sku = models.CharField(max_length=255, null=True, blank=True)
    product_price = models.DecimalField(max_digits=6, decimal_places=2)

    def __str__(self):
        return self.product_name

    class Meta:
        verbose_name = 'Product'
        verbose_name_plural = 'Products'

class ProductVariation(models.Model):
    """
    ProductVariation model to allow creating product with different size and color
    """
    product_var_uuid = models.UUIDField(
        default=uuid.uuid4, editable=False, unique=True)
    product = models.ForeignKey(
        'Product', on_delete=models.CASCADE, related_name='product_variations')
    size = models.ForeignKey('Size', on_delete=models.CASCADE, related_name="product_sizes")
    color = models.ForeignKey('Color', on_delete=models.CASCADE, related_name="color")
    quantity = models.IntegerField()

    def __str__(self):
        return self.product.product_name

    class Meta:
        constraints = [
            models.UniqueConstraint(
                fields=['product', 'color', 'size'],
                name='unique_prod_color_size_combo'
            )
        ]
        verbose_name = 'Product Variation'
        verbose_name_plural = 'Product Variations'


class Size(models.Model):
    product_size_uuid = models.UUIDField(
        default=uuid.uuid4, editable=False, unique=True)
    product_size = models.CharField(max_length=20)

    def __str__(self):
        return self.product_size

    class Meta:
        verbose_name = 'Product Size'
        verbose_name_plural = 'Product Sizes'

class Color(models.Model):
    product_color_uuid = models.UUIDField(
        default=uuid.uuid4, editable=False, unique=True)
    product_color = models.CharField(max_length=20)

    def __str__(self):
        return self.product_color

    class Meta:
        verbose_name = 'Product Color'
        verbose_name_plural = 'Product Colors'

解决方案与代码解释 #

1. 拆解你看不懂的那段代码

那段代码核心是反向关联查询+去重，但和你的模型存在命名不匹配：

• 你的变体模型是ProductVariation，Django反向关联默认用模型名小写，所以代码里的productvariant__应该改成productvariation__
• some_product是详情页对应的具体Product实例

2. 适配你模型的正确查询代码

在商品详情页的视图中，假设你已经拿到当前商品对象current_product，执行以下代码就能得到该商品的唯一颜色查询集：

unique_colors = Color.objects.filter(productvariation__product=current_product).distinct()

代码逻辑拆解：

• Color.objects.filter(...)：从颜色表中筛选数据
• productvariation__product=current_product：通过ProductVariation中间表反向关联，找到所有属于当前商品的变体，再关联对应的颜色
• .distinct()：同一个颜色可能对应多个尺寸变体（比如你的红色对应small和large），去重后就能得到唯一的颜色列表，也就是你要的仅包含红色的查询集

3. 额外优化建议

你的ProductVariation中Color字段的related_name设为"color"语义模糊，建议改成"product_variations"或者"variations"，方便后续反向查询：

color = models.ForeignKey('Color', on_delete=models.CASCADE, related_name="product_variations")

另外，也可以从当前商品对象直接出发查询颜色：

# 获取颜色名称的列表
color_names = current_product.product_variations.values_list('color__product_color', flat=True).distinct()
# 获取Color对象的查询集
color_objects = Color.objects.filter(id__in=current_product.product_variations.values_list('color_id', flat=True)).distinct()

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