From ad973de0b3b87ac9d0c56399a2f4dd6702008c59 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Mon, 29 Dec 2014 14:40:34 +0100 Subject: fix mitmproxy -w --- libmproxy/console/__init__.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'libmproxy/console/__init__.py') diff --git a/libmproxy/console/__init__.py b/libmproxy/console/__init__.py index 7d25d428..0e42e69d 100644 --- a/libmproxy/console/__init__.py +++ b/libmproxy/console/__init__.py @@ -432,20 +432,20 @@ class ConsoleMaster(flow.FlowMaster): print >> sys.stderr, "Script load error:", err sys.exit(1) - if options.wfile: - err = self.start_stream(options.wfile) + if options.outfile: + err = self.start_stream_to_path(options.outfile[0], options.outfile[1]) if err: - print >> sys.stderr, "Script load error:", err + print >> sys.stderr, "Stream file error:", err sys.exit(1) if options.app: self.start_app(self.options.app_host, self.options.app_port) - def start_stream(self, path): + def start_stream_to_path(self, path, mode="wb"): path = os.path.expanduser(path) try: - f = file(path, "wb") - flow.FlowMaster.start_stream(self, f, None) + f = file(path, mode) + self.start_stream(f, None) except IOError, v: return str(v) self.stream_path = path -- cgit v1.2.3