improve script loading

1 files changed, 21 insertions, 1 deletions

diff --git a/mitmproxy/exceptions.py b/mitmproxy/exceptions.py
index 86bf75ae..52fd36d1 100644
--- a/mitmproxy/exceptions.py
+++ b/mitmproxy/exceptions.py
@@ -7,6 +7,10 @@ See also: http://lucumr.pocoo.org/2014/10/16/on-error-handling/
 """
 from __future__ import (absolute_import, print_function, division)
 
+import traceback
+
+import sys
+
 
 class ProxyException(Exception):
     """
@@ -59,7 +63,23 @@ class ReplayException(ProxyException):
 
 
 class ScriptException(ProxyException):
-    pass
+    @classmethod
+    def from_exception_context(cls, cut_tb=1):
+        """
+        Must be called while the current stack handles an exception.
+
+        Args:
+            cut_tb: remove N frames from the stack trace to hide internal calls.
+        """
+        exc_type, exc_value, exc_traceback = sys.exc_info()
+
+        while cut_tb > 0:
+            exc_traceback = exc_traceback.tb_next
+            cut_tb -= 1
+
+        tb = "".join(traceback.format_exception(exc_type, exc_value, exc_traceback))
+
+        return cls(tb)
 
 
 class FlowReadException(ProxyException):