diff --git a/lerobot/scripts/find_motors_bus_port.py b/lerobot/scripts/find_motors_bus_port.py index 51ef6d41c..67b92ad7d 100644 --- a/lerobot/scripts/find_motors_bus_port.py +++ b/lerobot/scripts/find_motors_bus_port.py @@ -1,30 +1,36 @@ +import os import time from pathlib import Path +from serial.tools import list_ports # Part of pyserial library + def find_available_ports(): - ports = [] - for path in Path("/dev").glob("tty*"): - ports.append(str(path)) + if os.name == "nt": # Windows + # List COM ports using pyserial + ports = [port.device for port in list_ports.comports()] + else: # Linux/macOS + # List /dev/tty* ports for Unix-based systems + ports = [str(path) for path in Path("/dev").glob("tty*")] return ports def find_port(): print("Finding all available ports for the MotorsBus.") ports_before = find_available_ports() - print(ports_before) + print("Ports before disconnecting:", ports_before) - print("Remove the usb cable from your MotorsBus and press Enter when done.") - input() + print("Remove the USB cable from your MotorsBus and press Enter when done.") + input() # Wait for user to disconnect the device - time.sleep(0.5) + time.sleep(0.5) # Allow some time for port to be released ports_after = find_available_ports() ports_diff = list(set(ports_before) - set(ports_after)) if len(ports_diff) == 1: port = ports_diff[0] print(f"The port of this MotorsBus is '{port}'") - print("Reconnect the usb cable.") + print("Reconnect the USB cable.") elif len(ports_diff) == 0: raise OSError(f"Could not detect the port. No difference was found ({ports_diff}).") else: @@ -32,5 +38,5 @@ def find_port(): if __name__ == "__main__": - # Helper to find the usb port associated to all your MotorsBus. + # Helper to find the USB port associated with your MotorsBus. find_port()