From 4bdad6ef159a95fd164ee36da51a14197187d1aa Mon Sep 17 00:00:00 2001
From: Kyle Barron <kylebarron2@gmail.com>
Date: Wed, 13 May 2020 22:01:11 -0600
Subject: [PATCH 1/2] Optionally define custom response parameters from
 function

---
 lambda_proxy/proxy.py | 27 +++++++++++++++------------
 1 file changed, 15 insertions(+), 12 deletions(-)

diff --git a/lambda_proxy/proxy.py b/lambda_proxy/proxy.py
index c16a363..fd6171c 100644
--- a/lambda_proxy/proxy.py
+++ b/lambda_proxy/proxy.py
@@ -704,15 +704,18 @@ def __call__(self, event, context):
                 json.dumps({"errorMessage": str(err)}),
             )
 
-        return self.response(
-            response[0],
-            response[1],
-            response[2],
-            cors=route_entry.cors,
-            accepted_methods=route_entry.methods,
-            accepted_compression=self.event["headers"].get("accept-encoding", ""),
-            compression=route_entry.compression,
-            b64encode=route_entry.b64encode,
-            ttl=route_entry.ttl,
-            cache_control=route_entry.cache_control,
-        )
+        custom_response_params = response[3] if len(response) > 3 else {}
+        response_params = {
+            **dict(
+                cors=route_entry.cors,
+                accepted_methods=route_entry.methods,
+                accepted_compression=self.event["headers"].get("accept-encoding", ""),
+                compression=route_entry.compression,
+                b64encode=route_entry.b64encode,
+                ttl=route_entry.ttl,
+                cache_control=route_entry.cache_control,
+            ),
+            **custom_response_params,
+        }
+
+        return self.response(response[0], response[1], response[2], **response_params)

From 6a47de89a40f63e0002eb59ae87dbd5abd13d644 Mon Sep 17 00:00:00 2001
From: Kyle Barron <kylebarron2@gmail.com>
Date: Wed, 13 May 2020 22:11:54 -0600
Subject: [PATCH 2/2] Remove f string to make flake8 happy

---
 lambda_proxy/proxy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lambda_proxy/proxy.py b/lambda_proxy/proxy.py
index fd6171c..d62929f 100644
--- a/lambda_proxy/proxy.py
+++ b/lambda_proxy/proxy.py
@@ -494,7 +494,7 @@ def new_func(*args, **kwargs) -> Callable:
 
     def setup_docs(self) -> None:
         """Add default documentation routes."""
-        openapi_url = f"/openapi.json"
+        openapi_url = "/openapi.json"
 
         def _openapi() -> Tuple[str, str, str]:
             """Return OpenAPI json."""