From 5d615683869895276d25c964d555ee5de7e03064 Mon Sep 17 00:00:00 2001
From: paov <firstov_ov@spacecorp.ru>
Date: Wed, 19 Oct 2022 12:43:26 +0300
Subject: [PATCH] 21

---
 choose/templates/choose.html |  2 +-
 choose/templates/vote.html   | 11 ++++++++++-
 choose/views.py              | 13 ++++++++++---
 3 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/choose/templates/choose.html b/choose/templates/choose.html
index 4f0f5b7..bc19007 100644
--- a/choose/templates/choose.html
+++ b/choose/templates/choose.html
@@ -108,7 +108,7 @@
     function sum_price() {
         s = 0
         for (let tea in choice) {
-            s += choice[tea] / 50 * parseInt(a[tea].price)
+            s += choice[tea] / a[tea].ms[1] * parseInt(a[tea].price)
         }
         const butt = document.getElementById('confirm_button');
         const sum_h = document.getElementById('sum_h');
diff --git a/choose/templates/vote.html b/choose/templates/vote.html
index e11f92f..86aa778 100644
--- a/choose/templates/vote.html
+++ b/choose/templates/vote.html
@@ -16,6 +16,7 @@
         <th>Выбрали</th>
         <th>Проголосовали</th>
         <th>Всего голосов</th>
+        <th>Цена</th>
         <th></th>
     </tr>
     {% for choice in choices %}
@@ -44,6 +45,13 @@
             <td>
                 {{ choice.votes }}
             </td>
+            <td>
+                {{ choice.sum }}
+                <br>
+                С человека:
+                <br>
+                {{ choice.sum_per_usr }}
+            </td>
             <td style="text-align: center;">
                 {% if choice.class == 'choice' %}
                      <button class="btn" onclick="vote({{ choice.id }})"
@@ -201,7 +209,8 @@
                 }
             })
     }
-        function cancel_vote(id) {
+
+    function cancel_vote(id) {
         const request = new Request(
             'confirm-cancel-vote',
             {
diff --git a/choose/views.py b/choose/views.py
index 31b0e91..e2aea0e 100644
--- a/choose/views.py
+++ b/choose/views.py
@@ -2,6 +2,7 @@ import json
 from collections import OrderedDict
 
 from django.contrib.auth import authenticate, login, logout
+from django.contrib.auth.models import User
 from django.http import JsonResponse, HttpResponse
 from django.shortcuts import render, redirect
 from django.template import loader
@@ -53,26 +54,32 @@ def list_of_teas(request):
 def list_of_choices(request):
     arr = []
     voted = False
+    user_num = len(list(User.objects.all())) - 1
     my_choice = Choice.objects.filter(users__in=[request.user]).first()
     if my_choice is not None:
         voted = True
-        json.loads(my_choice.choice)
+        summ = sum([Tea.objects.get(id=tea_id).price*mass/Tea.objects.get(id=tea_id).m for tea_id, mass in json.loads(my_choice.choice).items()])
         arr.append({'voted': [user.username for user in my_choice.votes.all()],
                     'chosed': [user.username for user in my_choice.users.all()],
                     'votes': len(list(my_choice.votes.all())) + len(list(my_choice.users.all())),
                     'teas': sorted([Tea.objects.get(id=int(tea_id)).name + ':' + str(mass) for tea_id, mass in
                                     json.loads(my_choice.choice).items()]),
-                    'class': 'my_choice', 'id': my_choice.id})
+                    'class': 'my_choice', 'id': my_choice.id,
+                    'sum': summ,
+                    'sum_per_usr': str(summ / user_num)})
         choices = list(Choice.objects.all().exclude(id=my_choice.id))
     else:
         choices = list(Choice.objects.all())
     for choice in choices:
+        summ = sum([Tea.objects.get(id=tea_id).price*mass/Tea.objects.get(id=tea_id).m for tea_id, mass in json.loads(choice.choice).items()])
         arr.append({'voted': [user.username for user in choice.votes.all()],
                     'chosed': [user.username for user in choice.users.all()],
                     'votes': len(list(choice.votes.all())) + len(list(choice.users.all())),
                     'teas': sorted([Tea.objects.get(id=int(tea_id)).name + ':' + str(mass) for tea_id, mass in
                                     json.loads(choice.choice).items()]),
-                    'class': 'choice', 'id': choice.id})
+                    'class': 'choice', 'id': choice.id,
+                    'sum': summ,
+                    'sum_per_usr': str(summ / user_num)})
     return {'choices': arr, 'voted': voted}