mirror of
https://github.com/correl/tornado-openapi3.git
synced 2024-11-23 19:19:55 +00:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import itertools
|
|
|
|
from openapi_core.validation.request.datatypes import ( # type: ignore
|
|
RequestParameters,
|
|
OpenAPIRequest,
|
|
)
|
|
from openapi_core.validation.request import validators # type: ignore
|
|
from tornado.httputil import HTTPServerRequest
|
|
from werkzeug.datastructures import ImmutableMultiDict, Headers
|
|
|
|
|
|
class TornadoRequestFactory:
|
|
@classmethod
|
|
def create(cls, request: HTTPServerRequest) -> OpenAPIRequest:
|
|
if request.uri:
|
|
path, _, _ = request.uri.partition("?")
|
|
else:
|
|
path = ""
|
|
query_arguments: ImmutableMultiDict[str, str] = ImmutableMultiDict(
|
|
itertools.chain(
|
|
*[
|
|
[(k, v.decode("utf-8")) for v in vs]
|
|
for k, vs in request.query_arguments.items()
|
|
]
|
|
)
|
|
)
|
|
return OpenAPIRequest(
|
|
full_url_pattern=path,
|
|
method=request.method.lower() if request.method else "get",
|
|
parameters=RequestParameters(
|
|
query=query_arguments, header=Headers(request.headers.get_all())
|
|
),
|
|
body=request.body.decode("utf-8"),
|
|
mimetype=request.headers.get(
|
|
"Content-Type", "application/x-www-form-urlencoded"
|
|
),
|
|
)
|
|
|
|
|
|
class RequestValidator(validators.RequestValidator):
|
|
def validate(self, request: HTTPServerRequest):
|
|
return super().validate(TornadoRequestFactory.create(request))
|