diff options
Diffstat (limited to 'day7/backend.py')
| -rw-r--r-- | day7/backend.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/day7/backend.py b/day7/backend.py index 9f9b118..99a62bd 100644 --- a/day7/backend.py +++ b/day7/backend.py @@ -1,7 +1,7 @@ import re from templater import render_template -from utils import parse_cookies, add_headers, SUCCESS, BAD_REQUEST, NOT_FOUND +from utils import parse_cookies, add_headers, SUCCESS, BAD_REQUEST, NOT_FOUND, HTTP_METHODS from config import TEXT_TEMPLATE_NAME import db @@ -34,7 +34,7 @@ def route(url_format, methods=None): def run(method, url: str, cookies: dict, query): - res = NOT_FOUND, NOT_FOUND + 'KAVO' + res = NOT_FOUND, NOT_FOUND for key, value in cookies.items(): db.set_cookie(key, value) @@ -45,6 +45,11 @@ def run(method, url: str, cookies: dict, query): return add_headers(*res) +@route('/.*', list(set(HTTP_METHODS) ^ {'GET', 'POST'})) +def fallback_wrong_method(query, *args): + return NOT_FOUND, 'This method is not allowed' + + @route('/') def index_get(query, *args): return SUCCESS, render_template('form', color=get_color()) |