diff options
Diffstat (limited to 'day9/task5/router.py')
| -rw-r--r-- | day9/task5/router.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/day9/task5/router.py b/day9/task5/router.py index c3857f7..0827bb4 100644 --- a/day9/task5/router.py +++ b/day9/task5/router.py @@ -1,5 +1,6 @@ import re from utils import NOT_FOUND_CODE, BAD_REQUEST_CODE +from core import Response, ErrorResponse def route(url_format, methods=None): @@ -12,7 +13,7 @@ def route(url_format, methods=None): match = re.fullmatch(pattern, url) if match is None or len(match.groups()) != pattern.groups: - return BAD_REQUEST_CODE + return ErrorResponse(BAD_REQUEST_CODE) return func(query, *match.groups(), *args, **kwargs) @@ -25,8 +26,8 @@ def route(url_format, methods=None): return wrapper -def run(request): - res = NOT_FOUND_CODE +def run(request) -> Response: + res = ErrorResponse(NOT_FOUND_CODE) method, url = request['method'], request['url'] for url_pattern in _router_tree: |