From d68cf6e576cc18dc80722a927db8942dc6fa1a15 Mon Sep 17 00:00:00 2001
From: Gael Le Mignot <gael@pilotsystems.net>
Date: Tue, 27 Nov 2018 16:47:57 +0100
Subject: [PATCH] Fixed regression on special case for pending scripts

---
 teleforma/exam/views.py                  |  4 ++--
 teleforma/templatetags/teleforma_tags.py | 26 +++++++++++-------------
 2 files changed, 14 insertions(+), 16 deletions(-)

diff --git a/teleforma/exam/views.py b/teleforma/exam/views.py
index b0e61918..5d8304cb 100755
--- a/teleforma/exam/views.py
+++ b/teleforma/exam/views.py
@@ -180,8 +180,8 @@ class ScriptsPendingView(ScriptsView):
         
         if self.request.GET.get('corrector') is None:
             user = self.request.user
-            # Exclude status=3 but not author=user
-            qs = qs.filter(~Q(status=3) | Q(author=user))
+            # Exclude status=2 but not author=user
+            qs = qs.filter(~Q(status=2) | Q(author=user))
 
         return qs
 
diff --git a/teleforma/templatetags/teleforma_tags.py b/teleforma/templatetags/teleforma_tags.py
index 304feb45..1e57ba8a 100644
--- a/teleforma/templatetags/teleforma_tags.py
+++ b/teleforma/templatetags/teleforma_tags.py
@@ -216,27 +216,25 @@ def published(doc):
     if doc:
         return doc.filter(is_published=True)
 
-@register.simple_tag
-def untreated_scripts_count(user, period):
-    Q1 = Q(status=3, author=user, period=period)
-    Q2 = Q(status=3, corrector=user, period=period)
-    scripts = Script.objects.filter(Q1 | Q2)
+def scripts_count(user, period, statuses):
+    if not period:
+        return ''
+    Q1 = Q(author=user)
+    Q2 = Q(corrector=user)
+    scripts = Script.objects.filter(Q1 | Q2).filter(status__in = statuses,
+                                                    period = period)
     if scripts:
         return ' (' + str(len(scripts)) + ')'
     else:
         return ''
+    
+@register.simple_tag
+def untreated_scripts_count(user, period):
+    return scripts_count(user, period, (3,))
 
 @register.simple_tag
 def treated_scripts_count(user, period):
-    if not period:
-        return ''
-    Q1 = Q(status=4, author=user, period=period)
-    Q2 = Q(status=4, corrector=user, period=period)
-    scripts = Script.objects.filter(Q1 | Q2)
-    if scripts:
-        return ' (' + str(len(scripts)) + ')'
-    else:
-        return ''
+    return scripts_count(user, period, (4,))
 
 @register.simple_tag
 def get_training_profile(user):
-- 
2.39.5