From 70d807b1641bde9c525429191ab3f5e6c7b71554 Mon Sep 17 00:00:00 2001 From: Sarah Date: Fri, 2 Aug 2024 15:19:30 +0200 Subject: [PATCH] Avoid AttributeError when key or value are already decoded --- mitmproxy2swagger/mitmproxy2swagger.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mitmproxy2swagger/mitmproxy2swagger.py b/mitmproxy2swagger/mitmproxy2swagger.py index e2e2f74..c535f27 100644 --- a/mitmproxy2swagger/mitmproxy2swagger.py +++ b/mitmproxy2swagger/mitmproxy2swagger.py @@ -262,7 +262,11 @@ def main(override_args: Optional[Sequence[str]] = None): did_find_anything = False for key, value in body_val_bytes.items(): did_find_anything = True - body_val[key.decode("utf-8")] = value.decode("utf-8") + if type(key) != str: + key = key.decode("utf-8") + if type(value) != str: + value = value.decode("utf-8") + body_val[key] = value if did_find_anything: content_type = "application/x-www-form-urlencoded" else: