diff options
| -rw-r--r-- | mitmproxy/script/script.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/mitmproxy/script/script.py b/mitmproxy/script/script.py index 70f74817..48efcde5 100644 --- a/mitmproxy/script/script.py +++ b/mitmproxy/script/script.py @@ -9,12 +9,21 @@ from __future__ import absolute_import, print_function, division import os import shlex import sys +import contextlib import six from mitmproxy import exceptions +@contextlib.contextmanager +def setargs(args): + oldargs = sys.argv + sys.argv = args + yield + sys.argv = oldargs + + class Script(object): """ @@ -113,7 +122,8 @@ class Script(object): f = self.ns.get(name) if f: try: - return f(self.ctx, *args, **kwargs) + with setargs(self.args): + return f(self.ctx, *args, **kwargs) except Exception: six.reraise( exceptions.ScriptException, |