Websocket test

2021-04-14 09:29:25 +01:00
parent 2ac9e75dac
commit f5487b2185
2 changed files with 8 additions and 12 deletions
@@ -6,6 +6,12 @@ from lnbits.decorators import check_user_exists, validate_uuids
 from . import copilot_ext
 from .crud import get_copilot

+@copilot_ext.websocket('/ws')
+async def ws():
+    while True:
+        data = await websocket.receive()
+        await websocket.send(f"echo {data}")
+        
@copilot_ext.route("/")
@validate_uuids(["usr"], required=True)
@check_user_exists()
@@ -25,10 +31,4 @@ async def panel(copilot_id):
    copilot = await get_copilot(copilot_id) or abort(
        HTTPStatus.NOT_FOUND, "Copilot link does not exist."
    )
-    return await render_template("copilot/panel.html", copilot=copilot)
-
-@copilot_ext.websocket('/ws/<copilot_id>')
-async def ws():
-    while True:
-        data = await websocket.receive()
-        await websocket.send(f"echo {data}")
+    return await render_template("copilot/panel.html", copilot=copilot)