From 2d84b3600027c4dc5ffc2811584334beb3902b6a Mon Sep 17 00:00:00 2001
From: grey-cat-1908 <61203964+grey-cat-1908@users.noreply.github.com>
Date: Sun, 11 Aug 2024 13:45:17 +0000
Subject: [PATCH] form selector question

---
 models/form.py | 60 +++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 59 insertions(+), 1 deletion(-)

diff --git a/models/form.py b/models/form.py
index 9f207d9..995903a 100644
--- a/models/form.py
+++ b/models/form.py
@@ -1,21 +1,79 @@
 from enum import Enum, auto
+from typing import TypeAlias
 
-from pydantic import Field
+from pydantic import Field, field_validator
 
 from models import BaseModel
 
 
 class QuestionType(Enum):
     text = auto()
+    choice = auto()
 
 
 class BaseQuestion(BaseModel):
     question_type: QuestionType
     label: str = Field(min_length=1)
     description: str | None = Field(None, min_length=1)
+    required: bool = True
+
+
+class Option(BaseModel):
+    label: str
 
 
 class TextQuestion(BaseQuestion):
     question_type = QuestionType.text
     min_length: int | None = None
     max_length: int | None = None
+
+    @field_validator('min_length')
+    @classmethod
+    def validate_min_length(cls, v, values):
+        if v is not None and v < 0:
+            raise ValueError('min_length must be greater than or equal to 0')
+        return v
+
+    @field_validator('max_length')
+    @classmethod
+    def validate_max_length(cls, v, values):
+        min_length = values.get('min_length')
+        if v is not None:
+            if v <= 0:
+                raise ValueError('max_length must be greater than 0')
+            if min_length is not None and v < min_length:
+                raise ValueError('max_length cannot be less than min_length')
+        return v
+
+
+class SelectorQuestion(BaseQuestion):
+    question_type = QuestionType.choice
+    min_values: int = 1
+    max_values: int | None = None
+    options: list[Option] = []
+
+    @field_validator('min_values')
+    @classmethod
+    def validate_min_values(cls, v, values):
+        options = values.get('options')
+        if v is not None and (v < 1 or v > len(options)):
+            raise ValueError('min_values must be greater than or equal to 1')
+        return v
+
+    @field_validator('max_values')
+    @classmethod
+    def validate_max_values(cls, v, values):
+        min_values = values.get('min_values')
+        options = values.get('options')
+        if v is not None and (v > len(options) or min_values > v):
+            raise ValueError('max_values cannot be less than min_length or greater than the number of options')
+        return v
+
+
+Question: TypeAlias = SelectorQuestion | TextQuestion
+
+
+class Form(BaseModel):
+    id: int
+    name: str
+    questions: list[Question] = []