1 files changed, 40 insertions, 0 deletions
diff --git a/day9/task5/router.py b/day9/task5/router.py
new file mode 100644
index 0000000..c3857f7
--- /dev/null
+++ b/day9/task5/router.py
@@ -0,0 +1,40 @@
+import re
+from utils import NOT_FOUND_CODE, BAD_REQUEST_CODE
+
+
+def route(url_format, methods=None):
+    if methods is None:
+        methods = ['GET']
+
+    def wrapper(func):
+        def inner(url, query, *args, **kwargs):
+            pattern = re.compile(url_format)
+            match = re.fullmatch(pattern, url)
+
+            if match is None or len(match.groups()) != pattern.groups:
+                return BAD_REQUEST_CODE
+
+            return func(query, *match.groups(), *args, **kwargs)
+
+        _router_tree[url_format] = _router_tree.get(url_format, {})
+        for method in methods:
+            _router_tree[url_format][method] = inner
+
+        return inner
+
+    return wrapper
+
+
+def run(request):
+    res = NOT_FOUND_CODE
+
+    method, url = request['method'], request['url']
+    for url_pattern in _router_tree:
+        if re.fullmatch(url_pattern, url) and method in _router_tree[url_pattern]:
+            res = _router_tree[url_pattern][method](url, request['query'])
+            break
+
+    return res
+
+
+_router_tree = {}