.

longs/models.py

@@ -175,12 +175,20 @@ class Long(models.Model):
     def get_image_url_array(self):
         return json.dumps(["https://www.penracourses.org.uk{}".format(i.url) for i in self.get_images()])
 
-    def get_unmarked_answer_count(self):
-        answers = self.cid_user_answers.all()
+    def get_unmarked_answer_count(self, exam_pk=None):
+        if exam_pk is None:
+            answers = self.cid_user_answers.all()
+        else:
+            answers = self.cid_user_answers.filter(exam__id=exam_pk)
+
         return len([ans for ans in answers if not ans.is_marked()])
 
-    def get_unmarked_answers(self):
-        answers = self.cid_user_answers.all()
+    def get_unmarked_answers(self, exam_pk=None):
+        if exam_pk is None:
+            answers = self.cid_user_answers.all()
+        else:
+            answers = self.cid_user_answers.filter(exam__id=exam_pk)
+        
         return [ans for ans in answers if not ans.is_marked()]
 
     def get_question_json(self, based=True):