✨ Implemented in Editor

classquiz/db/models.py

@@ -85,24 +85,33 @@ class RangeQuizAnswer(BaseModel):
     max_correct: int
 
 
+class VotingQuizAnswer(BaseModel):
+    answer: str
+    image: str | None = None
+    color: str | None
+
+
 class QuizQuestionType(str, Enum):
     ABCD = "ABCD"
     RANGE = "RANGE"
+    VOTING = "VOTING"
 
 
 class QuizQuestion(BaseModel):
     question: str
     time: str  # in Secs
     type: None | QuizQuestionType = QuizQuestionType.ABCD
-    answers: list[ABCDQuizAnswer] | RangeQuizAnswer
+    answers: list[ABCDQuizAnswer] | RangeQuizAnswer | list[VotingQuizAnswer]
     image: str | None = None
 
     @validator("answers")
     def answers_not_none_if_abcd_type(cls, v, values):
-        if values["type"] == QuizQuestionType.ABCD and len(v) == 0:
+        if values["type"] == QuizQuestionType.ABCD and type(v[0]) != ABCDQuizAnswer:
             raise ValueError("Answers can't be none if type is ABCD")
         if values["type"] == QuizQuestionType.RANGE and type(v) != RangeQuizAnswer:
             raise ValueError("Answer must be from type RangeQuizAnswer if type is RANGE")
+        if values["type"] == QuizQuestionType.VOTING and type(v[0]) != VotingQuizAnswer:
+            raise ValueError("Answer must be from type RangeQuizAnswer if type is RANGE")
         return v