scripts: handle SystemExit from scripts explicitly

2016-10-16 11:34:27 +13:00 · 2016-10-16 11:34:27 +13:00 · a6c7a1ff91
parent dc19ff7a9d
commit a6c7a1ff91
1 changed files with 2 additions and 0 deletions
--- a/mitmproxy/builtins/script.py
+++ b/mitmproxy/builtins/script.py
@ -88,6 +88,8 @@ def scriptenv(path, args):
    sys.path.append(script_dir)
    try:
        yield
+    except SystemExit as v:
+        ctx.log.error("Script exited with code %s" % v.code)
    except Exception:
        etype, value, tb = sys.exc_info()
        tb = cut_traceback(tb, "scriptenv").tb_next