From 8a54620d570533a4c1edd92806eada946e39a6d7 Mon Sep 17 00:00:00 2001
From: AURUMVORXX <aurumvorax.enthroned@gmail.com>
Date: Tue, 20 May 2025 16:52:16 +0300
Subject: [PATCH] fix: Handling ConnectionClosedError

---
 src/pyg2o/server.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/pyg2o/server.py b/src/pyg2o/server.py
index fae57ff..e463a80 100644
--- 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')