Extract the notification part of controller to a dedicated class

gns3server/controller/__init__.py

@@ -24,6 +24,7 @@ import aiohttp
 from ..config import Config
 from .project import Project
 from .compute import Compute
+from .notification import Notification
 from ..version import __version__
 
 import logging
@@ -36,6 +37,7 @@ class Controller:
     def __init__(self):
         self._computes = {}
         self._projects = {}
+        self._notification = Notification(self)
 
         if sys.platform.startswith("win"):
             config_path = os.path.join(os.path.expandvars("%APPDATA%"), "GNS3")
@@ -116,6 +118,13 @@ class Controller:
                                           password=server_config.get("password", ""))
         return self._computes["local"]
 
+    @property
+    def notification(self):
+        """
+        The notification system
+        """
+        return self._notification
+
     @property
     def computes(self):
         """
@@ -180,17 +189,3 @@ class Controller:
             Controller._instance = Controller()
         return Controller._instance
 
-    def emit(self, action, event, **kwargs):
-        """
-        Send a notification to clients scoped by projects
-        """
-
-        if "project_id" in kwargs:
-            try:
-                project_id = kwargs.pop("project_id")
-                self._projects[project_id].emit(action, event, **kwargs)
-            except KeyError:
-                pass
-        else:
-            for project_instance in self._projects.values():
-                project_instance.emit(action, event, **kwargs)

gns3server/controller/compute.py

@@ -196,7 +196,7 @@ class Compute:
             msg = json.loads(response.data)
             action = msg.pop("action")
             event = msg.pop("event")
-            self._controller.emit(action, event, compute_id=self.id, **msg)
+            self._controller.notification.emit(action, event, compute_id=self.id, **msg)
 
     def _getUrl(self, path):
         return "{}://{}:{}/v2/compute{}".format(self._protocol, self._host, self._port, path)

gns3server/controller/notification.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2016 GNS3 Technologies Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from contextlib import contextmanager
+
+from ..notification_queue import NotificationQueue
+
+
+class Notification:
+    """
+    Manage notification for the controller
+    """
+
+    def __init__(self, controller):
+        self._controller = controller
+        self._listeners = {}
+
+    @contextmanager
+    def queue(self, project):
+        """
+        Get a queue of notifications
+
+        Use it with Python with
+        """
+        queue = NotificationQueue()
+        self._listeners.setdefault(project.id, set())
+        self._listeners[project.id].add(queue)
+        yield queue
+        self._listeners[project.id].remove(queue)
+
+    def emit(self, action, event, **kwargs):
+        """
+        Send a notification to clients scoped by projects
+
+        :param action: Action name
+        :param event: Event to send
+        :param kwargs: Add this meta to the notification
+        """
+        if "project_id" in kwargs:
+            project_id = kwargs.pop("project_id")
+            self._send_event_to_project(project_id, action, event, **kwargs)
+        else:
+            self._send_event_to_all(action, event, **kwargs)
+
+    def _send_event_to_project(self, project_id, action, event, **kwargs):
+        """
+        Send an event to all the client listening for notifications for
+        this project
+
+        :param project: Project where we need to send the event
+        :param action: Action name
+        :param event: Event to send
+        :param kwargs: Add this meta to the notification
+        """
+        try:
+            project_listeners = self._listeners[project_id]
+        except KeyError:
+            return
+        for listener in project_listeners:
+            listener.put_nowait((action, event, kwargs))
+
+    def _send_event_to_all(self, action, event, **kwargs):
+        """
+        Send an event to all the client listening for notifications on all
+        projects
+
+        :param action: Action name
+        :param event: Event to send
+        :param kwargs: Add this meta to the notification
+        """
+        for project_listeners in self._listeners.values():
+            for listener in project_listeners:
+                listener.put_nowait((action, event, kwargs))

gns3server/controller/project.py

@@ -21,11 +21,9 @@ import aiohttp
 import shutil
 
 from uuid import UUID, uuid4
-from contextlib import contextmanager
 
 from .node import Node
 from .udp_link import UDPLink
-from ..notification_queue import NotificationQueue
 from ..config import Config
 from ..utils.path import check_path_allowed, get_default_project_directory
 
@@ -59,7 +57,6 @@ class Project:
         self._computes = set()
         self._nodes = {}
         self._links = {}
-        self._listeners = set()
 
         # Create the project on demand on the compute node
         self._project_created_on_compute = set()
@@ -203,29 +200,6 @@ class Project:
             yield from compute.delete("/projects/{}".format(self._id))
         shutil.rmtree(self.path, ignore_errors=True)
 
-    @contextmanager
-    def queue(self):
-        """
-        Get a queue of notifications
-
-        Use it with Python with
-        """
-        queue = NotificationQueue()
-        self._listeners.add(queue)
-        yield queue
-        self._listeners.remove(queue)
-
-    def emit(self, action, event, **kwargs):
-        """
-        Send an event to all the client listening for notifications
-
-        :param action: Action name
-        :param event: Event to send
-        :param kwargs: Add this meta to the notification (project_id for example)
-        """
-        for listener in self._listeners:
-            listener.put_nowait((action, event, kwargs))
-
     @classmethod
     def _get_default_project_directory(cls):
         """