根据公共字段合并两个Django查询集

Question

我有两个查询集（实际上是字典列表），如：

q1 = M1.objects.filter(id=pk).values('p_id', 'q1_quantity')
# q1: <Queryset[{'p_id': 2, 'q1_quantity': 4}, {'p_id': 3, 'q1_quantity': 5}]>

q2 = M2.objects.filter(p_id__in=[q1[x]['p_id'] for x in range(len(q1))]).values('p_id', 'q2_quantity')
# q2: <Queryset[{'p_id': 2, 'q2_quantity': 2}, {'p_id': 2, 'q2_quantity': 5}, {'p_id': 3, 'q2_quantity': 1}, {'p_id': 3, 'q2_quantity': 7}]>

q1具有不同的键：值对，而q2具有重复的键。

1）我想通过公共p_id对q2的所有值求和，以使q2变为：

# q2: <Queryset[{'p_id': 2, 'q2_quantity': 7}, {'p_id': 3, 'q2_quantity': 8}]>

2）然后，基于公共p_id将q1和q2合并为q3，例如：

q3 = ?
# q3: <Queryset[{'p_id': 2, 'q1_quantity': 4, 'q2_quantity': 7}, {'p_id': 3, 'q1_quantity': 5, 'q2_quantity': 8}]>

我调查了union()。但是不知道如何对查询集（q2）求和，然后将其与q1合并。

有人可以帮我吗？

Answer 1

问题是您正在实现效率低下的模型，具有2个具有重复字段的单独模型将迫使您进行2个查询。您可能需要考虑将它们全部合并在一个模型中，或者M2模型扩展了M1。

models.py

class M(models.Model):
    p_id = #Your Field...
    q1_quantity = #Your Field...
    q2_quantity = #Your Field...

然后在您的views.py

q = M.objects.filter(id=pk).values('p_id', 'q1_quantity', 'q2_quantity')

潜在问题：在发布的代码中，带注释的部分显示了一个包含多个对象的查询集，并且pk作为主键应该是唯一的，因此应该返回唯一的对象queryset。

Answer 2

1）我想通过公共p_id对q2的所有值求和，以使q2变为：

# q2: <Queryset[{'p_id': 2, 'q2_quantity': 7}, {'p_id': 3, 'q2_quantity': 8}]>

使用过的itertools.combinations：

from itertools import combinations

compare = []
for a, b in combinations(q2, 2):
    if a['p_id'] == b ['p_id']:
        a['q2_quantity'] += b['q2_quantity']
        if len(compare) <= 0:
            compare.append(a)
        else:
            [compare[d]['q2_quantity'] for d in range(len(compare)) if a['p_id'] == compare[d]['p_id']]

    else:
        if len(compare) <= 0:
            compare.append(a)
            compare.append(b)
        else:
            if any([a['p_id'] == compare[d]['p_id'] for d in range(len(compare))]):
                pass
            else:
                compare.append(a)
            if any([b['p_id'] == compare[d]['p_id'] for d in range(len(compare))]):
                pass
            else:
                compare.append(b)

2）然后，基于公共p_id将q1和q2合并为q3，例如：

q3 = ?
# q3: <Queryset[{'p_id': 2, 'q1_quantity': 4, 'q2_quantity': 7}, {'p_id': 3, 'q1_quantity': 5, 'q2_quantity': 8}]>

按照此SO post：

from collections import defaultdict
from itertools import chain

collector = defaultdict(dict)

for collectible in chain(cp, compare):
    collector[collectible['p_id']].update(collectible.items())

products = list(collector.values())