Merge pull request #8521 from OpenMined/feature/add_notifier_aliases

ADD Activate/Deactivate notifier aliases for the notification service
OpenMined · Feb 27, 2024 · a7c7622 · a7c7622
2 parents 033a0af + 760c5be
commit a7c7622
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/packages/syft/src/syft/service/notification/notification_service.py b/packages/syft/src/syft/service/notification/notification_service.py
@@ -87,6 +87,32 @@ def reply(
                 message="The target notification id {reply.target_msg} was not found!"
             )
 
+    @service_method(
+        path="notifications.activate",
+        name="activate",
+        roles=DATA_SCIENTIST_ROLE_LEVEL,
+    )
+    def activate(
+        self,
+        context: AuthedServiceContext,
+    ) -> Union[Notification, SyftError]:
+        notifier_service = context.node.get_service("notifierservice")
+        result = notifier_service.activate(context.node)
+        return result
+
+    @service_method(
+        path="notifications.deactivate",
+        name="deactivate",
+        roles=DATA_SCIENTIST_ROLE_LEVEL,
+    )
+    def deactivate(
+        self,
+        context: AuthedServiceContext,
+    ) -> Union[Notification, SyftError]:
+        notifier_service = context.node.get_service("notifierservice")
+        result = notifier_service.deactivate(context.node)
+        return result
+
     @service_method(
         path="notifications.get_all",
         name="get_all",