prevent ws endpoint from running without valid token

2022-11-24 13:41:10 -07:00
parent 101dec461e
commit fc59b02255
2 changed files with 6 additions and 7 deletions
--- a/freqtrade/rpc/api_server/api_ws.py
+++ b/freqtrade/rpc/api_server/api_ws.py
@@ -97,8 +97,9 @@ async def message_endpoint(
    rpc: RPC = Depends(get_rpc),
    message_stream: MessageStream = Depends(get_message_stream)
 ):
-    async with create_channel(websocket) as channel:
-        await channel.run_channel_tasks(
-            channel_reader(channel, rpc),
-            channel_broadcaster(channel, message_stream)
-        )
+    if token:
+        async with create_channel(websocket) as channel:
+            await channel.run_channel_tasks(
+                channel_reader(channel, rpc),
+                channel_broadcaster(channel, message_stream)
+            )