From 657810edc35217ec08ef4dcc352cf3a9a35c895d Mon Sep 17 00:00:00 2001
From: Kay0u <pierre@kayou.io>
Date: Thu, 17 Jun 2021 16:13:10 +0200
Subject: [PATCH] revert the way we remove websockets in a list

---
 run.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/run.py b/run.py
index 937ca69..08aa4ad 100644
--- a/run.py
+++ b/run.py
@@ -470,13 +470,16 @@ async def broadcast(message, channels):
 
     for channel in channels:
         ws_list = subscriptions[channel]
+        dead_ws = []
 
         for ws in ws_list:
             try:
                 await ws.send(json.dumps(message, default=datetime_to_epoch_json_converter))
             except ConnectionClosed:
-                ws_list.remove(ws)
+                dead_ws.append(ws)
 
+        for to_remove in dead_ws:
+            ws_list.remove(to_remove)
 
 def subscribe(ws, channel):
     subscriptions[channel].append(ws)