wip

Author: d-perl

bec_widgets/widgets/control/procedure_control/procedure_logs.py

@@ -1,7 +1,8 @@
 from bec_lib.endpoints import MessageEndpoints
 from bec_lib.logger import bec_logger
+from bec_qthemes import material_icon
 from bec_server.scan_server.procedures.helper import FrontendProcedureHelper
-from PySide6.QtWidgets import QLabel
+from PySide6.QtWidgets import QComboBox, QHBoxLayout, QLabel, QToolButton
 from qtpy.QtGui import QFont
 from qtpy.QtWidgets import QSizePolicy, QTextEdit, QVBoxLayout, QWidget
 
@@ -19,11 +20,38 @@ def __init__(self, parent=None, client=None, config=None, gui_id: str | None = N
         super().__init__(parent=parent, client=client, config=config, gui_id=gui_id, **kwargs)
         self._conn = self.bec_dispatcher.client.connector
         self._queue: str | None = None
+        self._available_streams = self._get_available_streams()
         self._setup_ui()
 
+    def _get_available_streams(self):
+        # TODO get from redis
+        return ["primary"]
+
+    def _update_selection_box(self):
+        self._selection_box.clear()
+        self._selection_box.addItems(self._available_streams)
+
     def _setup_ui(self):
         self._layout = QVBoxLayout()
         self.setLayout(self._layout)
+        self._setup_tools()
+        self._setup_display()
+
+    def _setup_tools(self):
+        self.tools = QWidget(self)
+        self._tools_layout = QHBoxLayout()
+        self.tools.setLayout(self._tools_layout)
+        self._selection_box = QComboBox()
+        self._update_selection_box()
+        self._selection_box.currentTextChanged.connect(self.set_queue)
+        self._refresh_button = QToolButton()
+        self._refresh_button.setIcon(material_icon("refresh", convert_to_pixmap=False))
+        self._tools_layout.addWidget(QLabel("Select logs stream: "))
+        self._tools_layout.addWidget(self._selection_box)
+        self._tools_layout.addWidget(self._refresh_button)
+        self._layout.addWidget(self.tools)
+
+    def _setup_display(self):
         self.widget = QTextEdit(lineWrapMode=QTextEdit.LineWrapMode.NoWrap, readOnly=True)
         font = QFont("Courier New")
         font.setStyleHint(QFont.StyleHint.Monospace)