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']