如何在ModelMultipleChoiceField中合并两个querysets作为选择?

3 投票
2 回答
3185 浏览
提问于 2025-04-18 04:58

我在两个不同的地方使用一段代码,目的是动态生成一些表单字段。在这两个地方,dynamic_fields 是一个字典,字典的键是对象,值是对象的列表(如果列表为空,值就设为 False):

class ExampleForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        dynamic_fields = kwargs.pop('dynamic_fields')
        super(ExampleForm, self).__init__(*args, **kwargs)

        for key in dynamic_fields:
            if dynamic_fields[key]:
                self.fields[key.description] = forms.ModelMultipleChoiceField(widget=forms.CheckboxSelectMultiple, queryset=dynamic_fields[key], required=False)


    class Meta:
        model = Foo
        fields = ()

在一个视图中,对于任何键,值是通过一次数据库查询返回的对象列表——这是一个正常的查询集。这个视图运行得很好。

在另一个视图中,我需要进行多次查询才能获取构建某个值所需的所有内容。我首先用空列表初始化字典,然后通过基本的列表推导式(dict[key] += queryset)将从这些多次查询中获得的查询集逐个添加到相应的列表中。这使得每个值变成了一个二维列表,然后我通过以下方式将其扁平化(并去除重复项):

for key in dict:
    dict[key] = list(set(dict[key]))

我尝试了几种不同的方法——直接将每个查询集的查询附加到值/列表中,保持为列表的列表,使用 append 而不是 +=——但每次都遇到同样的错误:'list' object has no attribute 'none'

查看错误追踪信息,错误出现在表单的清理方法中。这是 django.forms.models 中相关的代码部分:

def clean(self, value):
    if self.required and not value:
        raise ValidationError(self.error_messages['required'], code='required')
    elif not self.required and not value:
        return self.queryset.none() # the offending line

到目前为止我的思考过程是:在第一个视图中,我通过一次查询生成了作为每个键值的列表,但在第二个视图中,我将多个查询组合成一个列表。这个列表没有像单个查询集那样的 none 方法。

我该如何在不失去这个方法的情况下组合多个查询集呢?

我找到了一篇帖子,但使用那里的 itertools.chain 后,我仍然遇到同样的问题。通过这个方法,我唯一能做到的就是把错误信息改成 'chain''set' object has no attribute 'none'


补充说明:这是关于查询集生成的一些额外信息。我有以下模型(只显示相关字段):

class Profile(models.Model):
    user = models.OneToOneField(User)
    preferred_genres = models.ManyToManyField(Genre, blank=True)

class Genre(models.Model):
    description = models.CharField(max_length=200, unique=True)
    parent = models.ForeignKey("Genre", null=True, blank=True)

class Trope(models.Model):
    description = models.CharField(max_length=200, unique=True)
    genre_relation = models.ManyToManyField(Genre)

在(正常工作的)视图 #1 中,我用来生成字段的字典,键是某个类型(Genre),值是该类型的父类型的列表。换句话说,对于每个键,查询集是 Genre.objects.filter(parent=key, **kwargs)

在不工作的视图 #2 中,我们需要从用户的 preferred_genres 字段开始。对于每个 preferred_genre,我需要提取相关的 Tropes 并将它们组合成一个查询集。目前,我正在循环遍历 preferred_genres,做类似这样的事情:

for g in preferred_genres:
    tropeset = g.trope_set.all()

这让我得到了许多包含所需信息的单独查询集,但我找不到将多个 tropesets 合并成一个大查询集的方法(而不是没有 none 属性的列表)。另外,这样也会对我的数据库造成很多查询压力。我也在尝试理解如何使用 prefetch_related 来减少查询次数,但我想一步一步来。

如果我不能将这些查询集合并成一个,但可以通过单个查询完成这些查找,我非常乐意听取建议!我现在正在阅读关于使用 Q 对象进行复杂查询的文档。这很吸引人——我能想象这将如何实现我想要的,但前提是我能一次性调用所有查询。由于我必须逐个调用它们,我不确定如何使用 Q 对象将它们 |& 组合在一起。

2 个回答

0

我找到了解决这个问题的“办法”。如果我把查询写成这样,我就能一次性获取我需要的所有内容,而不需要事后再把查询结果合并在一起:

desired_value = Trope.objects.filter(genre_relation__in=preferred_genres).distinct()

不过,我还是不知道怎么把多个查询结果合并成一个,而不失去那种看起来对表单正常显示很重要的“查询结果特性”。不过,对于我具体的使用情况,重新调整查询的方式就解决了这个问题。

5

你可以通过使用 |& 这两个符号来组合查询集。

from functools import reduce
from operator import and_, or_

querysets = [q1, q2, q3, ...]  # List of querysets you want to combine.

# Objects that are present in *at least one* of the queries
combined_or_querysets = reduce(or_, querysets[1:], querysets[0])

# Objects that are present in *all* of the queries
combined_and_querysets = reduce(and_, querysets[1:], querysets[0])

从 Django 1.11 版本开始,你还可以使用 unionintersection 这两个方法。

撰写回答