fix bug

elephantrobotics · Dec 1, 2023 · 5088495 · 5088495
1 parent 79bcb94
commit 5088495
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git 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
@@ -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,