From 085dc8c536a5968342d9632b3adb57ef4cd1d5bc Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Tue, 29 Dec 2020 18:54:25 +0200
Subject: Add validator for Discord role field for form

---
 backend/models/form.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'backend/models')

diff --git a/backend/models/form.py b/backend/models/form.py
index 64b198b..5aacc8b 100644
--- a/backend/models/form.py
+++ b/backend/models/form.py
@@ -1,7 +1,7 @@
 import typing as t
 
 import httpx
-from pydantic import BaseModel, Field, validator
+from pydantic import BaseModel, Field, root_validator, validator
 from pydantic.error_wrappers import ErrorWrapper, ValidationError
 
 from backend.constants import FormFeatures, WebHook
@@ -52,6 +52,16 @@ class Form(BaseModel):
 
         return value
 
+    @root_validator
+    def validate_role(cls, values: dict[str, t.Any]) -> t.Optional[dict[str, t.Any]]:
+        """Validates does Discord role provided when flag provided."""
+        if FormFeatures.ASSIGN_ROLE.value in values.get("features", []) and not values.get("discord_role"):  # noqa
+            raise ValueError(
+                "discord_role field is required when ASSIGN_ROLE flag provided."
+            )
+
+        return values
+
     def dict(self, admin: bool = True, **kwargs: t.Any) -> dict[str, t.Any]:
         """Wrapper for original function to exclude private data for public access."""
         data = super().dict(**kwargs)
-- 
cgit v1.2.3