Merge pull request faif#85 from JDegner0129/fix-pubsub-conditional

faif · faif · commit cf4b435aa8a5 · 2015-05-09T19:19:26.000+02:00
DRY up the subscribe conditional a bit
diff --git a/publish_subscribe.py b/publish_subscribe.py
@@ -17,11 +17,7 @@ def notify(self, msg):
         self.msg_queue.append(msg)
 
     def subscribe(self, msg, subscriber):
-        if msg not in self.subscribers:
-            self.subscribers[msg] = []
-            self.subscribers[msg].append(subscriber)  # unfair
-        else:
-            self.subscribers[msg].append(subscriber)
+        self.subscribers.setdefault(msg, []).append(subscriber)
 
     def unsubscribe(self, msg, subscriber):
         self.subscribers[msg].remove(subscriber)
