diff --git a/apps/user/models.py b/apps/user/models.py index db34018e..6f18d9e8 100644 --- a/apps/user/models.py +++ b/apps/user/models.py @@ -61,6 +61,9 @@ class User(AbstractUser): user_data = dumps(user_data, ensure_ascii=False) return user_data + def balance(self): + return self.balances.aggregate(models.Sum('amount')).get('amount__sum', 0) + @receiver(post_save, sender=User) def create_auth_token(sender, instance=None, created=False, **kwargs): diff --git a/project/templates/lilcity/index.html b/project/templates/lilcity/index.html index d9adc061..1ae53983 100644 --- a/project/templates/lilcity/index.html +++ b/project/templates/lilcity/index.html @@ -136,9 +136,9 @@
