status bar

Author: meowmeowahr

src/kevinbotlib_dashboard/app.py

@@ -3,7 +3,7 @@
 from typing import override
 
 from kevinbotlib.comm import KevinbotCommClient
-from PySide6.QtCore import QObject, QPointF, QRect, QRectF, QRegularExpression, QSettings, QSize, Qt, Signal
+from PySide6.QtCore import QObject, QPointF, QRect, QRectF, QRegularExpression, QSettings, QSize, Qt, Signal, QTimer
 from PySide6.QtGui import QAction, QBrush, QCloseEvent, QColor, QPainter, QPen, QRegularExpressionValidator
 from PySide6.QtWidgets import (
     QDialog,
@@ -21,6 +21,7 @@
     QStyleOptionGraphicsItem,
     QVBoxLayout,
     QWidget,
+    QLabel,
 )
 
 from kevinbotlib_dashboard.grid_theme import Themes
@@ -467,6 +468,8 @@ def __init__(self):
         self.client = KevinbotCommClient(
             host=self.settings.value("ip", "10.0.0.2", str),  # type: ignore
             port=self.settings.value("port", 8765, int),  # type: ignore
+            on_disconnect=self.on_disconnect,
+            on_connect=self.on_connect,
         )
         self.client.connect()
 
@@ -477,6 +480,17 @@ def __init__(self):
 
         self.settings_action = self.edit_menu.addAction("Settings", self.open_settings)
 
+        self.status = self.statusBar()
+
+        self.connection_status = QLabel("Robot Disconnected")
+        self.status.addWidget(self.connection_status)
+
+        self.ip_status = QLabel(str(self.settings.value("ip", "10.0.0.2", str)), alignment=Qt.AlignmentFlag.AlignCenter)
+        self.status.addWidget(self.ip_status, 1)
+
+        self.latency_status = QLabel("Latency: 0.00")
+        self.status.addPermanentWidget(self.latency_status)
+
         main_widget = QWidget()
         self.setCentralWidget(main_widget)
 
@@ -492,18 +506,36 @@ def __init__(self):
         layout.addWidget(self.graphics_view)
         layout.addWidget(palette)
 
+        self.latency_timer = QTimer()
+        self.latency_timer.setInterval(1000)
+        self.latency_timer.timeout.connect(self.update_latency)
+        self.latency_timer.start()
+
         self.controller = WidgetGridController(self.graphics_view)
         self.controller.load(self.item_loader, self.settings.value("layout", [], type=list))  # type: ignore
 
         self.settings_window = SettingsWindow(self, self.settings)
         self.settings_window.on_applied.connect(self.refresh_settings)
 
+    def update_latency(self):
+        if self.client.websocket:
+            self.latency_status.setText(f"Latency: {self.client.websocket.latency:.2f}ms")
+
+    def on_connect(self):
+        self.connection_status.setText("Robot Connected")
+
+    def on_disconnect(self):
+        self.connection_status.setText("Robot Disconnected")
+
+
     def refresh_settings(self):
         self.settings.setValue("ip", self.settings_window.net_ip.text())
         self.settings.setValue("port", self.settings_window.net_port.value())
         self.client.host = self.settings.value("ip", "10.0.0.2", str)  # type: ignore
         self.client.port = self.settings.value("port", 8765, int) # type: ignore
 
+        self.ip_status.setText(str(self.settings.value("ip", "10.0.0.2", str)))
+
         self.settings.setValue("grid", self.settings_window.grid_size.value())
         self.settings.setValue("rows", self.settings_window.grid_rows.value())
         self.settings.setValue("cols", self.settings_window.grid_cols.value())