fix get_basic_input() and get_digital_input()

elephantrobotics · Nov 15, 2021 · 1ccd1f8 · 1ccd1f8
1 parent 21afccc
commit 1ccd1f8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pymycobot/common.py b/pymycobot/common.py
@@ -145,12 +145,17 @@ def _process_received(self, data, genre):
                     break
         else:
             return []
-
+        unique_data = [ProtocolCode.GET_BASIC_INPUT,
+                       ProtocolCode.GET_DIGITAL_INPUT]
         # compare send header and received header
         cmd_id = data[idx + 3]
         if cmd_id != genre:
             return []
-        data_pos = idx + 4
+        if cmd_id in unique_data:
+            data_pos = idx + 5
+            data_len -= 1
+        else:
+            data_pos = idx + 4
         valid_data = data[data_pos: data_pos + data_len]
 
         # process valid data