mirror of
https://github.com/correl/openapi-core.git
synced 2024-11-25 11:09:53 +00:00
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
"""OpenAPI core responses models module"""
|
|
from openapi_core.schema.content.exceptions import MimeTypeNotFound
|
|
from openapi_core.schema.content.models import Content
|
|
from openapi_core.schema.media_types.exceptions import InvalidContentType
|
|
from openapi_core.schema.responses.exceptions import MissingResponseContent
|
|
|
|
|
|
class Response(object):
|
|
|
|
def __init__(
|
|
self, http_status, description, headers=None, content=None,
|
|
links=None):
|
|
self.http_status = http_status
|
|
self.description = description
|
|
self.headers = headers and dict(headers) or {}
|
|
self.content = content and Content(content) or Content()
|
|
self.links = links and dict(links) or {}
|
|
|
|
def __getitem__(self, mimetype):
|
|
return self.get_content_type(mimetype)
|
|
|
|
def get_content_type(self, mimetype):
|
|
try:
|
|
return self.content[mimetype]
|
|
except MimeTypeNotFound:
|
|
raise InvalidContentType(
|
|
"Invalid mime type `{0}`".format(mimetype))
|
|
|
|
def get_value(self, response):
|
|
if not response.data:
|
|
raise MissingResponseContent("Missing response content")
|
|
|
|
return response.data
|