3

我的 django 应用程序中有 3 个模型

class Company(models.Model):

    name = models.CharField(_("Name"), max_length = 100, blank = True, null = True)
    address = models.CharField(_("Address"), max_length = 300, blank = True, null = True)

class WebSite(models.Model):

    WEBSITE_TYPE_CHOICES = (
        ('Google+', 'Google+'),
        ('Facebook', 'Facebook'),
        ('Orkut', 'Orkut'),
    )
    user = models.ForeignKey(User)
    website = models.URLField(blank = True, null = True)
    website_type = models.CharField(max_length = 15, choices = WEBSITE_TYPE_CHOICES, default = 'Google+')
    company = models.ForeignKey(Company, null=True, blank=True)

class Review(models.Model):

    rating = models.FloatField(blank = True, null = True) 
    review = models.TextField(blank = True, null = True)
    company = models.ForeignKey(Company, null=True, blank=True)
    website = models.ForeignKey(Website, null=True, blank=True)

我想显示我的数据,例如

A公司:

  1. 谷歌 - 5 条评论,平均 3.4
  2. Orkut - 9 条评论,平均 4.1
  3. Facebook - 12 条评论,平均 4.3
  4. 其他 - 2 条评论,平均 2

B公司:

  1. Google - 2 条评论,平均 3 条
  2. Facebook - 5 条评论,平均 4 条

为此我做了

    final_list = company_list = []
    websites = Website.objects.select_related().filter(user = user)

    for website in websites:
        company = website.company
        if company is not None and company not in company_list:
            company_list.append(brand)
            company_websites = websites.filter(company = company)
            info = []
            for company_website in company_websites:
                company_type = company_website.website_type
                reviews = Review.objects.filter(company = company, website = company_website)
                no_of_reviews = len(reviews)
                average = reviews.aggregate(Avg('rating'))
                info.append({"type": company_type, "no_of_reviews": no_of_reviews,
                             "average": average['rating__avg']})
            final_list.append({"company": company.name, "stats": info})

代码有2个问题

  1. 它浪费了大量的计算
  2. 它还以某种方式在 final_list 中附加了空列表(BUGGY)

仅供参考:我相信 .annotate() 可以使用,但我不明白如何基于列使用它,在我的情况下是 (company__name)

4

2 回答 2

1

这将为您提供数据。

Review.objects.values(company, website__website_type).annotate(review_count=Count('id'), average=Avg('rating')).order_by('company')

使用{% regroup %}在模板中输出。

于 2012-05-21T13:19:07.523 回答
0

我没有对此进行测试,它可能不如您收到的其他答案那么好,但以防万一它有帮助......

companies = Company.objects.all()
company_ratings = []

for company in companies:
    websites = company.website_set.filter(user=user)
    new_data = []

    for website in websites:
        reviews = Reviews.objects.filter(website=website, company=company)
        ratings = [x.rating for x in reviews]
        count = len(ratings)
        new_data.append({'website': website, 'count': count, 
                         'average': float(sum(ratings))/count})

    company_ratings.append({'company':company, 'ratings':new_data})
于 2012-05-23T14:52:04.910 回答