fix: formatting (python)

Author: StellaLupus

tools/serial_parser.py

@@ -1,26 +1,26 @@
-import enum
 import serial
 import serial.tools
 import serial.tools.list_ports
 
 
-def substring_after(str: str, searchStr: str):
-    return str[str.index(searchStr) + len(searchStr):]
+def substring_after(str: str, search_str: str):
+    return str[str.index(search_str) + len(search_str):]
 
 
-def getPort():
+def get_port():
     ports = sorted(serial.tools.list_ports.comports())
     print("Available ports to listing:")
-    for id, portInfo in enumerate(ports):
-        print("{}. - {}: {} [{}]".format(id, portInfo.device,
-                                         portInfo.description, portInfo.hwid))
+    for id, port_info in enumerate(ports):
+        print("{}. - {}: {} [{}]".format(id, port_info.device,
+                                         port_info.description,
+                                         port_info.hwid))
     print("Select port: ", end="")
-    selectId = int(input())
-    if selectId < len(ports) and selectId >= 0:
-        return ports[selectId]
+    select_id = int(input())
+    if select_id < len(ports) and select_id >= 0:
+        return ports[select_id]
     else:
         print("Unrecognized port number")
-        return getPort()
+        return get_port()
 
 
 zero = 756
@@ -29,7 +29,7 @@ def getPort():
 interv = 0.2
 
 
-def getBitFromInterv(value: int):
+def get_bit_from_interv(value: int):
     if value > zero - zero * interv and value < zero + zero * interv:
         return 0
     elif value > one - one * interv and value < one + one * interv:
@@ -38,57 +38,58 @@ def getBitFromInterv(value: int):
         return 2
 
 
-def bitListToInt(bitlist):
+def bit_list_to_int(bit_list):
     out = 0
-    for bit in bitlist:
+    for bit in bit_list:
         out = (out << 1) | bit
     return out
 
 
 def main():
-    port = getPort()
+    port = get_port()
     print("Selected port:" + port.device)
 
     ser = serial.Serial(
         port=port.device,
         baudrate=115200,
     )
 
-    zSum = 0
-    zCnt = 0
-    oSum = 0
-    oCnt = 0
-    sSum = 0
-    sCnt = 0
+    zero_sum = 0
+    zero_count = 0
+    one_sum = 0
+    one_cnt = 0
+    space_sum = 0
+    space_cnt = 0
 
     while True:
         try:
-            dataStr = ser.readline().decode()
+            data_str = ser.readline().decode()
         except:
             continue
-        # print(dataStr)
-        if "uint16_t rawData" in dataStr:
-            dataStrArray = str(dataStr[dataStr.index("{") +
-                                       1:dataStr.index("}")]).split(",")
-            data = [int(i.strip()) for i in dataStrArray]
+        if "uint16_t rawData" in data_str:
+            data_str_array = str(data_str[data_str.index("{") +
+                                          1:data_str.index("}")]).split(",")
+            data = [int(i.strip()) for i in data_str_array]
             data = data[2:]
-            clearData = [i for idi, i in enumerate(data) if idi % 2 == 1]
-            bitData = [getBitFromInterv(i) for i in data]
-            clearBitData = [getBitFromInterv(i) for i in clearData]
+            clear_data = [i for idi, i in enumerate(data) if idi % 2 == 1]
+            bit_data = [get_bit_from_interv(i) for i in data]
+            clear_bit_data = [get_bit_from_interv(i) for i in clear_data]
 
             for idd, d in enumerate(data):
                 if idd % 2 == 0:
-                    sSum += d
-                    sCnt += 1
+                    space_sum += d
+                    space_cnt += 1
                 else:
-                    if getBitFromInterv(d) == 0:
-                        zSum += d
-                        zCnt += 1
-                    elif getBitFromInterv(d) == 1:
-                        oSum += d
-                        oCnt += 1
-
-            print("".join([str(i) for i in clearBitData]))
+                    if get_bit_from_interv(d) == 0:
+                        zero_sum += d
+                        zero_count += 1
+                    elif get_bit_from_interv(d) == 1:
+                        one_sum += d
+                        one_cnt += 1
+
+            print("Bit data = " + "".join([str(i) for i in bit_data]))
+            print("Clear bit data = " +
+                  "".join([str(i) for i in clear_bit_data]))
 
 
 if __name__ == "__main__":