aboutsummaryrefslogtreecommitdiffstats
path: root/backend/models/form.py
blob: e49026829a77cfc2f2be81c78e6fd58076c7ad20 (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
import typing as t

from pydantic import BaseModel, Field, validator

from backend.constants import FormFeatures
from backend.models import Question


class Form(BaseModel):
    """Schema model for form."""

    id: str = Field(alias="_id")
    features: t.List[str]
    questions: t.List[Question]

    @validator("features")
    def validate_features(self, value: t.List[str]) -> t.Optional[t.List[str]]:
        """Validates is all features in allowed list."""
        # Uppercase everything to avoid mixed case in DB
        value = [v.upper() for v in value]
        if not all(v in FormFeatures.__members__.values() for v in value):
            raise ValueError("Form features list contains one or more invalid values.")

        if (
            FormFeatures.COLLECT_EMAIL in value and
            FormFeatures.REQUIRES_LOGIN not in value
        ):
            raise ValueError("COLLECT_EMAIL feature require REQUIRES_LOGIN feature.")

        return value