fix: dictionary changed size during iteration

lekoOwO · May 14, 2021 · 3f34011 · 3f34011
1 parent b8b6d25
commit 3f34011
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/index.py b/index.py
@@ -131,13 +131,17 @@ def get_channel_name_by_video_id(video_id):
         def clear_chat():
             utils.log(f" Running chat instance clearing task.")
             global chats
+            to_del = []
             for video_id in chats:
                 if chats[video_id].is_finished():
                     if const.CHAT_CALLBACK_AFTER_EXPIRY:
                         channel_name = get_channel_name_by_video_id(video_id)
                         chat_callback.callback(chats[video_id], channel_name=channel_name, video_id=video_id)
-                    chats.pop(video_id)
-                    utils.log(f" Chat instance {video_id} has been cleared.")
+                    to_del.append(video_id)
+                    utils.log(f" Chat instance {video_id} has been queued to be cleared.")
+            for video_id in to_del:
+                chats.pop(video_id)
+                utils.log(f" Chat instance {video_id} has been cleared.")
 
         chat_expiry_task = utils.RepeatedTimer(const.CHAT_TASK_CLEAR_INTERVAL, clear_chat)