如何在ModelMultipleChoiceField中合并两个querysets作为选择?
我在两个不同的地方使用一段代码,目的是动态生成一些表单字段。在这两个地方,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 个回答
我找到了解决这个问题的“办法”。如果我把查询写成这样,我就能一次性获取我需要的所有内容,而不需要事后再把查询结果合并在一起:
desired_value = Trope.objects.filter(genre_relation__in=preferred_genres).distinct()
不过,我还是不知道怎么把多个查询结果合并成一个,而不失去那种看起来对表单正常显示很重要的“查询结果特性”。不过,对于我具体的使用情况,重新调整查询的方式就解决了这个问题。
你可以通过使用 |
和 &
这两个符号来组合查询集。
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 版本开始,你还可以使用 union
和 intersection
这两个方法。