Django 如何基于任意长度标签列表动态生成多组合Q查询对象

实现思路 #

每个标签对应两种查询逻辑：正向匹配、反向排除，n个标签的场景总共有2^n种组合。直接用itertools.product生成所有布尔值排列即可覆盖全部场景，无需手写递归：

• 布尔值True对应当前标签使用正向Q查询
• 布尔值False对应当前标签使用反向~Q查询
• 遍历组合的过程中统计True的数量，就是当前组合的正向条件计数

完整代码 #

from django.db.models import Q
import itertools

def generate_tag_q_combinations(tag_list):
    result_list = []
    positive_count_list = []
    # 生成所有正负逻辑组合
    for combination in itertools.product([True, False], repeat=len(tag_list)):
        current_q = Q()
        positive_count = 0
        for is_positive, tag_val in zip(combination, tag_list):
            if is_positive:
                current_q &= Q(tag=tag_val)
                positive_count += 1
            else:
                current_q &= ~Q(tag=tag_val)
        result_list.append(current_q)
        positive_count_list.append(positive_count)
    return result_list, positive_count_list

测试示例 #

标签长度为1的场景 #

tag_list = ["t1"]
result_list, q_count_list = generate_tag_q_combinations(tag_list)
print(len(result_list)) # 输出 2
print(q_count_list) # 输出 [1, 0]

标签长度为2的场景 #

tag_list = ["1", "2"]
result_list, q_count_list = generate_tag_q_combinations(tag_list)
print(len(result_list)) # 输出 4
print(q_count_list) # 输出 [2, 1, 1, 0]

扩展说明 #

如果你的标签是多对多关联字段，只需要把代码里的Q(tag=tag_val)调整为对应你的模型字段的查询语句即可，比如Q(tags__name=tag_val)，核心逻辑不需要修改。

内容的提问来源于stack exchange，提问作者David Foie Gras