Improve robustness of address changer script

Author: sebromero

examples/change_address.py

@@ -11,42 +11,63 @@
 from time import sleep
 from modulino import Modulino
 
-print()
-bus = None # Change this to the I2C bus you are using on 3rd party host boards
-devices = Modulino.available_devices(bus)
+def main():
+    print()
+    bus = None # Change this to the I2C bus you are using on 3rd party host boards
+    devices = Modulino.available_devices(bus)
 
-if len(devices) == 0:
-    print("No devices found on the bus. Try resetting the board.")
-    exit(1)
+    if len(devices) == 0:
+        print("No devices found on the bus. Try resetting the board.")
+        return
 
-print("The following devices were found on the bus:")
+    print("The following devices were found on the bus:")
 
-for index, device in enumerate(devices):
-    print(f"{index + 1}) {device.device_type} at {hex(device.address)}")
+    for index, device in enumerate(devices):
+        print(f"{index + 1}) {device.device_type} at {hex(device.address)}")
 
-choice = int(input("\nEnter the device number for which you want to change the address: "))
+    choice_is_valid = False
+    while not choice_is_valid:
+        try:
+            choice = int(input("\nEnter the device number for which you want to change the address: "))
+        except ValueError:
+            print("Invalid input. Please enter a valid device number.")
+            continue
+        
+        if choice < 1 or choice > len(devices):
+            print("Invalid choice. Please select a valid device number.")
+        else:
+            choice_is_valid = True
 
-if choice < 1 or choice > len(devices):
-    print("Invalid choice. Please select a valid device number.")
-    exit(1)
+    selected_device = devices[choice - 1]
 
-selected_device = devices[choice - 1]
-new_address = int(input("Enter the new address (hexadecimal or decimal): "), 0)
 
-if new_address < 0 or new_address > 127:
-    print("Invalid address. Address must be between 0 and 127")
-    exit(1)
+    new_address_is_valid = False
+    while not new_address_is_valid:
+        try:
+            new_address = int(input("Enter the new address (hexadecimal or decimal): "), 0)
+        except ValueError:
+            print("Invalid input. Please enter a valid hexadecimal (e.g., 0x2A) or decimal (e.g., 42) address.")
+            continue
 
-if new_address == 100:
-    print("The address 0x64 is reserved for bootloader mode. Please choose a different address.")
-    exit(1)
+        if new_address < 1 or new_address > 127:
+            print("Invalid address. Address must be between 1 and 127")
+        elif new_address == 100:
+            print("The address 0x64 (100) is reserved for bootloader mode. Please choose a different address.")
+        else:
+            new_address_is_valid = True
 
-print(f"Changing address of device at {hex(selected_device.address)} to {hex(new_address)}...")
-selected_device.change_address(new_address)
-sleep(1) # Give the device time to reset
+    print(f"Changing address of device at {hex(selected_device.address)} to {hex(new_address)}...")
+    selected_device.change_address(new_address)
+    sleep(1) # Give the device time to reset
 
-# Check if the address was successfully changed
-if selected_device.connected:
-    print(f"✅ Address changed successfully to {hex(new_address)}")
-else:
-    print("❌ Failed to change address. Please try again.")
+    # Check if the address was successfully changed
+    if selected_device.connected:
+        print(f"✅ Address changed successfully to {hex(new_address)}")
+    else:
+        print("❌ Failed to change address. Please try again.")
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        print("\Aborted by user")