handle path arguments correctly

miguelgrinberg · Apr 2, 2021 · a9a216b · a9a216b
1 parent 30caa40
commit a9a216b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/flask_sock/__init__.py b/src/flask_sock/__init__.py
@@ -1,3 +1,4 @@
+from functools import wraps
 from flask import Blueprint, request, abort
 from .ws import WebSocket, ConnectionClosed
 
@@ -18,9 +19,10 @@ def init_app(self, app):
 
     def route(self, path, **kwargs):
         def decorator(f):
-            def websocket_route():
+            @wraps(f)
+            def websocket_route(*args, **kwargs):
                 try:
-                    f(WebSocket(request.environ))
+                    f(WebSocket(request.environ), *args, **kwargs)
                 except ConnectionClosed:
                     pass
                 return ''