From 70cd632270083acc2bb835069d6e00f07a728cb3 Mon Sep 17 00:00:00 2001 From: Jesse Luehrs Date: Tue, 23 Sep 2014 14:01:46 -0400 Subject: reorganize --- termcast_server/pubsub.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 termcast_server/pubsub.py (limited to 'termcast_server/pubsub.py') 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) -- cgit v1.2.3-54-g00ecf