Agrégation Django dans le modèle

Question

Bonjour, Je suis en train de coder un site qui utilise des dons en tant qu'éléments multiples sur un projet. Je calcule le montant total des dons, mais cela ne semble pas fonctionner. Ai-je besoin d'une table intermédiaire entre les deux modèles?

Je me demandais s'il y avait une meilleure façon de le faire avec 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 

Merci pour votre aide aimable!

Answer 1

je me débarrasser de la méthode progess et annoter simplement le queryset:

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

Vous pouvez même utiliser une coutume 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() 

Puis:

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

Ou vous pouvez même remplacer le jeu de requête par défaut afin que la progression soit toujours renvoyée:

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')) 

Puis:

>>> 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 

Je pense que cela vous donnera ce que vous voulez. Vous devrez jouer avec un peu, et laissez-moi savoir si vous avez des problèmes. Assurez-vous également que la version de Django prend en charge l'agrégation.

EDIT: Correction d'un problème où aucun n'était retourné.