aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/script.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2011-08-03 16:36:20 +1200
committerAldo Cortesi <aldo@nullcube.com>2011-08-03 16:36:20 +1200
commit179cf758624152a4988e04a767759cb15b9c358a (patch)
tree6e9351565b9241c9f8d751418cae93cd012d0550 /libmproxy/script.py
parentf7e4e89b124e0bc20cf8192227add04bd92846ae (diff)
downloadmitmproxy-179cf758624152a4988e04a767759cb15b9c358a.tar.gz
mitmproxy-179cf758624152a4988e04a767759cb15b9c358a.tar.bz2
mitmproxy-179cf758624152a4988e04a767759cb15b9c358a.zip
Add script hooks, enable new engine for mitmdump.
Diffstat (limited to 'libmproxy/script.py')
-rw-r--r--libmproxy/script.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/libmproxy/script.py b/libmproxy/script.py
index da3131a8..03eff958 100644
--- a/libmproxy/script.py
+++ b/libmproxy/script.py
@@ -8,7 +8,7 @@ class Context:
self.master, self.state = master, state
def log(self, *args, **kwargs):
- self.master.log(*args, **kwargs)
+ self.master.add_event(*args, **kwargs)
class Script:
@@ -32,9 +32,14 @@ class Script:
Raises ScriptError on failure, with argument equal to an error
message that may be a formatted traceback.
"""
+ path = os.path.expanduser(self.path)
+ if not os.path.exists(path):
+ raise ScriptError("No such file: %s"%self.path)
+ if not os.path.isfile(path):
+ raise ScriptError("Not a file: %s"%self.path)
ns = {}
try:
- self.mod = execfile(os.path.expanduser(self.path), {}, ns)
+ self.mod = execfile(path, ns, ns)
except Exception, v:
raise ScriptError(traceback.format_exc(v))
self.ns = ns