From 6daf583b4b7dea68e41d2d9acb9e7a9098c1ca1e Mon Sep 17 00:00:00 2001 From: shortcutme Date: Thu, 29 Mar 2018 02:45:33 +0200 Subject: [PATCH] Only log detailed message type error with debug_socket argument --- src/Connection/Connection.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/Connection/Connection.py b/src/Connection/Connection.py index 5e465640..88f3ac4d 100644 --- a/src/Connection/Connection.py +++ b/src/Connection/Connection.py @@ -141,7 +141,6 @@ class Connection(object): self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.sock.connect((self.ip, int(self.port))) - # Detect protocol self.send({"cmd": "handshake", "req_id": 0, "params": self.getHandshakeInfo()}) event_connected = self.event_connected @@ -206,10 +205,9 @@ class Connection(object): except StopIteration: break if not type(message) is dict: - raise Exception( - "Invalid message type: %s, content: %r, buffer: %r" % - (type(message), message, buff[0:16]) - ) + if config.debug_socket: + self.log("Invalid message type: %s, content: %r, buffer: %r" % (type(message), message, buff[0:16])) + raise Exception("Invalid message type: %s" % type(message)) # Stats self.incomplete_buff_recv = 0