diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-05-03 19:54:16 -0700 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-05-03 19:54:16 -0700 |
commit | ff53e7a6ebc878b0a8557112d070d8ca6e91562d (patch) | |
tree | 43cffd81e8c565dbc5645f9a8e44290a6d367b2d /mitmproxy/script | |
parent | 714ba1611125601e4eb21a2ceb5fa7f742d008b9 (diff) | |
download | mitmproxy-ff53e7a6ebc878b0a8557112d070d8ca6e91562d.tar.gz mitmproxy-ff53e7a6ebc878b0a8557112d070d8ca6e91562d.tar.bz2 mitmproxy-ff53e7a6ebc878b0a8557112d070d8ca6e91562d.zip |
scripts: allow package imports
Diffstat (limited to 'mitmproxy/script')
-rw-r--r-- | mitmproxy/script/script.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/mitmproxy/script/script.py b/mitmproxy/script/script.py index 5a8334c4..4d0f73fa 100644 --- a/mitmproxy/script/script.py +++ b/mitmproxy/script/script.py @@ -74,6 +74,7 @@ class Script(object): script_dir = os.path.dirname(os.path.abspath(self.args[0])) self.ns = {'__file__': os.path.abspath(self.args[0])} sys.path.append(script_dir) + sys.path.append(os.path.join(script_dir, "..")) try: with open(self.filename) as f: code = compile(f.read(), self.filename, 'exec') @@ -86,6 +87,7 @@ class Script(object): ) finally: sys.path.pop() + sys.path.pop() return self.run("start", self.args) def unload(self): |