From 670799fe8492ea205d863c9773d9de59bc8c142e Mon Sep 17 00:00:00 2001 From: Volodymyr Pavlenko Date: Tue, 10 Nov 2015 13:19:40 +0200 Subject: [PATCH] Fixes linkedin/simoorg#13 --- src/simoorg/__main__.py | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/src/simoorg/__main__.py b/src/simoorg/__main__.py index 5da2993..a028360 100755 --- a/src/simoorg/__main__.py +++ b/src/simoorg/__main__.py @@ -18,18 +18,23 @@ import simoorg.moirai as moi import sys -if len(sys.argv) != 2: - print "Please provide config path" - exit(1) -CONFIG_PATH = sys.argv[1] -MOIRAI = moi.Moirai(CONFIG_PATH, verbose=True, debug=False) +def main(): + if len(sys.argv) != 2: + print "Please provide config path" + exit(1) + CONFIG_PATH = sys.argv[1] -MOIRAI.spawn_atropos() + MOIRAI = moi.Moirai(CONFIG_PATH, verbose=True, debug=False) -for service_name, proc_handler in MOIRAI.atropos_army.iteritems(): - print service_name, proc_handler.pid + MOIRAI.spawn_atropos() -# Wait for all the atropos to finish + for service_name, proc_handler in MOIRAI.atropos_army.iteritems(): + print service_name, proc_handler.pid -MOIRAI.finish() + # Wait for all the atropos to finish + + MOIRAI.finish() + +if __name__ == '__main__': + main()