Queryset过滤器:检索manytomany字段作为每个对象的列表

时间:2017-08-09 12:21:42

标签: python django django-queryset

我想检索模型的每个对象,只包含他们的id和他们在多对多字段中的id列表。

我的模特:

class Wordlist(models.Model):
  words = models.ManyToManyField(Word)


class Word(models.Model):
  word = models.CharField(max_length=256)

我有这段代码:

list(Wordlist.objects.all().annotate(w=F('words')).values_list('pk', 'w'))

它给了我这个:

[{'pk': 1, 'w': 7},
 {'pk': 1, 'w': 13},
 {'pk': 1, 'w': 17},
 {'pk': 2, 'w': 29},
 {'pk': 1, 'w': 42},
 {'pk': 3, 'w': 52},
 {'pk': 2, 'w': 65}
 ...
]

我想要的是:

[{'pk': 1, 'w': [7, 13, 17, 42,...]},
 {'pk': 2, 'w': [29, 65,...]},
 {'pk': 3, 'w': [52,...]},
 ...
]

一个简单的解决方案是根据他们的ID组合dicts,但我认为这不是一个好的做法,而且非常有效,因为我们可能会有数十万个dicts。

另外,我想知道是否可以通过单个请求执行相反的操作;在Wordlist的请求中检索每个单词的Word个字词列表。

1 个答案:

答案 0 :(得分:3)

如果您使用PostgreSQL,您可以使用在django.contrib.postgres包中实现的array_agg函数。您的查询将如下所示:

from django.contrib.postgres.aggregates.general import ArrayAgg

Wordlist.objects.annotate(arr=ArrayAgg('words')).values_list('id', 'arr')