mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-21 22:58:24 +00:00
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
"""
|
|
Add a custom message body pretty-printer for use inside mitmproxy.
|
|
|
|
This example shows how one can add a custom contentview to mitmproxy,
|
|
which is used to pretty-print HTTP bodies for example.
|
|
The content view API is explained in the mitmproxy.contentviews module.
|
|
"""
|
|
from typing import Optional
|
|
|
|
from mitmproxy import contentviews, flow
|
|
from mitmproxy import http
|
|
|
|
|
|
class ViewSwapCase(contentviews.View):
|
|
name = "swapcase"
|
|
|
|
def __call__(
|
|
self,
|
|
data: bytes,
|
|
*,
|
|
content_type: Optional[str] = None,
|
|
flow: Optional[flow.Flow] = None,
|
|
http_message: Optional[http.Message] = None,
|
|
**unknown_metadata,
|
|
) -> contentviews.TViewResult:
|
|
return "case-swapped text", contentviews.format_text(data.swapcase())
|
|
|
|
def render_priority(
|
|
self,
|
|
data: bytes,
|
|
*,
|
|
content_type: Optional[str] = None,
|
|
flow: Optional[flow.Flow] = None,
|
|
http_message: Optional[http.Message] = None,
|
|
**unknown_metadata,
|
|
) -> float:
|
|
if content_type == "text/plain":
|
|
return 1
|
|
else:
|
|
return 0
|
|
|
|
|
|
view = ViewSwapCase()
|
|
|
|
|
|
def load(l):
|
|
contentviews.add(view)
|
|
|
|
|
|
def done():
|
|
contentviews.remove(view)
|