mitmproxy/test/helper_tools/dumperview.py
Aldo Cortesi 67381ae550 Revamp options
- Options are now explicitly initialized with an add_option method
- We have one canonical Options class - ditch dump.Options
2017-03-05 15:10:03 +13:00

58 lines
1.3 KiB
Python
Executable File

#!/usr/bin/env python
import click
from mitmproxy.addons import dumper
from mitmproxy.test import tflow
from mitmproxy.test import taddons
from mitmproxy.tools import options
def show(flow_detail, flows):
d = dumper.Dumper()
with taddons.context(options=options.Options()) as ctx:
ctx.configure(d, flow_detail=flow_detail)
for f in flows:
ctx.cycle(d, f)
@click.group()
def cli():
pass
@cli.command()
@click.option('--level', default=1, help='Detail level')
def tcp(level):
f1 = tflow.ttcpflow(client_conn=True, server_conn=True)
show(level, [f1])
@cli.command()
@click.option('--level', default=1, help='Detail level')
def large(level):
f1 = tflow.tflow(client_conn=True, server_conn=True, resp=True)
f1.response.headers["content-type"] = "text/html"
f1.response.content = b"foo bar voing\n" * 100
show(level, [f1])
@cli.command()
@click.option('--level', default=1, help='Detail level')
def small(level):
f1 = tflow.tflow(client_conn=True, server_conn=True, resp=True)
f1.response.headers["content-type"] = "text/html"
f1.response.content = b"<html><body>Hello!</body></html>"
f2 = tflow.tflow(client_conn=True, server_conn=True, err=True)
show(
level,
[
f1, f2,
]
)
if __name__ == "__main__":
cli()