From e7b61995adcb8f6e5391847043d5be3c69bfc3f9 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Tue, 14 Jun 2016 12:34:22 +1200
Subject: A tiny wrapper to make script invocations nicer

---
 mitmproxy/script/script.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

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,
-- 
cgit v1.2.3