模型中的 Django 聚合

Question

您好，我正在编写一个网站，该网站将捐赠用作项目中的多个项目。 我正在计算捐款总额，但似乎不起作用。我需要两个模型之间的中间表吗？

我想知道是否有更好的方法来使用 Django Aggregation 来做到这一点。

class Donation(models.Model):
    user = models.ForeignKey(User)
    amount= models.DecimalField(max_digits=12, decimal_places=2)

    def __unicode__(self):
        return self.user

class Project(models.Model):
    name = models.CharField(max_length=200)
    donations=models.ManyToManyField(Donation, null=True, blank=True)

    def __unicode__(self):
        return self.name

    def progress(self):
        donations = self.donations.all()
        total_donations = 0
        for item in donations:
            total_donations += item.amount
        return total_donations

感谢您的热心帮助！

Answer 1

我将摆脱 progess 方法并简单地注释查询集：

from django.db.models import Sum
Project.objects.annotate(progress=Sum('donations__amount'))

您甚至可以使用自定义 Manager：

from django.db.models import Sum

class ProjectManager(models.Manager):
    def all_with_progress(self):
        return self.get_query_set().annotate(progress=Sum('donations__amount'))

class Project(models.Model)
    ...
    objects = ProjectManager()

然后：

>>> projects = Project.objects.all_with_progress()
>>> projects[0].progress
700

或者您甚至可以覆盖默认查询集，因此进度是总是返回：

class ProjectManager(models.Manager):
    def get_query_set(self, *args, **kwargs):
        qs = super(ProjectManager, self).get_query_set(*args, **kwargs)
        return qs.annotate(progress=Sum('donations__amount'))

然后：

>>> projects = Project.objects.all()
>>> projects[0].progress
700

Answer 2

def progress(self):
    from django.db.models import Sum
    amount = self.donations.exclude(amount=None).aggregate(progress=Sum('amount'))['progress']
    if amount:
        return amount
    return 0

我想这会给你你想要的。您必须尝试一下，如果有任何问题请告诉我。另外，请确保您的 Django 版本支持聚合。

编辑：修复了没有返回任何内容的问题。