56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
# Copyright 2024 The HuggingFace Inc. team. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
|
|
from serial.tools import list_ports # Part of pyserial library
|
|
|
|
|
|
def find_available_ports():
|
|
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 disconnecting:", ports_before)
|
|
|
|
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) # 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.")
|
|
elif len(ports_diff) == 0:
|
|
raise OSError(f"Could not detect the port. No difference was found ({ports_diff}).")
|
|
else:
|
|
raise OSError(f"Could not detect the port. More than one port was found ({ports_diff}).")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Helper to find the USB port associated with your MotorsBus.
|
|
find_port()
|