Remove all the calls to websocket.state

replit · Apr 23, 2024 · e94cace · e94cace
1 parent 0294ed2
commit e94cace
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/replit_river/client_transport.py b/replit_river/client_transport.py
@@ -219,9 +219,7 @@ async def _establish_handshake(
         logging.debug("river client waiting for handshake response")
         while True:
             try:
-                logging.debug(
-                    f"websocket while waiting for response : {websocket.id} {websocket.state}"
-                )
+                logging.debug(f"websocket while waiting for response : {websocket.id}")
                 data = await websocket.recv()
             except ConnectionClosed:
                 logging.debug(

diff --git a/replit_river/server.py b/replit_river/server.py
@@ -36,7 +36,6 @@ def add_rpc_handlers(
     async def serve(self, websocket: WebSocketServerProtocol) -> None:
         logging.debug(
             f"River server started establishing session with ws: {websocket.id}"
-            f" {websocket.state}"
         )
         try:
             session = await self._transport.handshake_to_get_session(websocket)

diff --git a/replit_river/session.py b/replit_river/session.py
@@ -162,7 +162,7 @@ async def _handle_messages_from_ws(
     ) -> None:
         logging.debug(
             f'{"server" if self._is_server else "client"} start handling messages from'
-            f" ws {websocket.id}, state {websocket.state}"
+            f" ws {websocket.id}"
         )
         try:
             async for message in websocket: