Skip to content

Commit 0feb8f3

Browse files
committed
update response validation information
1 parent 40fb407 commit 0feb8f3

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

flask_openapi3/utils.py

+2-10
Original file line numberDiff line numberDiff line change
@@ -275,10 +275,6 @@ def get_responses(
275275
}
276276
)
277277
_schemas[UnprocessableEntity.__name__] = Schema(**UnprocessableEntity.schema())
278-
# if not responses.get("500"):
279-
# _responses["500"] = Response(description=HTTP_STATUS["500"])
280-
# handle extra_responses
281-
282278
for key, response in responses.items():
283279
# Verify that the response is a class and that class is a subclass of `pydantic.BaseModel`
284280
if inspect.isclass(response) and issubclass(response, BaseModel):
@@ -302,17 +298,13 @@ def get_responses(
302298
if definitions:
303299
for name, value in definitions.items():
304300
_schemas[name] = Schema(**value)
305-
306-
# Verify that if the key is "204", the response is None,
307-
# because http status code "204" means return "No Content"
301+
# Verify that if the response is None, because http status code "204" means return "No Content"
308302
elif response is None:
309303
_responses[key] = Response(
310304
description=HTTP_STATUS.get(key, ""),
311305
)
312-
313306
else:
314-
raise AttributeError(f'{response} is invalid `pydantic.BaseModel` '
315-
f'or if the key is "204" the type should be None')
307+
raise TypeError(f'{response} is invalid `pydantic.BaseModel`.')
316308
# handle extra_responses
317309
for key, value in extra_responses.items():
318310
# key "200" value {"content":{"text/csv":{"schema":{"type": "string"}}}}

0 commit comments

Comments
 (0)