如何在Mongoengine中动态构建Q组合查询？解决+=操作报错问题

我正在通过Mongoengine对接数据库，根据不同输入提取帖子数据。最初采用循环方式每次执行单次查询，查阅资料后发现使用|操作符构建单次查询效率更高。静态查询场景下该方法可正常工作，但因筛选条件数量不固定，需实现动态构建查询的方案。

当前循环查询代码：

console_list = []
for console in console_list:
    query = Q(**{f'{console}__company': company_name})
    console_list.extend(console_document.objects(query))

静态查询成功的代码：

query = Q(**{f'Dreamcast__company': company_name}) | Q(**{f'Playstation__company': company_name})
console_list.extend(console_document.objects(query))

尝试动态构建查询时出现错误：

query = Q(**{f'{console}__company': company_name})
for idx, console in enumerate(console_list):
    if idx==0:
        continue
    query += Q(**{f'{console}__company': company_name})

报错信息：

TypeError: unsupported operand type(s) for +=: 'QCombination' and 'Q'

请问如何动态生成这些QCombination？

Mongoengine的Q对象不支持+=操作符，需要使用|操作符来组合多个Q条件，以下是两种可行的实现方式：

方式一：手动循环组合 #

如果console_list不为空，先基于第一个元素创建初始Q对象，再遍历剩余元素用|叠加条件：

if not console_list:
    results = []
else:
    # 初始化第一个查询条件
    query = Q(**{f'{console_list[0]}__company': company_name})
    # 遍历剩余平台，组合OR条件
    for console in console_list[1:]:
        query |= Q(**{f'{console}__company': company_name})
    # 执行查询并扩展结果
    results = console_document.objects(query)
    console_list.extend(results)

方式二：用functools.reduce简化代码 #

借助functools.reduce和operator.or_可以一行完成条件组合，代码更简洁：

from functools import reduce
import operator

if console_list:
    # 生成所有单个Q条件的列表，再用reduce组合成OR查询
    query = reduce(operator.or_, [Q(**{f'{console}__company': company_name}) for console in console_list])
    results = console_document.objects(query)
    console_list.extend(results)
else:
    results = []

原理说明 #

• operator.or_等价于|操作符，reduce会依次将列表中的每个Q对象用该操作符组合，最终生成包含所有OR条件的QCombination对象。
• 两种方式都能实现动态构建多条件OR查询，避免了循环多次查询的效率问题。

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