35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from pybricks.hubs import PrimeHub
|
|
from pybricks.pupdevices import Motor, ColorSensor, UltrasonicSensor, ForceSensor
|
|
from pybricks.parameters import Button, Color, Direction, Port, Side, Stop
|
|
from pybricks.robotics import DriveBase
|
|
from pybricks.tools import wait, StopWatch
|
|
hub = PrimeHub()
|
|
from BatteryDiagnostics import BatteryDiagnostics
|
|
battery = BatteryDiagnostics()
|
|
clearConfirmation = input("Do you want to clear the console before proceeding? Y/N (default: yes): ")
|
|
if(clearConfirmation == "Y" or clearConfirmation == "y" or clearConfirmation == "yes" or clearConfirmation == ""):
|
|
print("Clearing console... \n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n")
|
|
|
|
while True:
|
|
print("\nWhat diagnostic do you want to perform?")
|
|
print("Enter 'b' for Battery diagnostics")
|
|
print("Enter 'm' for Motor diagnostics")
|
|
print("Enter 'q' to Quit")
|
|
|
|
choice = input("Your choice: ").strip().lower()
|
|
|
|
if choice == "b":
|
|
print("-----------------------BATTERY DIAGNOSTICS-----------------------")
|
|
battery.printAll()
|
|
|
|
elif choice == "m":
|
|
print("------------------------MOTOR DIAGNOSTICS------------------------")
|
|
# motor.printAll() # call your motor diagnostics here
|
|
print("Motor diagnostics would run here.")
|
|
|
|
elif choice == "q":
|
|
print("Diagnostics completed successfully. Exiting with code 0. Good luck in the robot game!")
|
|
break
|
|
|
|
else:
|
|
print("Invalid choice. Please enter 'b', 'm', or 'q'.") |