From a6c7a1ff918c5aa0285decb995096190888a2f51 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sun, 16 Oct 2016 11:34:27 +1300
Subject: scripts: handle SystemExit from scripts explicitly

---
 mitmproxy/builtins/script.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mitmproxy/builtins/script.py b/mitmproxy/builtins/script.py
index bbefc5c7..15ee4936 100644
--- 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
-- 
cgit v1.2.3