openapi-core/openapi_core/contrib/flask/handlers.py

"""OpenAPI core contrib flask handlers module"""
from flask.globals import current_app
from flask.json import dumps

from openapi_core.templating.media_types.exceptions import MediaTypeNotFound
from openapi_core.templating.paths.exceptions import (
    ServerNotFound, OperationNotFound, PathNotFound,
)


class FlaskOpenAPIErrorsHandler(object):

    OPENAPI_ERROR_STATUS = {
        ServerNotFound: 400,
        OperationNotFound: 405,
        PathNotFound: 404,
        MediaTypeNotFound: 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)),
        }