Background loading, initially implemented for search window

[jamaendo] / jamaui / postoffice.py

diff --git a/jamaui/postoffice.py b/jamaui/postoffice.py
index 675f9bf..a84158b 100644 (file)
--- a/jamaui/postoffice.py
+++ b/jamaui/postoffice.py
@@ -23,38 +23,45 @@
 #
 # message central
 
 #
 # message central
 

+from __future__ import with_statement

 import logging
 import logging

+import threading

 
 log = logging.getLogger(__name__)
 
 class PostOffice(object):
 
     def __init__(self):
 
 log = logging.getLogger(__name__)
 
 class PostOffice(object):
 
     def __init__(self):

+        self.lock = threading.RLock()

         self.tags = {} # tag -> [callback]
 
     def notify(self, tag, *data):
         self.tags = {} # tag -> [callback]
 
     def notify(self, tag, *data):

-        clients = self.tags.get(tag)
-        if clients:
-            log.debug("(%s %s) -> [%s]",
-                      tag,
-                      " ".join(str(x) for x in data),
-                      " ".join(str(x) for x in clients))
-            for client in clients:
-                client(*data)
-
-    def connect(self, tag, callback):
-        if tag not in self.tags:
-            self.tags[tag] = []
-        clients = self.tags[tag]
-        if callback not in clients:
-            clients.append(callback)
-
-    def disconnect(self, tag, callback):
-        if tag not in self.tags:
-            self.tags[tag] = []
-        clients = self.tags[tag]
-        if callback in clients:
-            clients.remove(callback)
+        with self.lock:
+            log.info("(%s %s)", tag, " ".join(str(x) for x in data))
+            clients = self.tags.get(tag)
+            if clients:
+                for ref, client in clients:
+                    client(*data)
+
+    def connect(self, tag, ref, callback):
+        with self.lock:
+            if not isinstance(tag, list):
+                tag = [tag]
+            for t in tag:
+                if t not in self.tags:
+                    self.tags[t] = []
+                clients = self.tags[t]
+                if callback not in clients:
+                    clients.append((ref, callback))
+
+    def disconnect(self, tag, ref):
+        with self.lock:
+            if not isinstance(tag, list):
+                tag = [tag]
+            for t in tag:
+                if t not in self.tags:
+                    self.tags[t] = []
+                self.tags[t] = [(_ref, cb) for _ref, cb in self.tags[t] if _ref != ref]

 
 postoffice = PostOffice()
 
 
 postoffice = PostOffice()