fix: Handling ConnectionClosedError

2025-05-20 16:52:16 +03:00
parent 78b44897aa
commit 8a54620d57
1 changed files with 6 additions and 0 deletions
--- a/src/pyg2o/server.py
+++ b/src/pyg2o/server.py
@@ -154,6 +154,12 @@ class PythonWebsocketServer:
                except Exception as e:
                    self.logger.exception(f'[PyG2O] Exception: {e}')
                    continue
+        except websockets.exceptions.ConnectionClosedError:
+            if (not self.silent):
+                self.logger.info('Client disconnected')
+            self.is_connected = None
+            self._connected_socket = None
+            asyncio.create_task(callEvent('onWebsocketDisconnect', **{}))
        finally:
            if (not self.silent):
                self.logger.info('Client disconnected')