Fix is_alive (#142)

xchwarze · Nov 28, 2024 · 51081bc · 51081bc
1 parent 8ee4242
commit 51081bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/samsungtvws/async_connection.py b/samsungtvws/async_connection.py
@@ -24,6 +24,7 @@
 
 from websockets.asyncio.client import ClientConnection, connect
 from websockets.exceptions import ConnectionClosed
+from websockets.protocol import State
 
 from . import connection, exceptions, helper
 from .command import SamsungTVCommand, SamsungTVSleepCommand
@@ -174,4 +175,4 @@ async def _send_command(
         await asyncio.sleep(delay)
 
     def is_alive(self) -> bool:
-        return self.connection is not None and not self.connection.closed
+        return self.connection is not None and self.connection.state is not State.CLOSED
diff --git a/samsungtvws/encrypted/remote.py b/samsungtvws/encrypted/remote.py
@@ -10,6 +10,7 @@
 import aiohttp
 from websockets.asyncio.client import ClientConnection, connect
 from websockets.exceptions import ConnectionClosed
+from websockets.protocol import State
 
 from ..exceptions import ConnectionFailure
 from .command import SamsungTVEncryptedCommand
@@ -183,4 +184,7 @@ async def close(self) -> None:
         LOGGER.debug("Connection closed")
 
     def is_alive(self) -> bool:
-        return self._connection is not None and not self._connection.closed
+        return (
+            self._connection is not None
+            and not self._connection.state is not State.CLOSED
+        )