Handle CancelledError in async pubsub managers (Fixes #750)

miguelgrinberg · Jul 13, 2021 · a813bde · a813bde
1 parent fb3ac95
commit a813bde
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/socketio/asyncio_pubsub_manager.py b/src/socketio/asyncio_pubsub_manager.py
@@ -1,3 +1,4 @@
+import asyncio
 from functools import partial
 import uuid
 
@@ -148,6 +149,8 @@ async def _thread(self):
         while True:
             try:
                 message = await self._listen()
+            except asyncio.CancelledError:  # pragma: no cover
+                break
             except:
                 import traceback
                 traceback.print_exc()