Provide a msgflo.main(), use in example

Author: jonnor

bin/msgflo-python

@@ -33,14 +33,7 @@ def main():
     module = load_module_file(modulepath)
     classes = find_participant_classes(module)
     Participant = classes[0]
-
-    participant = Participant(role)
-    d = participant.definition
-    waiter = gevent.event.AsyncResult()
-    engine = msgflo.run(participant, done_cb=waiter.set)
-    print "%s(%s) running on %s" % (d['role'], d['component'], engine.broker_url)
-    sys.stdout.flush()
-    waiter.wait()    
+    msgflo.main(Participant)
 
 if __name__ == '__main__':
     logging.basicConfig()

examples/repeat.py

@@ -18,20 +18,6 @@ def process(self, inport, msg):
     self.send('out', msg.data)
     self.ack(msg)
 
-
-def main():
-  waiter = gevent.event.AsyncResult()
-  role = sys.argv[1] if len(sys.argv) > 1 else 'repeat'
-  repeater = Repeat(role)
-  engine = msgflo.run(repeater, done_cb=waiter.set)
-
-  print "Repeat running on %s" % (engine.broker_url)
-  sys.stdout.flush()
-  waiter.wait()
-  print "Shutdown"
-  sys.stdout.flush()
-
 if __name__ == '__main__':
-  logging.basicConfig()
-  main()
+  msgflo.main(Repeat)
 

msgflo/msgflo.py

@@ -288,3 +288,12 @@ def run(participant, broker=None, done_cb=None):
     engine.run()
 
     return engine
+
+def main(Participant):
+    participant = Participant(role)
+    d = participant.definition
+    waiter = gevent.event.AsyncResult()
+    engine = msgflo.run(participant, done_cb=waiter.set)
+    print "%s(%s) running on %s" % (d['role'], d['component'], engine.broker_url)
+    sys.stdout.flush()
+    waiter.wait()