improved input request

pus_tc/devs/ploc_supervisor.py

@@ -32,6 +32,8 @@ MANUAL_INPUT = "1"
 update_file_dict = {
     MANUAL_INPUT: ["manual input", ""],
     "2": ["/mnt/sd0/ploc/supervisor/update.bin", "/mnt/sd0/ploc/supervisor/update.bin"],
+    "3": ["/mnt/sd0/ploc/supervisor/update-large.bin", "/mnt/sd0/ploc/supervisor/update-large.bin"],
+    "4": ["/mnt/sd0/ploc/supervisor/update-small.bin", "/mnt/sd0/ploc/supervisor/update-small.bin"],
 }
 
 event_buffer_path_dict = {
@@ -161,7 +163,7 @@ def pack_ploc_supv_commands(
         tc_queue.appendleft(
             (QueueCommands.PRINT, "PLOC Supervisor: Set max restart tries: ")
         )
-        restart_tries = int(input("Set maximum restart tries:"))
+        restart_tries = int(input("Set maximum restart tries: "))
         command = (
             object_id
             + struct.pack("!I", SupvActionIds.SET_MAX_RESTART_TRIES)
@@ -581,7 +583,7 @@ def get_latchup_id() -> int:
         key_string = key.ljust(key_column_width)
         description_string = latchup_id_dict[key].ljust(description_column_width)
         print(f"{key_string} | {description_string}")
-    return int(input("Specify latchup ID:"))
+    return int(input("Specify latchup ID: "))
 
 
 def pack_set_alert_limit_cmd(object_id: bytearray) -> bytearray:
@@ -631,7 +633,7 @@ def pack_set_adc_enabled_channels_cmd(object_id: bytearray) -> bytearray:
     @brief  This function packs the command to enable or disable channels of the ADC.
     @param object_id    The object id of the PLOC supervisor handler.
     """
-    ch = int(input("Specify ch:"))
+    ch = int(input("Specify ch: 0x"), 16)
     command = bytearray()
     command = object_id + struct.pack("!I", SupvActionIds.SET_ADC_ENABLED_CHANNELS)
     command = command + struct.pack("!H", ch)
@@ -649,7 +651,7 @@ def pack_set_adc_window_and_stride_cmd(object_id: bytearray) -> bytearray:
 
 
 def pack_set_adc_threshold_cmd(object_id: bytearray) -> bytearray:
-    threshold = int(input("Specify threshold:"))
+    threshold = int(input("Specify threshold: "))
     command = bytearray()
     command = object_id + struct.pack("!I", SupvActionIds.SET_ADC_THRESHOLD)
     command = command + struct.pack("!I", threshold)
@@ -767,10 +769,9 @@ def get_debug_verbosity() -> int:
 
 
 def pack_set_gpio_cmd(object_id: bytearray) -> bytearray:
-    port = int(input("Specify port: "))
-    pin = int(input("Specify pin: "))
-    val = int(input("Specify val: "))
-    command = bytearray()
+    port = int(input("Specify port : 0x"), 16)
+    pin = int(input("Specify pin: 0x"), 16)
+    val = int(input("Specify val: 0x"), 16)
     command = object_id + struct.pack("!I", SupvActionIds.SET_GPIO)
     command = command + struct.pack("!B", port)
     command = command + struct.pack("!B", pin)
@@ -779,9 +780,8 @@ def pack_set_gpio_cmd(object_id: bytearray) -> bytearray:
 
 
 def pack_read_gpio_cmd(object_id: bytearray) -> bytearray:
-    port = int(input("Specify port: "))
-    pin = int(input("Specify pin: "))
-    command = bytearray()
+    port = int(input("Specify port: 0x"), 16)
+    pin = int(input("Specify pin: 0x"), 16)
     command = object_id + struct.pack("!I", SupvActionIds.READ_GPIO)
     command = command + struct.pack("!B", port)
     command = command + struct.pack("!B", pin)