blob: ac585f9517486b2170174154998ea0ad3532f172 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import typing as t
from pydantic import BaseModel, Field, validator
from backend.constants import QUESTION_TYPES, REQUIRED_QUESTION_TYPE_DATA
from backend.models.types import ObjectId
class Question(BaseModel):
"""Schema model for form question."""
id: ObjectId = Field(alias="_id")
name: str
type: str
data: t.Dict[str, t.Any]
@validator("type", pre=True)
def validate_question_type(self, value: str) -> t.Optional[str]:
"""Checks if question type in currently allowed types list."""
value = value.lower()
if value not in QUESTION_TYPES:
raise ValueError(
f"{value} is not valid question type. "
f"Allowed question types: {QUESTION_TYPES}."
)
return value
@validator("data")
def validate_question_data(
self,
value: t.Dict[str, t.Any]
) -> t.Optional[t.Dict[str, t.Any]]:
"""Check does required data exists for question type and remove other data."""
# When question type don't need data, don't add anything to keep DB clean.
if self.type not in REQUIRED_QUESTION_TYPE_DATA:
return {}
# Required keys (and values) will be stored to here
# to remove all unnecessary stuff
result = {}
for key, data_type in REQUIRED_QUESTION_TYPE_DATA[self.type].items():
if key not in value:
raise ValueError(f"Required question data key '{key}' not provided.")
if not isinstance(value[key], data_type):
raise ValueError(
f"Question data key '{key}' expects {data_type.__name__}, "
f"got {type(value[key]).__name__} instead."
)
result[key] = value[key]
return result
|
