From bccee15dcbc186d0014809830c5b996a3fb2f442 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sat, 16 Nov 2019 12:09:38 +0100 Subject: [PATCH] minor type fixes --- mitmproxy/addons/export.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/mitmproxy/addons/export.py b/mitmproxy/addons/export.py index d87cd7872..7bd0aef0b 100644 --- a/mitmproxy/addons/export.py +++ b/mitmproxy/addons/export.py @@ -12,8 +12,8 @@ from mitmproxy.net.http.http1 import assemble from mitmproxy.utils import strutils -def cleanup_request(f: flow.Flow): - if not hasattr(f, "request") or not f.request: # type: ignore +def cleanup_request(f: flow.Flow) -> http.HTTPRequest: + if not hasattr(f, "request") or not f.request: raise exceptions.CommandError("Can't export flow with no request.") assert isinstance(f, http.HTTPFlow) request = f.request.copy() @@ -27,13 +27,15 @@ def cleanup_request(f: flow.Flow): return request -def cleanup_response(f: flow.Flow): - if not hasattr(f, "response") or not f.response: # type: ignore +def cleanup_response(f: flow.Flow)-> http.HTTPResponse: + if not hasattr(f, "response") or not f.response: raise exceptions.CommandError("Can't export flow with no response.") + assert isinstance(f, http.HTTPFlow) response = f.response.copy() # type: ignore response.decode(strict=False) return response + def request_content_for_console(request: http.HTTPRequest) -> str: try: text = request.get_text(strict=True)