def convert_response(result: RequestsResponse) -> Response:
"""
Utility function to convert a response for the requests library to our internal (Werkzeug based) Response object.
"""
if result is None:
return Response()
if isinstance(result, LambdaResponse):
headers = Headers(dict(result.headers))
for k, values in result.multi_value_headers.items():
for value in values:
headers.add(k, value)
else:
headers = dict(result.headers)
response = Response(status=result.status_code, headers=headers)
if isinstance(result.content, dict):
response.set_json(result.content)
elif isinstance(result.content, (str, bytes)):
response.data = result.content
else:
raise ValueError(f"Unhandled content type {type(result.content)}")
return response
def __call__(
self,
chain: HandlerChain,
exception: Exception,
context: RequestContext,
response: Response,
):
if response.data:
# response already set
return
LOG.debug("setting internal failure response for %s", exception)
response.status_code = 500
response.set_json(
{
"message": "Unexpected exception",
"error": str(exception),
"type": str(exception.__class__.__name__),
}
)
def handler_bar(_request: Request, bar, baz) -> Dict[str, any]:
response = Response()
response.set_json({"bar": bar, "baz": baz})
return response
def on_get(self, req, resp: Response):
resp.set_json({"message": "GET/OK"})
def ep_index2(_: Request, args) -> Response:
response = Response()
response.set_json(dict(method="2", **args))
return response
def ep_all(_: Request, args) -> Response:
response = Response()
response.set_json(dict(method="all", **args))
return response
def users(_: Request, args) -> Response:
response = Response()
response.set_json(args)
return response
def index(_: Request, args) -> Response:
response = Response()
response.set_json(args)
return response
def echo_json(request: Request, args) -> Response:
response = Response()
response.set_json(request.json)
return response
