refactor: use identity checks for WebSocket message types (#56)

Harmony-Libs · Jan 24, 2025 · f1e9192 · f1e9192
1 parent 2188bfa
commit f1e9192
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/aioharmony/hubconnector_websocket.py b/src/aioharmony/hubconnector_websocket.py
@@ -370,7 +370,7 @@ async def _listener(self, websocket=None) -> None:
                     "%s: Response payload: %s", self._ip_address, response.data
                 )
 
-                if response.type == aiohttp.WSMsgType.CLOSED:
+                if response.type is aiohttp.WSMsgType.CLOSED:
                     close_code = (
                         ""
                         if response.data is None
@@ -383,11 +383,11 @@ async def _listener(self, websocket=None) -> None:
                     have_connection = False
                     break
 
-                if response.type == aiohttp.WSMsgType.ERROR:
+                if response.type is aiohttp.WSMsgType.ERROR:
                     _LOGGER.error("%s: Response error", self._ip_address)
                     continue
 
-                if response.type != aiohttp.WSMsgType.TEXT:
+                if response.type is not aiohttp.WSMsgType.TEXT:
                     continue
 
                 response_json = response.json()