mirror of
https://github.com/correl/openapi-core.git
synced 2024-11-29 03:00:12 +00:00
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
"""OpenAPI core contrib requests requests module"""
|
|
from typing import Union
|
|
from werkzeug.datastructures import ImmutableMultiDict
|
|
from requests import Request, PreparedRequest
|
|
from urllib.parse import urlparse, parse_qs
|
|
|
|
from openapi_core.validation.request.datatypes import (
|
|
RequestParameters, OpenAPIRequest,
|
|
)
|
|
|
|
|
|
class RequestsOpenAPIRequestFactory(object):
|
|
|
|
@classmethod
|
|
def create(cls, request: Union[Request, PreparedRequest]) -> OpenAPIRequest:
|
|
"""
|
|
Converts a requests request to an OpenAPI one
|
|
|
|
Internally converts to a `PreparedRequest` first to parse the exact
|
|
payload being sent
|
|
"""
|
|
if isinstance(request, Request):
|
|
request = request.prepare()
|
|
|
|
# Method
|
|
method = request.method.lower()
|
|
|
|
# Cookies
|
|
cookie = {}
|
|
if request._cookies is not None:
|
|
# cookies are stored in a cookiejar object
|
|
cookie = request._cookies.get_dict()
|
|
|
|
# Preparing a request formats the URL with params, strip them out again
|
|
o = urlparse(request.url)
|
|
params = parse_qs(o.query)
|
|
# extract the URL without query parameters
|
|
url = o._replace(query=None).geturl()
|
|
|
|
# gets deduced by path finder against spec
|
|
path = {}
|
|
|
|
# Order matters because all python requests issued from a session include
|
|
# Accept */* which does not necessarily match the content type
|
|
mimetype = request.headers.get('Content-Type') or \
|
|
request.headers.get('Accept')
|
|
|
|
# Headers - request.headers is not an instance of dict, which is expected
|
|
header = dict(request.headers)
|
|
|
|
# Body
|
|
# TODO: figure out if request._body_position is relevant
|
|
body = request.body
|
|
|
|
parameters = RequestParameters(
|
|
query=ImmutableMultiDict(params),
|
|
header=header,
|
|
cookie=cookie,
|
|
path=path,
|
|
)
|
|
return OpenAPIRequest(
|
|
full_url_pattern=url,
|
|
method=method,
|
|
parameters=parameters,
|
|
body=body,
|
|
mimetype=mimetype,
|
|
)
|