From d17b9d62305507a93836eb6ee085fa3edd029bed Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sat, 16 Nov 2019 13:54:44 +0100 Subject: [PATCH] make mypy happy --- mitmproxy/addons/export.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mitmproxy/addons/export.py b/mitmproxy/addons/export.py index 9d04997ff..68df93744 100644 --- a/mitmproxy/addons/export.py +++ b/mitmproxy/addons/export.py @@ -13,7 +13,7 @@ from mitmproxy.utils import strutils def cleanup_request(f: flow.Flow) -> http.HTTPRequest: - if not hasattr(f, "request") or not f.request: + if not getattr(f, "request", None): raise exceptions.CommandError("Can't export flow with no request.") assert isinstance(f, http.HTTPFlow) request = f.request.copy() @@ -28,7 +28,7 @@ def cleanup_request(f: flow.Flow) -> http.HTTPRequest: def cleanup_response(f: flow.Flow) -> http.HTTPResponse: - if not hasattr(f, "response") or not f.response: + if not getattr(f, "response", None): raise exceptions.CommandError("Can't export flow with no response.") assert isinstance(f, http.HTTPFlow) response = f.response.copy() # type: ignore