Django values 方法 - 列表中的相关对象答案

【问题标题】：Django values method - related objects in listDjango values 方法 - 列表中的相关对象
【发布时间】：2021-10-15 21:27:12
【问题描述】：

您好，帮助查询集的 values() 方法。显示相关字段（外键）时，数据重复，这个数据可以分组吗？

class Product(models.Model):
    category = models.ForeignKey(Category, related_name='product', on_delete=models.CASCADE)
    title = models.CharField(max_length=250)
    slug = models.SlugField(max_length=250, unique=True, db_index=True)

class ProductImage(models.Model):
    product = models.ForeignKey(Product, related_name='product_image', on_delete=models.CASCADE)
    image = models.ImageField(upload_to='img/models/')
    is_main = models.BooleanField()

查看代码

data = Product.objects.all().values('pk', 'title', 'product_image')

此处为示例

这就是它的显示方式

[
    {'pk': 1, 'title': 'Product 1', 'product_image__image': 'img/models/mod_wpYzlnm.png'}, 
    {'pk': 2, 'title': 'Product 2', 'product_image__image': 'img/models/mod2_wEr0D2q.png'},
    {'pk': 2, 'title': 'Product 2', 'product_image__image': 'img/models/mod_pPQqmjB_we175uR.png'},
    {'pk': 10, 'title': 'Product 3', 'product_image__image': 'img/models/mod_3mTxkb9_z4lKV3l.png'}, 
    {'pk': 10, 'title': 'Product 3', 'product_image__image': 'img/models/heart.png'}
]

应该是这样的

[
    {'pk': 1, 'title': 'Product 1, 'product_image':[ 
        {'image':'img/models/mod_wpYzlnm.png'}
    ]}, 
    {'pk': 2, 'title': 'Product 2', 'product_image': [
        {'image':'img/models/mod2_wEr0D2q.png'}, 
        {'image':'img/models/mod_pPQqmjB_we175uR.png'}
    ]},
    {'pk': 10, 'title': 'Product 3',  'product_image': [
        {'image':'img/models/mod_3mTxkb9_z4lKV3l.png'}, 
        {'image':'img/models/heart.png'}
    ]}, 
]

【问题讨论】：

当你做什么查询...？
data = Product.objects.all().values('pk', 'title', 'product_image')
我觉得很奇怪，如果有多个项目，则使用列表，但如果有一个则使用字符串。这会让“读者”的逻辑更加复杂。
这能回答你的问题吗？ Output of values() on a QuerySet with ManyToMany fields
对不起，是的，必须有一个列表code {'pk': 1, 'title': 'Product 1, 'product_image':[ {'image':'img/models/mod_wpYzlnm.png'} ]},

标签： django

【解决方案1】：

您应该使用groupby(…) function [python-doc] 的itertools module [python-doc] 对数据进行后处理。然而，对于查询，我们应该首先对pk 进行排序，或者至少对查询集ProductImages 进行排序，使得相同Product 的所有ProductImages 彼此相邻。

因此，我们可以使用以下方式对其进行转换：

from itertools import groupby
from operator import itemgetter

data = Product.objects.values('pk', 'title', 'product_image').order_by('pk')

result = [
    {'pk': pk, 'title': title,
    'product_image': [ {'image': pi['product_image__image'] } for pi in pis ] }
    for (pk, title), pis in groupby(data, itemgetter('pk', 'title'))
]

对于给定的样本数据，这给了我们：

[{'pk': 1,
  'product_image': [{'image': 'img/models/mod_wpYzlnm.png'}],
  'title': 'Product 1'},
 {'pk': 2,
  'product_image': [{'image': 'img/models/mod2_wEr0D2q.png'},
                    {'image': 'img/models/mod_pPQqmjB_we175uR.png'}],
  'title': 'Product 2'},
 {'pk': 10,
  'product_image': [{'image': 'img/models/mod_3mTxkb9_z4lKV3l.png'},
                    {'image': 'img/models/heart.png'}],
  'title': 'Product 3'}]

【讨论】：