Вычислить разность 2 агрегированного Сумма в одном запросе

Question

У меня есть модель выглядит следующим образом:

class Loan(models.Model): 
    lender = models.ForeignKey(User, related_name='lender') 
    borrower = models.ForeignKey(User, related_name='borrower') 
    money = models.IntegerField() 

То, что я хочу сделать, это рассчитать валовой общей для пользователя:

gross_total = Loans.object.filter(lender=user).annonate(Sum('money')) 
       - Loans.object.filter(borrower=user).annonate(Sum('money')) 

Мой текущий общий расчет использует два запроса. Мне нужен один запрос для этого.

Спасибо.

Answer 1

Это может быть сделано с conditional expressions

from django.db.models import Case, Value, When, IntegerField 

gross_total = Loans.object.aggregate(
    gross=Sum(
     Case(
      When(lender=user, then=F('money')), 
      default=0, output_field=IntegerField) 
     ) 
    ) - Sum(
     Case(
      When(borrower=user, then=F('money')), 
      default=0, output_field=IntegerField) 
     ) 
    ) 
) 

Answer 2

Во-первых, как вы используете два внешних ключей из одной и той же модели пользователя, вы должны дать им «related_name», как это:

class Loan(models.Model): 
    lender = models.ForeignKey(User, related_name='lender') 
    borrower = models.ForeignKey(User, related_name='borrower') 
    money = models.IntegerField() 

Теперь приходя на ваш вопрос, что вам нужно что-то дано в этом посте ->How to group by AND aggregate with Django

Таким образом, требуемый запрос на ваш вопрос, скорее всего, будет:

gross_total = Loans.objects.filter(lender=user).values('user').annotate(score = Sum('money')) - Loans.objects.filter(borrower=user).values('user').annotate(score = Sum('money'))