From 854c6d3d653238bc2d1b5b57f5939fcd606efbab Mon Sep 17 00:00:00 2001 From: taoufik07 Date: Tue, 26 Feb 2019 16:44:12 +0100 Subject: [PATCH] Add @before_request --- responder/api.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/responder/api.py b/responder/api.py index 5db9362..899478c 100644 --- a/responder/api.py +++ b/responder/api.py @@ -179,6 +179,16 @@ class API: start_response("404 NOT FOUND", [("Content-Type", "text/plain")]) return [b"Not Found."] + def before_request(self, websocket=False): + def decorator(f): + if websocket: + self.before_requests.setdefault("ws", []).append(f) + else: + self.before_requests.setdefault("http", []).append(f) + return f + + return decorator + @property def before_http_requests(self): return self.before_requests.get("http", [])