diff options
Diffstat (limited to 'termcast_server/pubsub.py')
-rw-r--r-- | termcast_server/pubsub.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/termcast_server/pubsub.py b/termcast_server/pubsub.py new file mode 100644 index 0000000..b5faf22 --- /dev/null +++ b/termcast_server/pubsub.py @@ -0,0 +1,31 @@ +class Publisher(object): + def __init__(self): + self.subscribers = [] + + def subscribe(self, who): + if who not in self.subscribers: + self.subscribers.append(who) + + def unsubscribe(self, who): + if who in self.subscribers: + self.subscribers.remove(who) + + def request_all(self, message, *args): + ret = [] + for subscriber in self.subscribers: + method = "request_" + message + if hasattr(subscriber, method): + ret.append(getattr(subscriber, method)(*args)) + return ret + + def request_one(self, message, *args): + for subscriber in self.subscribers: + method = "request_" + message + if hasattr(subscriber, method): + return getattr(subscriber, method)(*args) + + def notify(self, message, *args): + for subscriber in self.subscribers: + method = "msg_" + message + if hasattr(subscriber, method): + getattr(subscriber, method)(*args) |