Add manager to WS class

1 files changed, 2 insertions, 0 deletions

diff --git a/pysite/websockets.py b/pysite/websockets.py
index 19f9bf83..d72f44b7 100644
--- a/pysite/websockets.py
+++ b/pysite/websockets.py
@@ -74,6 +74,8 @@ class WS:
         if not cls.path or not cls.name:
             raise RuntimeError("Websockets must have both `path` and `name` defined")
 
+        cls.manager = manager
+
         def handle(socket: WebSocket):
             """
             Wrap the current WS class, dispatching events to it as necessary. We're using gevent, so there's