Convert status code to int to allow HTTPStatus members

spec-first · May 28, 2024 · b7ec3da · b7ec3da
1 parent 39da1f2
commit b7ec3da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/connexion/decorators/response.py b/connexion/decorators/response.py
@@ -138,7 +138,7 @@ def _unpack_handler_response(
         elif len(handler_response) == 2:
             data, status_code_or_headers = handler_response
             if isinstance(status_code_or_headers, int):
-                status_code = status_code_or_headers
+                status_code = int(status_code_or_headers)
             elif isinstance(status_code_or_headers, Enum) and isinstance(
                 status_code_or_headers.value, int
             ):

diff --git a/connexion/middleware/response_validation.py b/connexion/middleware/response_validation.py
@@ -97,11 +97,7 @@ async def wrapped_send(message: t.MutableMapping[str, t.Any]) -> None:
                 if message["status"] < 400:
                     self.validate_mime_type(mime_type)
 
-                try:
-                    status = str(message["status"].value)
-                except AttributeError:
-                    status = str(message["status"])
-
+                status = str(message["status"])
                 response_definition = self._operation.response_definition(
                     status, mime_type
                 )