adjusted notification models

app/helpers/notifications/email/models/account.py

@@ -1,11 +1,12 @@
-from typing import Optional, List, Dict
-
-from pydantic import BaseModel
-
+from pydantic import BaseModel, Field
 
 class Account(BaseModel):
-    sender: str
-    account_name: str
-    password: str
-    server_address: str
-    port: int
+    sender: str = Field(...)
+    account_name: str = Field(...)
+    password: str = Field(...)
+    server_address: str = Field(...)
+    port: int = Field(...)
+
+    model_config = {
+        'validate_assignment': True
+    }

app/helpers/notifications/email/models/email.py

@@ -1,14 +1,18 @@
-from typing import Optional, List, Dict
-
-from pydantic import BaseModel
-
+from pydantic import BaseModel, Field
 
 class Services(BaseModel):
-    neo4j: str
-    swate: str
-    rabbitmq: str
+    neo4j: str = Field(default="")
+    swate: str = Field(default="")
+    rabbitmq: str = Field(default="")
 
+    model_config = {
+        'validate_assignment': True
+    }
 
 class Health(BaseModel):
     services: Services
-    status: str
+    status: str = Field(default="")
+
+    model_config = {
+        'validate_assignment': True
+    }

app/helpers/notifications/email/models/images.py

@@ -1,9 +1,11 @@
-from typing import Optional, List, Dict
-
-from pydantic import BaseModel
-
+from typing import Optional
+from pydantic import BaseModel, Field
 
 class Images(BaseModel):
-    success: Optional[bool]
-    warning: Optional[bool]
-    webhook: Optional[bool]
+    success: Optional[bool] = Field(default=None)
+    warning: Optional[bool] = Field(default=None)
+    webhook: Optional[bool] = Field(default=None)
+
+    model_config = {
+        'validate_assignment': True
+    }