From 90adff64b33c301126c36e692805afed29034e73 Mon Sep 17 00:00:00 2001 From: Ross Date: Sun, 12 Sep 2021 10:04:34 +0100 Subject: [PATCH] . --- longs/models.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/longs/models.py b/longs/models.py index 3c45d652..123364b2 100644 --- a/longs/models.py +++ b/longs/models.py @@ -178,13 +178,11 @@ 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, exam_pk=None): + def get_unmarked_answer_count(self, exam_pk=None, marker=None): if exam_pk is None: - answers = self.cid_user_answers.all() - else: - answers = self.cid_user_answers.filter(exam__id=exam_pk) + return self.cid_user_answers.all().count() - return len([ans for ans in answers if not ans.is_marked()]) + return len(self.get_unmarked_answers(exam_pk=exam_pk, marker=marker)) def get_unmarked_answers(self, exam_pk=None, marker=None): if exam_pk is None: