Source code for jivago.wsgi.invocation.parameter_selection.query_parameter_selector

from jivago.lang.annotations import Override
from jivago.wsgi.invocation.missing_route_invocation_argument import MissingRouteInvocationArgument
from jivago.wsgi.invocation.parameter_selection.parameter_selector import ParameterSelector
from jivago.wsgi.invocation.parameters import QueryParam
from jivago.wsgi.request.request import Request

ALLOWED_QUERY_PARAMETER_TYPES = (str, int, float)




class QueryParameterSelector(ParameterSelector):



    @Override
    def matches(self, paramater_declaration: type) -> bool:
        return hasattr(paramater_declaration, '__origin__') and paramater_declaration.__origin__ == QueryParam




    @Override
    def format_parameter(self, parameter_name: str, parameter_declaration: type, request: Request) -> object:
        parameter_type = parameter_declaration.__args__[0]
        if parameter_type in ALLOWED_QUERY_PARAMETER_TYPES:
            try:
                return parameter_type(request.query_form[parameter_name])
            except (KeyError, ValueError):
                raise MissingRouteInvocationArgument(parameter_name)
        raise MissingRouteInvocationArgument(parameter_name)