openapi-core/openapi_core/contrib/flask/views.py
2020-01-26 23:29:19 +00:00

27 lines
1.1 KiB
Python

"""OpenAPI core contrib flask views module"""
from flask.views import MethodView
from openapi_core.contrib.flask.decorators import FlaskOpenAPIViewDecorator
from openapi_core.contrib.flask.handlers import FlaskOpenAPIErrorsHandler
from openapi_core.validation.request.validators import RequestValidator
from openapi_core.validation.response.validators import ResponseValidator
class FlaskOpenAPIView(MethodView):
"""Brings OpenAPI specification validation and unmarshalling for views."""
openapi_errors_handler = FlaskOpenAPIErrorsHandler
def __init__(self, spec):
super(FlaskOpenAPIView, self).__init__()
self.request_validator = RequestValidator(spec)
self.response_validator = ResponseValidator(spec)
def dispatch_request(self, *args, **kwargs):
decorator = FlaskOpenAPIViewDecorator(
request_validator=self.request_validator,
response_validator=self.response_validator,
openapi_errors_handler=self.openapi_errors_handler,
)
return decorator(super(FlaskOpenAPIView, self).dispatch_request)(
*args, **kwargs)