openapi-core/openapi_core/templating/responses/finders.py
2021-05-02 22:07:35 +01:00

21 lines
645 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'):
if http_status in self.responses:
return self.responses / http_status
# 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'