Django + Queryset + m2m + ForeignKey

时间:2016-10-13 00:33:56

标签: django django-queryset

我正在尝试获取一个包含唯一“来源”的列表,并且所有用户都为其添加了“条款”,例如

从:

user1 s1 [t3,t4,t5]
user1 s2 [t1,t2]
user2 s1 [t1,t2]
user2 s2 [t3,t4,t5]

为:

s1 t1,t2,t3,t4,t5
s2 t1,t2,t3,t4,t5

我的模特:

class User_Filter(models.Model):
    user = models.ForeignKey(User)
    source = models.ForeignKey(Source, null=True)
    terms = models.ManyToManyField(Term)

class Source(models.Model):
    title = models.CharField('name', max_length=100, blank=True)
    url = models.URLField('url', unique=True)

class Term(models.Model):
    title = models.CharField('title', max_length=100, unique=True)

我正在玩这个代码:

sources = User_Filter.objects.values('source').distinct()

此行使用源ID吐出dict:

{'source': 1}
{'source': 2}

从这一点来说,我被困住了。任何帮助都非常感谢。

丑陋的解决方案:

list = []
for s in User_Filter.objects.values('source').distinct():
    objs = User_Filter.objects.all().filter(source=s['source'])
    source = Source.objects.get(id=s['source']).url
    source_dict = {}
    terms = []
    for obj in objs:
        for term in obj.terms.all():
            if term.title not in terms: terms.append(term.title)
    source_dict['terms']=terms
    source_dict['url']=source
    to_scraper.append(source_dict)
print (list)

1 个答案:

答案 0 :(得分:2)

您可以使用以下方式获取数据:

data = []

for source in Source.objects.all():
    terms = []

    for user in source.user_filter_set.all():
        terms.extend(user.terms.all().values_list('title', flat=True))

    data.append({'url': source.url, 'terms': list(set(terms))})

您遍历每个Source并获取其相关的User_Filter个实例。从那里开始,每个terms获得user,并将其添加到当前source的条款中。

然后,您将data列表添加到正在处理的source的表示中。 list(set(terms))删除了terms中可能存在的重复项。

有关user_filter_set的说明,请参阅有关多对一关系的official doc