.

rad/views.py

@@ -88,41 +88,56 @@ def cid_scores(request, pk, passcode):
     print(cid_user.passcode)
 
     # questions = exam.exam_questions.all()
-    physics_answers = PhysicsCidUserAnswer.objects.filter(cid=cid)
-    anatomy_answers = AnatomyCidUserAnswer.objects.filter(cid=cid)
-    rapids_answers = RapidsCidUserAnswer.objects.filter(cid=cid)
-    longs_answers = LongsCidUserAnswer.objects.filter(cid=cid)
-    sbas_answers = SbasCidUserAnswer.objects.filter(cid=cid)
-
-    # if (
-    #    not physics_answers
-    #    and not anatomy_answers
-    #    and not rapids_answers
-    #    and not longs_answers
-    #    and not sbas_answers
-    # ):
-    #    raise Http404("cid not found")
-
-    #kwargs = {"publish_results": True}
+    EXAM_ANSWER_MAP = {
+        "physics" : (PhysicsCidUserAnswer, PhysicsExam),
+        "anatomy" : (AnatomyCidUserAnswer, AnatomyExam),
+        "rapids" : (RapidsCidUserAnswer, RapidsExam),
+        "longs" : (LongsCidUserAnswer, LongsExam),
+        "sbas" : (SbasCidUserAnswer, SbasExam),
+    }
     kwargs = {"exam_mode": True}
 
-    #if request.user.groups.filter(name="view_all_results").exists():
-    #    kwargs = {}
+    exams = []
+    for exam_type in EXAM_ANSWER_MAP:
+        exam_answers = EXAM_ANSWER_MAP[exam_type].objects.filter(cid=cid)
+        exam_ids = exam_answers.values_list("exam").distinct()
+        exams.append((exam_type, PhysicsExam.objects.filter(id__in=exam_ids, **kwargs)))
 
-    physics_exam_ids = physics_answers.values_list("exam").distinct()
-    physics_exams = PhysicsExam.objects.filter(id__in=physics_exam_ids, **kwargs)
 
-    anatomy_exam_ids = anatomy_answers.values_list("exam").distinct()
-    anatomy_exams = AnatomyExam.objects.filter(id__in=anatomy_exam_ids, **kwargs)
+    #physics_answers = PhysicsCidUserAnswer.objects.filter(cid=cid)
+    #anatomy_answers = AnatomyCidUserAnswer.objects.filter(cid=cid)
+    #rapids_answers = RapidsCidUserAnswer.objects.filter(cid=cid)
+    #longs_answers = LongsCidUserAnswer.objects.filter(cid=cid)
+    #sbas_answers = SbasCidUserAnswer.objects.filter(cid=cid)
 
-    rapids_exam_ids = rapids_answers.values_list("exam").distinct()
-    rapid_exams = RapidsExam.objects.filter(id__in=rapids_exam_ids, **kwargs)
+    ## if (
+    ##    not physics_answers
+    ##    and not anatomy_answers
+    ##    and not rapids_answers
+    ##    and not longs_answers
+    ##    and not sbas_answers
+    ## ):
+    ##    raise Http404("cid not found")
 
-    longs_exam_ids = longs_answers.values_list("exam").distinct()
-    longs_exams = LongsExam.objects.filter(id__in=longs_exam_ids, **kwargs)
+    ##kwargs = {"publish_results": True}
 
-    sbas_exam_ids = sbas_answers.values_list("exam").distinct()
-    sba_exams = SbasExam.objects.filter(id__in=sbas_exam_ids, **kwargs)
+    ##if request.user.groups.filter(name="view_all_results").exists():
+    ##    kwargs = {}
+
+    #physics_exam_ids = physics_answers.values_list("exam").distinct()
+    #physics_exams = PhysicsExam.objects.filter(id__in=physics_exam_ids, **kwargs)
+
+    #anatomy_exam_ids = anatomy_answers.values_list("exam").distinct()
+    #anatomy_exams = AnatomyExam.objects.filter(id__in=anatomy_exam_ids, **kwargs)
+
+    #rapids_exam_ids = rapids_answers.values_list("exam").distinct()
+    #rapid_exams = RapidsExam.objects.filter(id__in=rapids_exam_ids, **kwargs)
+
+    #longs_exam_ids = longs_answers.values_list("exam").distinct()
+    #longs_exams = LongsExam.objects.filter(id__in=longs_exam_ids, **kwargs)
+
+    #sbas_exam_ids = sbas_answers.values_list("exam").distinct()
+    #sba_exams = SbasExam.objects.filter(id__in=sbas_exam_ids, **kwargs)
 
     available_exams = cid_user.get_cid_exams()
 
@@ -130,11 +145,12 @@ def cid_scores(request, pk, passcode):
         request,
         "cid_scores.html",
         {
-            "physics_exams": physics_exams,
-            "anatomy_exams": anatomy_exams,
-            "rapid_exams": rapid_exams,
-            "longs_exams": longs_exams,
-            "sba_exams": sba_exams,
+            #"physics_exams": physics_exams,
+            #"anatomy_exams": anatomy_exams,
+            #"rapid_exams": rapid_exams,
+            #"longs_exams": longs_exams,
+            #"sba_exams": sba_exams,
+            "all_exams": exams,
             "cid": cid,
             "passcode": passcode,
             "cid_user": cid_user,