diff --git a/mitmproxy2swagger/mitmproxy2swagger.py b/mitmproxy2swagger/mitmproxy2swagger.py index 3f60136..f0bf08e 100644 --- a/mitmproxy2swagger/mitmproxy2swagger.py +++ b/mitmproxy2swagger/mitmproxy2swagger.py @@ -213,19 +213,22 @@ def main(override_args: Optional[Sequence[str]] = None): ) params = swagger_util.url_to_params(url, path_template_to_set) + all_params = [] if args.headers: headers_request = swagger_util.request_to_headers( req.get_request_headers() ) if headers_request is not None and len(headers_request) > 0: - set_key_if_not_exists( - swagger["paths"][path_template_to_set][method], - "parameters", - headers_request, - ) + all_params.extend(headers_request) + if params is not None and len(params) > 0: + all_params.extend(params) + + if len(all_params) > 0: set_key_if_not_exists( - swagger["paths"][path_template_to_set][method], "parameters", params + swagger["paths"][path_template_to_set][method], + "parameters", + all_params, ) if method not in ["get", "head"]: