ee: Graceful exit if StopToolException is thrown.

1 files changed, 7 insertions, 2 deletions

diff --git a/src/ee/__main__.py b/src/ee/__main__.py
index 3feec33..b6fdd29 100644
--- a/src/ee/__main__.py
+++ b/src/ee/__main__.py
@@ -56,8 +56,13 @@ def main():
         if t.name != name:
             continue
         sys.argv[0] = t.name
-        importlib.import_module(t.module_name)
-        exit(0)
+        try:
+            importlib.import_module(t.module_name)
+            exit(0)
+        except ee.StopToolException as e:
+            if e.args:
+                print("{}: {}".format(t.name, " ".join(e.args)))
+            exit(1)
 
     eprint("No such tool: {}".format(name))
     exit(1)