X-Git-Url: https://scripts.mit.edu/gitweb/wizard.git/blobdiff_plain/e9659460546dcbdfe603b20d5d41e45e72fb9f65..3a532cd77a685f25452fc076f0caf0a58c530ef3:/bin/wizard

diff --git a/bin/wizard b/bin/wizard
index 1dc3144..e492a53 100755
--- a/bin/wizard
+++ b/bin/wizard
@@ -12,7 +12,7 @@ import decorator
 sys.path.insert(0,os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 import wizard
-from wizard import command
+from wizard import command, prompt
 
 def main():
     usage = """usage: %prog COMMAND [ARGS]
@@ -89,6 +89,9 @@ See '%prog help COMMAND' for more information on a specific command."""
     command_module = get_command(command_name)
     try:
         command_module.main(rest_argv, baton)
+    except prompt.UserCancel as e:
+        print str(e)
+        sys.exit(1)
     except Exception as e:
         # log the exception
         msg = traceback.format_exc()