Ignore ConnectionError in disconnect()

Fixes #8
NitorCreations · Dec 13, 2024 · cbc0342 · cbc0342
1 parent 3a388a6
commit cbc0342
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/extron/extron.py b/custom_components/extron/extron.py
@@ -69,8 +69,13 @@ async def connect(self):
 
     async def disconnect(self):
         self._connected = False
-        self._writer.close()
-        await self._writer.wait_closed()
+
+        # Ignore potential connection errors here, we're about to disconnect after all
+        try:
+            self._writer.close()
+            await self._writer.wait_closed()
+        except ConnectionError:
+            pass
 
     async def reconnect(self):
         await self.disconnect()