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]
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()