From e7d5f2bdfcaf99718027e22b44083fae54f7aab8 Mon Sep 17 00:00:00 2001
From: Atharv Nagavarapu <30nagava@elmbrookstudents.org>
Date: Fri, 9 Jan 2026 19:20:14 +0000
Subject: [PATCH] Update diagnostics/FullDiagnostics.py

---
 diagnostics/FullDiagnostics.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/diagnostics/FullDiagnostics.py b/diagnostics/FullDiagnostics.py
index 2a55560..12d5adb 100644
--- a/diagnostics/FullDiagnostics.py
+++ b/diagnostics/FullDiagnostics.py
@@ -7,9 +7,11 @@ HUB = PrimeHub()
 from BatteryDiagnostics import BatteryDiagnostics
 from MotorDiagnostics import MotorDiagnostics
 from ColorSensorDiagnostics import ColorSensorDiagnostics
+from DriveBaseDiagnostics import DriveBaseDiagnostics
 battery = BatteryDiagnostics(HUB)
 motor = MotorDiagnostics(HUB, Motor)
 colorsensor = ColorSensorDiagnostics(HUB, ColorSensor)
+drivebase = DriveBaseDiagnostics(HUB, Motor, DriveBase)
 CLEARCONFIRM = input("Clear the console before proceeding? Y/N (default: yes): ")
 if(CLEARCONFIRM == "Y" or CLEARCONFIRM == "y" or CLEARCONFIRM == "yes" or CLEARCONFIRM == ""):
     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")
@@ -56,5 +58,9 @@ while True:
         print("---------------------COLOR SENSOR DIAGNOSTICS---------------------")
         colorsensor.printAll()
         print("Color sensor diagnostics completed.")
+    elif choice == "db":
+        print("----------------------DRIVE BASE DIAGNOSTICS----------------------")
+        drivebase.printAll()
+        print("Drivebase diagnostics completed.")
     else:
         print("Invalid choice. Please enter 'b', 'm', or 'q'.")
\ No newline at end of file