From 305690d7860200a2550eb9eca6a2b6e76ba02506 Mon Sep 17 00:00:00 2001 From: Ross Date: Mon, 3 May 2021 12:44:51 +0100 Subject: [PATCH] . --- anatomy/models.py | 8 +++++--- rapids/models.py | 12 +++++++----- 2 files changed, 12 insertions(+), 8 deletions(-) diff --git a/anatomy/models.py b/anatomy/models.py index 9786b5cb..6d52f4e5 100644 --- a/anatomy/models.py +++ b/anatomy/models.py @@ -147,8 +147,8 @@ class AnatomyQuestion(models.Model): return reverse("anatomy:question_detail", kwargs={"pk": self.pk}) def GetPrimaryAnswer(self): - if len(self.answers.all()) > 0: - return self.answers.all().first().answer + if len(self.answers.filter(proposed=False)) > 0: + return self.answers.filter(proposed=False).first().answer else: return "None yet..." @@ -183,7 +183,7 @@ class AnatomyQuestion(models.Model): return set( [ i.answer_compare - for i in self.answers.all() + for i in self.answers.filter(proposed=False) if i.status != i.MarkOptions.UNMARKED ] ) @@ -222,6 +222,8 @@ class Answer(models.Model): max_length=1, choices=MarkOptions.choices, default=MarkOptions.UNMARKED ) + proposed = models.BooleanField(default=False) + def __str__(self): return self.answer diff --git a/rapids/models.py b/rapids/models.py index ccef00fa..cd252793 100644 --- a/rapids/models.py +++ b/rapids/models.py @@ -89,6 +89,8 @@ class Answer(models.Model): max_length=1, choices=MarkOptions.choices, default=MarkOptions.UNMARKED ) + proposed = models.BooleanField(default=False) + def __str__(self): return self.answer @@ -212,10 +214,10 @@ class Rapid(models.Model): return "{} / {} {}".format(exams, lat, n) def GetPrimaryAnswer(self): - if len(self.answers.all()) > 0: - return self.answers.all().first().answer - elif self.normal: + if self.normal: return "Normal" + elif len(self.answers.filter(proposed=False)) > 0: + return self.answers.filter(proposed=False).first().answer else: return "None yet..." @@ -254,7 +256,7 @@ class Rapid(models.Model): return set( [ i.answer_compare - for i in self.answers.all() + for i in self.answers.filter(proposed=False) if i.status != i.MarkOptions.UNMARKED ] ) @@ -275,7 +277,7 @@ class Rapid(models.Model): return set( [ str(i) - for i in self.answers.all() + for i in self.answers.filter(proposed=False) if i.status == i.MarkOptions.CORRECT ] )