mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-22 15:37:45 +00:00
f51019cb63
* [#4235] Automatic view based on should_render method instead of content_types property * [#4235] Update CHENGELOG * [#4235] Fix linter warnings * Add an explicit test for the new forward-compatible behaviour * wip * contentviews: introduce render_priority (2/2) * coverage++, lint! * minor fixes Co-authored-by: Maximilian Hils <git@maximilianhils.com>
44 lines
1.0 KiB
Python
44 lines
1.0 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.net import http
|
|
|
|
|
|
class ViewSwapCase(contentviews.View):
|
|
name = "swapcase"
|
|
|
|
def __call__(self, data, **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)
|