Получить сумму на основе другого поля django

У меня есть эти модели:

status_choices = ['not received', 'received']

class Investment(Model):
    # Offering
    offering = models.ForeignKey(Offering, blank=False, null=False, on_delete=models.CASCADE)
    invested = models.DecimalField(max_digits=9, default=0, decimal_places=2, blank=True, null=True)
    status = models.CharField(max_length=200, choices=status_choices, blank=True, null=True, default='not received')
    
class Offering(Model):
    
    ...
    

Я хочу сделать свойство для отображения суммы всех инвестиций внутри модели Offering.

Я делал это следующим образом:

    @property
    def amount_invested(self):
        return Investment.objects.filter(offering_id=self.id).aggregate(Sum('invested'))['invested__sum'] or 0

Но я также хочу отобразить amount_invested, если status внутри Investment модели 'received'.

Могу ли я узнать, как я должен уметь это делать?

Вы можете отфильтровать investment_set и таким образом сохранить только Investment с помощью status='received':

@property
def amount_invested(self):
    return self.investment_set.filter(
        status='received'
    ).aggregate(Sum('invested'))['invested__sum'] or 0
Вернуться на верх