From f53daf45257757e8f33139e65a455f17d6e184b6 Mon Sep 17 00:00:00 2001 From: Ivlev Denis Date: Wed, 21 Feb 2018 15:51:39 +0300 Subject: [PATCH] LIL-267, LIL-271. Fix summary price --- apps/payment/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/apps/payment/models.py b/apps/payment/models.py index 9beb1947..75645e08 100644 --- a/apps/payment/models.py +++ b/apps/payment/models.py @@ -69,6 +69,9 @@ class CoursePayment(Payment): amount=self.amount, payment=self, ) + else: + author_balance.amount = self.amount + author_balance.save() class SchoolPayment(Payment): @@ -85,7 +88,7 @@ class SchoolPayment(Payment): weekday__in=self.weekdays, ).aggregate( models.Sum('month_price'), - ).month_price_sum + ).get('month_price__sum', 0) self.amount = month_price_sum super().save(*args, **kwargs) author_balance = getattr(self, 'authorbalance', None) @@ -94,3 +97,6 @@ class SchoolPayment(Payment): amount=month_price_sum, payment=self, ) + else: + author_balance.amount = self.amount + author_balance.save()