add: unsubscribe method for Subscriber class

faif · Mar 6, 2016 · e9619ad · e9619ad
1 parent a73a054
commit e9619ad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/publish_subscribe.py b/publish_subscribe.py
@@ -48,6 +48,9 @@ def __init__(self, name, msg_center):
     def subscribe(self, msg):
         self.provider.subscribe(msg, self)
 
+    def unsubscribe(self, msg):
+        self.provider.unsubscribe(msg, self)
+
     def run(self, msg):
         print("{} got {}".format(self.name, msg))
 
@@ -63,6 +66,9 @@ def main():
     jack.subscribe("music")
     gee = Subscriber("gee", message_center)
     gee.subscribe("movie")
+    vani = Subscriber("vani", message_center)
+    vani.subscribe("movie")
+    vani.unsubscribe("movie")
 
     fftv.publish("cartoon")
     fftv.publish("music")