fix rejected copies

diff --git a/teleforma/exam/views.py b/teleforma/exam/views.py
index 0cf9cc0ff1eadacab66fead0d9789613bfe33c9e..de320f54b07d9ed0a216b31de373d62d0ebac0ca 100644 (file)
--- a/teleforma/exam/views.py
+++ b/teleforma/exam/views.py
@@ -131,14 +131,16 @@ class ScriptsRejectedView(ScriptsView):
     def get_queryset(self):
         user = self.request.user
         period = Period.objects.get(id=self.kwargs['period_id'])
-        if user.professor.all():
-            Q1 = Q(status=0)
-            scripts = Script.objects.filter(Q1)
-        else:
-            Q1 = Q(status=0, author=user)
-            Q2 = Q(status=0, corrector=user)
-            scripts = Script.objects.filter(Q1 | Q2)
-        return scripts
+        QT = Q(status=0, author=user)
+        QT = Q(status=0, corrector=user) | QT
+
+        professor = user.professor.all()
+        if professor:
+            professor = professor[0]
+            for course in professor.courses.all():
+                QT = Q(status=0, period=period, course=course) | QT
+
+        return Script.objects.filter(QT)
 
     def get_context_data(self, **kwargs):
         context = super(ScriptsRejectedView, self).get_context_data(**kwargs)