sonarcloud

ValvePython · Sep 8, 2024 · 694491c · 694491c
1 parent 5ce82c2
commit 694491c
Showing 1 changed file with 25 additions and 23 deletions.
diff --git a/steam/core/connection.py b/steam/core/connection.py
@@ -234,29 +234,31 @@ def _reader_loop(self):
 
                 logger.debug("Received {} bytes".format(len(data)))
                 self.ws.receive_data(data)
-
-                for event in self.ws.events():
-                    if isinstance(event, wsevents.AcceptConnection):
-                        logger.debug("WebSocket negotiation complete. Connected.")
-                        self.event_connected.set()
-                    elif isinstance(event, wsevents.RejectConnection):
-                        logger.debug("WebSocket connection was rejected. That's probably not good.")
-                    elif isinstance(event, wsevents.TextMessage):
-                        logger.debug("Received websocket text message of length: {}".format(len(event.data)))
-                    elif isinstance(event, wsevents.BytesMessage):
-                        logger.debug("Received websocket bytes message of length: {}".format(len(event.data)))
-                        self.recv_queue.put(event.data)
-                    elif isinstance(event, wsevents.Pong):
-                        logger.debug("Received pong: {}".format(repr(event.payload)))
-                    elif isinstance(event, wsevents.CloseConnection):
-                        logger.debug('Connection closed: code={} reason={}'.format(
-                            event.code, event.reason
-                        ))
-                        if self.ws.state == ConnectionState.REMOTE_CLOSING:
-                            self.socket.send(self.ws.send(event.response()))
-                        self.event_wsdisconnected.set()
-                    else:
-                        raise TypeError("Do not know how to handle event: {}".format((event)))
+                self._handle_events()
+
+    def _handle_events(self):
+        for event in self.ws.events():
+            if isinstance(event, wsevents.AcceptConnection):
+                logger.debug("WebSocket negotiation complete. Connected.")
+                self.event_connected.set()
+            elif isinstance(event, wsevents.RejectConnection):
+                logger.debug("WebSocket connection was rejected. That's probably not good.")
+            elif isinstance(event, wsevents.TextMessage):
+               logger.debug("Received websocket text message of length: {}".format(len(event.data)))
+            elif isinstance(event, wsevents.BytesMessage):
+                logger.debug("Received websocket bytes message of length: {}".format(len(event.data)))
+                self.recv_queue.put(event.data)
+            elif isinstance(event, wsevents.Pong):
+                logger.debug("Received pong: {}".format(repr(event.payload)))
+            elif isinstance(event, wsevents.CloseConnection):
+                logger.debug('Connection closed: code={} reason={}'.format(
+                    event.code, event.reason
+                ))
+                if self.ws.state == ConnectionState.REMOTE_CLOSING:
+                    self.socket.send(self.ws.send(event.response()))
+                self.event_wsdisconnected.set()
+            else:
+                raise TypeError("Do not know how to handle event: {}".format((event)))
 
     def disconnect(self):
         self.event_wsdisconnected.clear()