cause BANE to shutdown gracefully when ctrl+C'd. reslove #38

PaulHancock · May 5, 2018 · 8f98336 · 8f98336
1 parent af97f5c
commit 8f98336
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/AegeanTools/BANE.py b/AegeanTools/BANE.py
@@ -415,7 +415,12 @@ def filter_mc_sharemem(filename, step_size, box_size, cores, shape, dobkg=True):
             args.append((filename, region, step_size, box_size, shape, dobkg))
 
     pool = multiprocessing.Pool(processes=cores)
-    pool.map(_sf2, args)
+    try:
+        pool.map_async(_sf2, args).get(timeout=10000000)
+    except KeyboardInterrupt:
+        logging.error("Caught keyboard interrupt")
+        pool.close()
+        sys.exit(1)
     pool.close()
     pool.join()