LIL-271. Add balance sum func & show in user menu for author & admin

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

project/templates/lilcity/index.html

@@ -136,9 +136,9 @@
           <div class="header__ava ava"><img class="ava__pic" src="{% static 'img/user.jpg' %}"></div>
           {% endif %}
           <div class="header__drop">
-            {% comment %} <a class="header__link header__link_border" href="#">234.120.345 руб.</a> {% endcomment %}
-            {% if request.user.auth_token %}
             {% if request.user.role == 1 or request.user.role == 2 %}
+            <a class="header__link header__link_border" href="#">{{ request.user.balance }} руб.</a>
+            {% if request.user.auth_token %}
             <a class="header__link header__link_green" href="{% url 'course_create' %}">
             {% comment %} <a class="header__link header__link_gray disabled" href="#"> {% endcomment %}
               <div class="header__title">ДОБАВИТЬ КУРС</div>