diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2011-08-03 16:36:20 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2011-08-03 16:36:20 +1200 |
commit | 179cf758624152a4988e04a767759cb15b9c358a (patch) | |
tree | 6e9351565b9241c9f8d751418cae93cd012d0550 /libmproxy/script.py | |
parent | f7e4e89b124e0bc20cf8192227add04bd92846ae (diff) | |
download | mitmproxy-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.py | 9 |
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 |