Add @before_request

2026-06-05 23:00:17 +00:00 · 2019-02-26 16:44:12 +01:00
parent f9a850a8fe
commit 854c6d3d65
1 changed files with 10 additions and 0 deletions
@@ -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", [])