Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity

minor type fixes

Browse Source
This commit is contained in:
Maximilian Hils 2019-11-16 12:09:38 +01:00 committed by GitHub
parent 8158349db5
commit bccee15dcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mitmproxy/addons/export.py
View File

@ -12,8 +12,8 @@ from mitmproxy.net.http.http1 import assemble
from mitmproxy.utils import strutils from mitmproxy.utils import strutils
def cleanup_request(f: flow.Flow): def cleanup_request(f: flow.Flow) -> http.HTTPRequest:
if not hasattr(f, "request") or not f.request: # type: ignore if not hasattr(f, "request") or not f.request:
raise exceptions.CommandError("Can't export flow with no request.") raise exceptions.CommandError("Can't export flow with no request.")
assert isinstance(f, http.HTTPFlow) assert isinstance(f, http.HTTPFlow)
request = f.request.copy() request = f.request.copy()
@ -27,13 +27,15 @@ def cleanup_request(f: flow.Flow):
return request return request
def cleanup_response(f: flow.Flow): def cleanup_response(f: flow.Flow)-> http.HTTPResponse:
if not hasattr(f, "response") or not f.response: # type: ignore if not hasattr(f, "response") or not f.response:
raise exceptions.CommandError("Can't export flow with no response.") raise exceptions.CommandError("Can't export flow with no response.")
assert isinstance(f, http.HTTPFlow)
response = f.response.copy() # type: ignore response = f.response.copy() # type: ignore
response.decode(strict=False) response.decode(strict=False)
return response return response
def request_content_for_console(request: http.HTTPRequest) -> str: def request_content_for_console(request: http.HTTPRequest) -> str:
try: try:
text = request.get_text(strict=True) text = request.get_text(strict=True)