fix(cli): removed hard-coded signal

bec_widgets/cli/client_utils.py

@@ -9,6 +9,7 @@ from typing import TYPE_CHECKING
 
 from bec_lib import MessageEndpoints, messages
 from bec_lib.connector import MessageObject
+from bec_lib.device import DeviceBase
 from qtpy.QtCore import QCoreApplication
 
 import bec_widgets.cli.client as client
@@ -36,6 +37,19 @@ def rpc_call(func):
     return wrapper
 
 
+def get_selected_device(monitored_devices, selected_device):
+    """
+    Get the selected device for the plot. If no device is selected, the first
+    device in the monitored devices list is selected.
+    """
+    if selected_device:
+        return selected_device
+    if len(monitored_devices) > 0:
+        sel_device = monitored_devices[0]
+        return sel_device
+    return None
+
+
 def update_script(figure: BECFigure, msg):
     """
     Update the script with the given data.
@@ -46,10 +60,15 @@ def update_script(figure: BECFigure, msg):
     scan_number = info.get("scan_number", 0)
     scan_name = info.get("scan_name", "Unknown")
     scan_report_devices = info.get("scan_report_devices", [])
+    monitored_devices = info.get("readout_priority", {}).get("monitored", [])
+    monitored_devices = [dev for dev in monitored_devices if dev not in scan_report_devices]
 
     if scan_name == "line_scan" and scan_report_devices:
         dev_x = scan_report_devices[0]
-        dev_y = figure._selected_device
+        dev_y = get_selected_device(monitored_devices, figure.selected_device)
+        print(f"Selected device: {dev_y}")
+        if not dev_y:
+            return
         figure.clear_all()
         plt = figure.plot(dev_x, dev_y)
         plt.set(title=f"Scan {scan_number}", x_label=dev_x, y_label=dev_y)
@@ -62,7 +81,9 @@ def update_script(figure: BECFigure, msg):
         plt.set(title=f"Scan {scan_number}", x_label=dev_x, y_label=dev_y)
     elif scan_report_devices:
         dev_x = scan_report_devices[0]
-        dev_y = figure._selected_device
+        dev_y = get_selected_device(monitored_devices, figure.selected_device)
+        if not dev_y:
+            return
         figure.clear_all()
         plt = figure.plot(dev_x, dev_y, label=f"Scan {scan_number}")
         plt.set(title=f"Scan {scan_number}", x_label=dev_x, y_label=dev_y)
@@ -74,7 +95,23 @@ class BECFigureClientMixin:
         self._process = None
         self.update_script = update_script
         self._target_endpoint = MessageEndpoints.scan_status()
-        self._selected_device = "bpm4i"
+        self._selected_device = None
+
+    @property
+    def selected_device(self):
+        """
+        Selected device for the plot.
+        """
+        return self._selected_device
+
+    @selected_device.setter
+    def selected_device(self, device: str | DeviceBase):
+        if isinstance(device, DeviceBase):
+            self._selected_device = device.name
+        elif isinstance(device, str):
+            self._selected_device = device
+        else:
+            raise ValueError("Device must be a string or a device object")
 
     def _start_update_script(self) -> None:
         self._client.connector.register(