From 5088495a25673b5b642c0f3e74b0c198810f6238 Mon Sep 17 00:00:00 2001 From: weiquan Date: Fri, 1 Dec 2023 11:47:46 +0800 Subject: [PATCH] fix bug --- pymycobot/common.py | 10 ++++++++-- pymycobot/mybuddysocket.py | 5 +++-- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/pymycobot/common.py b/pymycobot/common.py index 298d4c4..5983a33 100644 --- a/pymycobot/common.py +++ b/pymycobot/common.py @@ -467,7 +467,9 @@ def write(self, command, method=None): if method == "socket": log_command = [] for i in command: - if isinstance(i, str): + if isinstance(command, str): + log_command.append(command) + elif isinstance(i, str): log_command.append(i) else: log_command.append(hex(i)) @@ -477,7 +479,11 @@ def write(self, command, method=None): if py_version == 2: self.sock.sendall("".join([chr(b) for b in command])) else: - self.sock.sendall(bytes(command)) + if isinstance(command, str): + command = command.encode() + else: + command = bytes(command) + self.sock.sendall(command) else: self._serial_port.reset_input_buffer() self.log.debug("_write: {}".format([hex(i) for i in command])) diff --git a/pymycobot/mybuddysocket.py b/pymycobot/mybuddysocket.py index b3ea99c..8407c92 100644 --- a/pymycobot/mybuddysocket.py +++ b/pymycobot/mybuddysocket.py @@ -93,8 +93,9 @@ def _mesg(self, genre, *args, **kwargs): """ real_command, has_reply = super( MyBuddySocket, self)._mesg(genre, *args, **kwargs) - data = self._write(self._flatten(real_command), "socket") - if data: + self._write(self._flatten(real_command), "socket") + if has_reply: + data = self._read(genre, 'socket') res = self._process_received(data, genre, arm=12) if genre in [ ProtocolCode.ROBOT_VERSION,