Refactor user enabled_notifications parameter to provide support for …

…other notification types.
OpenMined · Feb 23, 2024 · f348928 · f348928
1 parent 99fcfa7
commit f348928
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/syft/src/syft/service/user/user.py b/packages/syft/src/syft/service/user/user.py
@@ -39,7 +39,7 @@
 from ..response import SyftError
 from ..response import SyftSuccess
 from .user_roles import ServiceRole
-
+from ..notifier.notifier_enums import NOTIFIERS
 
 @serializable()
 class UserV1(SyftObject):
@@ -97,7 +97,12 @@ def make_email(cls, v: EmailStr) -> EmailStr:
         return EmailStr(v)
 
     # fields
-    email_notifications_enabled: bool = True
+    notifications_enabled: Dict[NOTIFIERS, bool] = {
+        NOTIFIERS.EMAIL: True,
+        NOTIFIERS.SMS: False,
+        NOTIFIERS.SLACK: False,
+        NOTIFIERS.APP: False,
+    }
     email: Optional[EmailStr]
     name: Optional[str]
     hashed_password: Optional[str]