happyz
/
fastapi
mirror of https://github.com/tiangolo/fastapi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(openapi.utils): checks whether to superimpose an additional response's media_type over the default route's response class's media_type

This commit is contained in:
Vinícius Aguiar 2024-03-01 19:13:37 -03:00
parent b6b160260e
commit 932659f49b
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
fastapi/openapi/utils.py
View File

@ -345,11 +345,15 @@ def get_openapi_path(
)
else:
response_schema = {}
operation.setdefault("responses", {}).setdefault(
status_code, {}
).setdefault("content", {}).setdefault(route_response_media_type, {})[
"schema"
] = response_schema
route_responses = dict((str(k), v) for k, v in route.responses.items())
if status_code not in route_responses or not route_responses.get(status_code).get('superimpose'):
operation.setdefault("responses", {}).setdefault(
status_code, {}
).setdefault("content", {}).setdefault(route_response_media_type, {})[
"schema"
] = response_schema
if route.responses:
operation_responses = operation.setdefault("responses", {})
for (
@ -358,6 +362,7 @@ def get_openapi_path(
) in route.responses.items():
process_response = additional_response.copy()
process_response.pop("model", None)
process_response.pop("superimpose", None)
status_code_key = str(additional_status_code).upper()
if status_code_key == "DEFAULT":
status_code_key = "default"