openapi-core/openapi_core/templating/responses/finders.py
2021-03-31 16:17:40 +01:00

23 lines
655 B
Python

from openapi_core.templating.responses.exceptions import ResponseNotFound
class ResponseFinder(object):
def __init__(self, responses):
self.responses = responses
def find(self, http_status='default'):
try:
return self.responses[http_status]
except KeyError:
pass
# try range
http_status_range = '{0}XX'.format(http_status[0])
if http_status_range in self.responses:
return self.responses[http_status_range]
if 'default' not in self.responses:
raise ResponseNotFound(http_status, self.responses)
return self.responses['default']