From 194ee67048b80655f3da6be1f5018fdc278ffc58 Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Wed, 25 Sep 2019 15:46:43 +0530
Subject: Multiple Changes: - Add teams - Fix bug that causes todays booked
 workshops to disappear from queue - Add a graph to display viz for accepted
 workshops in a team - Remove unnecessary loops

---
 statistics_app/views.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'statistics_app/views.py')

diff --git a/statistics_app/views.py b/statistics_app/views.py
index 1507196..8c3a6f0 100644
--- a/statistics_app/views.py
+++ b/statistics_app/views.py
@@ -10,6 +10,8 @@ from workshop_app.models import (
     BookedWorkshop, ProposeWorkshopDate,
     Testimonial
 )
+from teams.models import Team
+
 from django.template.loader import get_template
 from django.template import RequestContext
 from datetime import datetime, date
@@ -582,3 +584,27 @@ def profile_stats(request):
     else:
         logout(request)
         return render(request, "workshop_app/logout.html")
+
+@login_required
+def team_stats(request, team_id):
+    user = request.user
+    team = Team.objects.get(team_id)
+    if user not in team.members.all():
+        if user.groups.filter(name='instructor').count() > 0:
+            return redirect('/manage/')
+        return redirect('/book/')
+
+    member_workshop_data = {}
+    for member in team.members.all():
+        workshop_count = ProposeWorkshopDate.objects.filter(proposed_workshop_instructor=member.id).count()
+        if member_workshop_data.get('data'):
+            member_workshop_data['data'].append(workshop_count)
+        else:
+            member_workshop_data['data'] = [workshop_count]
+        if member_workshop_data.get('labels'):
+            member_workshop_data['labels'].append(member.user.username)
+        else:
+            member_workshop_data['labels'] = [member.user.username]
+    return render(request, 'statistics_app/team_stats.html', {'workshop_data': member_workshop_data})
+
+
-- 
cgit