diff options
Diffstat (limited to 'libmproxy/dump.py')
| -rw-r--r-- | libmproxy/dump.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/libmproxy/dump.py b/libmproxy/dump.py index c2a3268a..65e212bf 100644 --- a/libmproxy/dump.py +++ b/libmproxy/dump.py @@ -1,6 +1,4 @@ from __future__ import absolute_import, print_function -import sys -import os import traceback import click @@ -10,7 +8,7 @@ from netlib.http import CONTENT_MISSING import netlib.utils from . import flow, filt, contentviews from .exceptions import ContentViewException -from .models import HTTPRequest + class DumpError(Exception): pass @@ -58,6 +56,7 @@ class Options(object): class DumpMaster(flow.FlowMaster): + def __init__(self, server, options, outfile=None): flow.FlowMaster.__init__(self, server, flow.State()) self.outfile = outfile @@ -171,7 +170,7 @@ class DumpMaster(flow.FlowMaster): "{}: {}".format( click.style(k, fg="blue", bold=True), click.style(v, fg="blue")) - for k, v in message.headers.fields + for k, v in message.headers.fields ) self.echo(headers, indent=4) if self.o.flow_detail >= 3: @@ -237,7 +236,7 @@ class DumpMaster(flow.FlowMaster): client = click.style("[replay]", fg="yellow", bold=True) method = flow.request.method - method_color=dict( + method_color = dict( GET="green", DELETE="red" ).get(method.upper(), "magenta") |