Allow to have more than one notification type in Service

[gregoa/zavai.git] / zavai / registry.py

diff --git a/zavai/registry.py b/zavai/registry.py
index 4af413c83e9e02c9dd4a313ece374d507c70de67..ec0a4a2c4f3b29c9bc497de4c6b1bedecc63bf77 100644 (file)
--- a/zavai/registry.py
+++ b/zavai/registry.py
@@ -145,6 +145,9 @@ class Registry(object):
         self.objects.clear()
 
 class Resource(object):
         self.objects.clear()
 
 class Resource(object):

+    def __init__(self):
+        super(Resource, self).__init__()
+

     def shutdown(self):
         """Shut down this resource.
 
     def shutdown(self):
         """Shut down this resource.
 


@@ -155,3 +158,49 @@ class Resource(object):
         releasing a FSO resource, restoring mixer settings and so on.
         """
         pass
         releasing a FSO resource, restoring mixer settings and so on.
         """
         pass

+
+class Service(Resource):
+    "Service that is activated only when someone is listening"
+    def __init__(self):
+        """
+        Initialise a service that can emit signals for the given event types
+        """
+        super(Service, self).__init__(types = [])
+        self.callbacks = dict()
+        for t in types:
+            self.callbacks[t] = set()
+
+    def shutdown(self):
+        self.stop()
+
+    def start(self):
+        "Activate the service"
+        pass
+
+    def stop(self):
+        "Deactivate the service"
+        pass
+
+    def notify(self, type, *args, **kw):
+        "Call all callbacks with the given parameters"
+        for cb in self.callbacks[type]:
+            cb(*args, **kw)
+
+    def has_callbacks(self):
+        for i in self.callbacks.values():
+            if i: return True
+        return False
+
+    def connect(self, type, callback):
+        "Connect a callback to this resource, activating it if needed"
+        do_start = not self.has_callbacks()
+        self.callbacks[type].add(callback)
+        if do_start:
+            self.start()
+
+    def disconnect(self, type, callback):
+        "Disconnect a callback to this resource, activating it if needed"
+        if not self.has_callbacks(): return
+        self.callbacks.discard(callback)
+        if not self.has_callbacks():
+            self.stop()