openapi-core/openapi_core/contrib/flask/handlers.py
2020-02-18 11:51:42 +00:00

39 lines
1.1 KiB
Python

"""OpenAPI core contrib flask handlers module"""
from flask.globals import current_app
from flask.json import dumps
from openapi_core.schema.media_types.exceptions import InvalidContentType
from openapi_core.schema.servers.exceptions import InvalidServer
class FlaskOpenAPIErrorsHandler(object):
OPENAPI_ERROR_STATUS = {
InvalidServer: 500,
InvalidContentType: 415,
}
@classmethod
def handle(cls, errors):
data_errors = [
cls.format_openapi_error(err)
for err in errors
]
data = {
'errors': data_errors,
}
data_error_max = max(data_errors, key=lambda x: x['status'])
status = data_error_max['status']
return current_app.response_class(
dumps(data),
status=status,
mimetype='application/json'
)
@classmethod
def format_openapi_error(cls, error):
return {
'title': str(error),
'status': cls.OPENAPI_ERROR_STATUS.get(error.__class__, 400),
'class': str(type(error)),
}