bec-widgets 0.87.0__tar.gz → 0.87.1__tar.gz

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/CHANGELOG.md

@@ -1,5 +1,23 @@
 # CHANGELOG
 
+## v0.87.1 (2024-07-18)
+
+### Fix
+
+* fix(dock): added hasattr to cleanup method for widgets ([`d75c55b`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/d75c55b2b1ccf156fb789c7813f1c5bdf256f860))
+
+* fix: add missing close() call, ensure jupyter console client.shutdown() is called in closeEvent ([`e52ee26`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/e52ee2604cb35096f1bd833ca9516d8a34197d35))
+
+* fix: BECWidget checks if it is a widget, and implements closeEvent and cleanup ([`d64758f`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/d64758f268cad69e6a17bd52dc9913a6367d3cde))
+
+* fix: add exit handlers for BECConnection objects ([`6202d22`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/6202d224fe85c103a4c33bd8c255f18cfd027303))
+
+### Refactor
+
+* refactor: BECWidget is a mixin based on BECConnector, for each QWidget in BEC
+
+Handles closeEvent() and RPC registering/unregistering ([`c7feb69`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/c7feb6952d590b569f7b0cba3b019a9af0ce0c93))
+
 ## v0.87.0 (2024-07-17)
 
 ### Feature
@@ -118,12 +136,6 @@
 
 * fix(motor_map): bug where motors without limits were selected ([`c78cd89`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/c78cd898f203f950d7cb589eb5609feaa88062cf))
 
-### Refactor
-
-* refactor(setting_dialog): moved to qt_utils ([`3826bb3`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/3826bb3d9e870e85709b5b20ef09a4d22641280c))
-
-* refactor(toolbar): toolbar moved from widgets to qt_utils ([`7ffc06f`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/7ffc06f3c7ddd86a1681408a75221b9bbadb236b))
-
 ### Test
 
 * test(setting_dialog): tests added ([`74a249b`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/74a249bd065d01006cb532bfff2a9bfedb34b592))
@@ -131,15 +143,3 @@
 ### Unknown
 
 * tests(motor_map_widget): tests added ([`734f4c7`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/734f4c77507a1edafd477d81b5f7401d8e759be2))
-
-* feat(settings_dialog):apply button ([`2020953`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/2020953b933b6fcad61ecc770588d39518c26fdd))
-
-## v0.82.0 (2024-07-07)
-
-### Feature
-
-* feat(toggle): added angular component-like toggle ([`b9bff38`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/b9bff38b64b86f06b3bc047922ef9df0c7d32e71))
-
-### Refactor
-
-* refactor(device_input): DeviceComboBox and DeviceLineEdit moved to top layer of widgets ([`f048629`](https://gitlab.psi.ch/bec/bec_widgets/-/commit/f04862933f049030554086adef3ec9e1aebd3eda))

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bec_widgets
-Version: 0.87.0
+Version: 0.87.1
 Summary: BEC Widgets
 Project-URL: Bug Tracker, https://gitlab.psi.ch/bec/bec_widgets/issues
 Project-URL: Homepage, https://gitlab.psi.ch/bec/bec_widgets

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/examples/jupyter_console/jupyter_console_window.py

@@ -175,12 +175,11 @@ class JupyterConsoleWindow(QWidget):  # pragma: no cover:
 
     def closeEvent(self, event):
         """Override to handle things when main window is closed."""
-        self.dock.clear_all()
         self.dock.cleanup()
         self.dock.close()
-        self.figure.clear_all()
         self.figure.cleanup()
         self.figure.close()
+        self.console.close()
 
         super().closeEvent(event)
 

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/utils/bec_connector.py

@@ -11,10 +11,10 @@ from bec_lib.utils.import_utils import lazy_import_from
 from pydantic import BaseModel, Field, field_validator
 from qtpy.QtCore import QObject, QRunnable, QThreadPool, Signal
 from qtpy.QtCore import Slot as pyqtSlot
+from qtpy.QtWidgets import QApplication
 
 from bec_widgets.cli.rpc_register import RPCRegister
 from bec_widgets.qt_utils.error_popups import ErrorPopupUtility
-from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.yaml_dialog import load_yaml, load_yaml_gui, save_yaml, save_yaml_gui
 
 BECDispatcher = lazy_import_from("bec_widgets.utils.bec_dispatcher", ("BECDispatcher",))
@@ -65,16 +65,30 @@ class Worker(QRunnable):
         self.signals.completed.emit()
 
 
-class BECConnector(BECWidget):
-    """Connection mixin class for all BEC widgets, to handle BEC client and device manager"""
+class BECConnector:
+    """Connection mixin class to handle BEC client and device manager"""
 
     USER_ACCESS = ["_config_dict", "_get_all_rpc"]
+    EXIT_HANDLERS = {}
 
     def __init__(self, client=None, config: ConnectionConfig = None, gui_id: str = None):
         # BEC related connections
         self.bec_dispatcher = BECDispatcher(client=client)
         self.client = self.bec_dispatcher.client if client is None else client
 
+        if not self.client in BECConnector.EXIT_HANDLERS:
+            # register function to clean connections at exit;
+            # the function depends on BECClient, and BECDispatcher
+            @pyqtSlot()
+            def terminate(client=self.client, dispatcher=self.bec_dispatcher):
+                print("Disconnecting", repr(dispatcher))
+                dispatcher.disconnect_all()
+                print("Shutting down BEC Client", repr(client))
+                client.shutdown()
+
+            BECConnector.EXIT_HANDLERS[self.client] = terminate
+            QApplication.instance().aboutToQuit.connect(terminate)
+
         if config:
             self.config = config
             self.config.widget_class = self.__class__.__name__
@@ -92,6 +106,8 @@ class BECConnector(BECWidget):
             self.gui_id = self.config.gui_id
 
         # register widget to rpc register
+        # be careful: when registering, and the object is not a BECWidget,
+        # cleanup has to called manually since there is no 'closeEvent'
         self.rpc_register = RPCRegister()
         self.rpc_register.add_rpc(self)
 
@@ -284,18 +300,3 @@ class BECConnector(BECWidget):
             return self.config.model_dump()
         else:
             return self.config
-
-    def cleanup(self):
-        """Cleanup the widget."""
-        self.rpc_register.remove_rpc(self)
-        all_connections = self.rpc_register.list_all_connections()
-        if len(all_connections) == 0:
-            print("No more connections. Shutting down GUI BEC client.")
-            self.bec_dispatcher.disconnect_all()
-            self.client.shutdown()
-        if hasattr(super(), "cleanup"):
-            super().cleanup()
-
-    # def closeEvent(self, event):
-    #     self.cleanup()
-    #     super().closeEvent(event)

bec_widgets-0.87.1/bec_widgets/utils/bec_widget.py

@@ -0,0 +1,23 @@
+from qtpy.QtWidgets import QWidget
+
+from bec_widgets.utils.bec_connector import BECConnector, ConnectionConfig
+
+
+class BECWidget(BECConnector):
+    """Mixin class for all BEC widgets, to handle cleanup"""
+
+    def __init__(self, client=None, config: ConnectionConfig = None, gui_id: str = None):
+        if not isinstance(self, QWidget):
+            raise RuntimeError(f"{repr(self)} is not a subclass of QWidget")
+        super().__init__(client, config, gui_id)
+
+    def cleanup(self):
+        """Cleanup the widget."""
+        pass
+
+    def closeEvent(self, event):
+        self.rpc_register.remove_rpc(self)
+        try:
+            self.cleanup()
+        finally:
+            super().closeEvent(event)

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/base_classes/device_input_base.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from bec_widgets.utils import BECConnector, ConnectionConfig
+from bec_widgets.utils import ConnectionConfig
+from bec_widgets.utils.bec_widget import BECWidget
 
 
 class DeviceInputConfig(ConnectionConfig):
@@ -9,7 +10,7 @@ class DeviceInputConfig(ConnectionConfig):
     arg_name: str | None = None
 
 
-class DeviceInputBase(BECConnector):
+class DeviceInputBase(BECWidget):
     """
     Mixin class for device input widgets. This class provides methods to get the device list and device object based
     on the current text of the widget.
@@ -120,6 +121,3 @@ class DeviceInputBase(BECConnector):
         """
         if device not in self.get_device_list(self.config.device_filter):
             raise ValueError(f"Device {device} is not valid.")
-
-    def cleanup(self):
-        super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/bec_queue/bec_queue.py

@@ -2,10 +2,11 @@ from bec_lib.endpoints import MessageEndpoints
 from qtpy.QtCore import Qt, Slot
 from qtpy.QtWidgets import QHeaderView, QTableWidget, QTableWidgetItem, QWidget
 
-from bec_widgets.utils.bec_connector import BECConnector, ConnectionConfig
+from bec_widgets.utils.bec_connector import ConnectionConfig
+from bec_widgets.utils.bec_widget import BECWidget
 
 
-class BECQueue(BECConnector, QTableWidget):
+class BECQueue(BECWidget, QTableWidget):
     """
     Widget to display the BEC queue.
     """

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/bec_status_box/bec_status_box.py

@@ -13,7 +13,7 @@ from bec_lib.utils.import_utils import lazy_import_from
 from qtpy.QtCore import QObject, QTimer, Signal, Slot
 from qtpy.QtWidgets import QHBoxLayout, QTreeWidget, QTreeWidgetItem, QWidget
 
-from bec_widgets.utils.bec_connector import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import apply_theme
 from bec_widgets.widgets.bec_status_box.status_item import StatusItem
 
@@ -57,7 +57,7 @@ class BECServiceStatusMixin(QObject):
         self.services_update.emit(self.client._services_info, self.client._services_metric)
 
 
-class BECStatusBox(BECConnector, QWidget):
+class BECStatusBox(BECWidget, QWidget):
     """An autonomous widget to display the status of BEC services.
 
     Args:
@@ -290,15 +290,6 @@ class BECStatusBox(BECConnector, QWidget):
             if objects["item"] == item:
                 objects["widget"].show_popup()
 
-    def closeEvent(self, event):
-        """Upon closing the widget, clean up the BECStatusBox and the QWidget.
-
-        Args:
-            event: The close event.
-        """
-        super().cleanup()
-        super().closeEvent(event)
-
 
 def main():
     """Main method to run the BECStatusBox widget."""

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/device_box/device_box.py

@@ -8,11 +8,11 @@ from qtpy.QtGui import QDoubleValidator
 from qtpy.QtWidgets import QDoubleSpinBox, QVBoxLayout, QWidget
 
 from bec_widgets.utils import UILoader
-from bec_widgets.utils.bec_connector import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import apply_theme
 
 
-class DeviceBox(BECConnector, QWidget):
+class DeviceBox(BECWidget, QWidget):
     device_changed = Signal(str, str)
 
     def __init__(self, parent=None, device=None, *args, **kwargs):

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/device_combobox/device_combobox.py

@@ -2,6 +2,7 @@ from typing import TYPE_CHECKING
 
 from qtpy.QtWidgets import QComboBox
 
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.widgets.base_classes.device_input_base import DeviceInputBase, DeviceInputConfig
 
 if TYPE_CHECKING:
@@ -82,11 +83,3 @@ class DeviceComboBox(DeviceInputBase, QComboBox):
         if device_obj is None:
             raise ValueError(f"Device {device_name} is not found.")
         return device_obj
-
-    def cleanup(self):
-        """Cleanup the widget."""
-        super().cleanup()
-
-    def closeEvent(self, event):
-        super().cleanup()
-        return QComboBox.closeEvent(self, event)

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/device_line_edit/device_line_edit.py

@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING
 from qtpy.QtCore import QSize
 from qtpy.QtWidgets import QCompleter, QLineEdit, QSizePolicy
 
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.widgets.base_classes.device_input_base import DeviceInputBase, DeviceInputConfig
 
 if TYPE_CHECKING:
@@ -33,8 +34,8 @@ class DeviceLineEdit(DeviceInputBase, QLineEdit):
         default: str | None = None,
         arg_name: str | None = None,
     ):
+        super().__init__(client=client, config=config, gui_id=gui_id)
         QLineEdit.__init__(self, parent=parent)
-        DeviceInputBase.__init__(self, client=client, config=config, gui_id=gui_id)
 
         self.completer = QCompleter(self)
         self.setCompleter(self.completer)
@@ -94,11 +95,3 @@ class DeviceLineEdit(DeviceInputBase, QLineEdit):
         if device_obj is None:
             raise ValueError(f"Device {device_name} is not found.")
         return device_obj
-
-    def cleanup(self):
-        """Cleanup the widget."""
-        super().cleanup()
-
-    def closeEvent(self, event):
-        super().cleanup()
-        return QLineEdit.closeEvent(self, event)

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/dock/dock.py

@@ -6,7 +6,8 @@ from pydantic import Field
 from pyqtgraph.dockarea import Dock
 
 from bec_widgets.cli.rpc_wigdet_handler import widget_handler
-from bec_widgets.utils import BECConnector, ConnectionConfig, GridLayoutManager
+from bec_widgets.utils import ConnectionConfig, GridLayoutManager
+from bec_widgets.utils.bec_widget import BECWidget
 
 if TYPE_CHECKING:
     from qtpy.QtWidgets import QWidget
@@ -24,7 +25,7 @@ class DockConfig(ConnectionConfig):
     )
 
 
-class BECDock(BECConnector, Dock):
+class BECDock(BECWidget, Dock):
     USER_ACCESS = [
         "_config_dict",
         "_rpc_id",
@@ -91,7 +92,7 @@ class BECDock(BECConnector, Dock):
             super().float()
 
     @property
-    def widget_list(self) -> list[BECConnector]:
+    def widget_list(self) -> list[BECWidget]:
         """
         Get the widgets in the dock.
 
@@ -101,7 +102,7 @@ class BECDock(BECConnector, Dock):
         return self.widgets
 
     @widget_list.setter
-    def widget_list(self, value: list[BECConnector]):
+    def widget_list(self, value: list[BECWidget]):
         self.widgets = value
 
     def hide_title_bar(self):
@@ -153,13 +154,13 @@ class BECDock(BECConnector, Dock):
 
     def add_widget(
         self,
-        widget: BECConnector | str,
+        widget: BECWidget | str,
         row=None,
         col=0,
         rowspan=1,
         colspan=1,
         shift: Literal["down", "up", "left", "right"] = "down",
-    ) -> BECConnector:
+    ) -> BECWidget:
         """
         Add a widget to the dock.
 
@@ -238,6 +239,7 @@ class BECDock(BECConnector, Dock):
         for widget in self.widgets:
             if hasattr(widget, "cleanup"):
                 widget.cleanup()
+        self.widgets.clear()
         super().cleanup()
 
     def close(self):

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/dock/dock_area.py

@@ -9,7 +9,8 @@ from qtpy.QtCore import Qt
 from qtpy.QtGui import QPainter, QPaintEvent
 from qtpy.QtWidgets import QWidget
 
-from bec_widgets.utils import BECConnector, ConnectionConfig, WidgetContainerUtils
+from bec_widgets.utils import ConnectionConfig, WidgetContainerUtils
+from bec_widgets.utils.bec_widget import BECWidget
 
 from .dock import BECDock, DockConfig
 
@@ -21,7 +22,7 @@ class DockAreaConfig(ConnectionConfig):
     )
 
 
-class BECDockArea(BECConnector, DockArea):
+class BECDockArea(BECWidget, DockArea):
     USER_ACCESS = [
         "_config_dict",
         "panels",
@@ -227,6 +228,7 @@ class BECDockArea(BECConnector, DockArea):
         self.attach_all()
         for dock in dict(self.docks).values():
             dock.remove()
+        self.docks.clear()
 
     def cleanup(self):
         """

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/figure.py

@@ -12,7 +12,8 @@ from qtpy.QtCore import Signal as pyqtSignal
 from qtpy.QtWidgets import QWidget
 from typeguard import typechecked
 
-from bec_widgets.utils import BECConnector, ConnectionConfig, WidgetContainerUtils
+from bec_widgets.utils import ConnectionConfig, WidgetContainerUtils
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import apply_theme
 from bec_widgets.widgets.figure.plots.image.image import BECImageShow, ImageConfig
 from bec_widgets.widgets.figure.plots.motor_map.motor_map import BECMotorMap, MotorMapConfig
@@ -108,7 +109,7 @@ class WidgetHandler:
         return widget
 
 
-class BECFigure(BECConnector, pg.GraphicsLayoutWidget):
+class BECFigure(BECWidget, pg.GraphicsLayoutWidget):
     USER_ACCESS = [
         "_rpc_id",
         "_config_dict",
@@ -728,14 +729,9 @@ class BECFigure(BECConnector, pg.GraphicsLayoutWidget):
         """Clear all widgets from the figure and reset to default state"""
         for widget in list(self._widgets.values()):
             widget.remove()
-        # self.clear()
-        self._widgets = defaultdict(dict)
+        self._widgets.clear()
         self.grid = []
         theme = self.config.theme
         self.config = FigureConfig(
             widget_class=self.__class__.__name__, gui_id=self.gui_id, theme=theme
         )
-
-    # def cleanup(self):
-    #     self.clear_all()
-    #     super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/plots/image/image.py

@@ -596,7 +596,4 @@ class BECImageShow(BECPlotBase):
             self.bec_dispatcher.disconnect_slot(
                 self.on_image_update, MessageEndpoints.device_monitor(monitor)
             )
-        for image in self.images:
-            image.cleanup()
-
-        super().cleanup()
+        self.images.clear()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/plots/motor_map/motor_map.py

@@ -518,4 +518,3 @@ class BECMotorMap(BECPlotBase):
     def cleanup(self):
         """Cleanup the widget."""
         self._disconnect_current_motors()
-        super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/plots/plot_base.py

@@ -296,9 +296,4 @@ class BECPlotBase(BECConnector, pg.GraphicsLayout):
     def remove(self):
         """Remove the plot widget from the figure."""
         if self.figure is not None:
-            self.cleanup()
             self.figure.remove(widget_id=self.gui_id)
-
-    def cleanup(self):
-        """Cleanup the plot widget."""
-        super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/plots/waveform/waveform.py

@@ -1368,6 +1368,4 @@ class BECWaveform(BECPlotBase):
                 self.on_async_readback,
                 MessageEndpoints.device_async_readback(self.scan_id, curve_id),
             )
-        for curve in self.curves:
-            curve.cleanup()
-        super().cleanup()
+        self.curves.clear()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/figure/plots/waveform/waveform_curve.py

@@ -262,4 +262,4 @@ class BECCurve(BECConnector, pg.PlotDataItem):
         """Remove the curve from the plot."""
         # self.parent_item.removeItem(self)
         self.parent_item.remove_curve(self.name())
-        self.cleanup()
+        self.rpc_register.remove_rpc(self)

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/jupyter_console/jupyter_console.py

@@ -10,6 +10,7 @@ class BECJupyterConsole(RichJupyterWidget):  # pragma: no cover:
         super().__init__()
 
         self.inprocess = None
+        self.client = None
 
         self.kernel_manager, self.kernel_client = self._init_kernel(inprocess=inprocess)
         self.set_default_style("linux")
@@ -60,6 +61,11 @@ class BECJupyterConsole(RichJupyterWidget):  # pragma: no cover:
         self.kernel_client.stop_channels()
         self.kernel_manager.shutdown_kernel()
 
+    def closeEvent(self, event):
+        self.shutdown_kernel()
+        if self.client:
+            self.client.shutdown()
+
 
 if __name__ == "__main__":  # pragma: no cover
     import sys

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/motor_map/motor_map_widget.py

@@ -6,7 +6,7 @@ from qtpy.QtWidgets import QVBoxLayout, QWidget
 
 from bec_widgets.qt_utils.settings_dialog import SettingsDialog
 from bec_widgets.qt_utils.toolbar import ModularToolBar
-from bec_widgets.utils import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.widgets.figure import BECFigure
 from bec_widgets.widgets.figure.plots.motor_map.motor_map import MotorMapConfig
 from bec_widgets.widgets.motor_map.motor_map_dialog.motor_map_settings import MotorMapSettings
@@ -18,7 +18,7 @@ from bec_widgets.widgets.motor_map.motor_map_dialog.motor_map_toolbar import (
 )
 
 
-class BECMotorMapWidget(BECConnector, QWidget):
+class BECMotorMapWidget(BECWidget, QWidget):
     USER_ACCESS = [
         "change_motors",
         "set_max_points",
@@ -208,10 +208,6 @@ class BECMotorMapWidget(BECConnector, QWidget):
         self.toolbar.widgets["motor_y"].device_combobox.cleanup()
         return super().cleanup()
 
-    def closeEvent(self, event):
-        self.cleanup()
-        QWidget().closeEvent(event)
-
 
 def main():  # pragma: no cover
     from qtpy.QtWidgets import QApplication

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/ring_progress_bar/ring.py

@@ -288,7 +288,3 @@ class Ring(BECConnector):
         value = msg.get("signals").get(device).get("value")
         self.set_value(value)
         self.parent_progress_widget.update()
-
-    def cleanup(self):
-        self.reset_connection()
-        super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/ring_progress_bar/ring_progress_bar.py

@@ -10,7 +10,8 @@ from qtpy import QtCore, QtGui
 from qtpy.QtCore import QSize, Slot
 from qtpy.QtWidgets import QSizePolicy, QWidget
 
-from bec_widgets.utils import BECConnector, Colors, ConnectionConfig, EntryValidator
+from bec_widgets.utils import Colors, ConnectionConfig, EntryValidator
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.widgets.ring_progress_bar.ring import Ring, RingConfig
 
 
@@ -66,7 +67,7 @@ class RingProgressBarConfig(ConnectionConfig):
     _validate_colormap = field_validator("color_map")(Colors.validate_color_map)
 
 
-class RingProgressBar(BECConnector, QWidget):
+class RingProgressBar(BECWidget, QWidget):
     USER_ACCESS = [
         "_get_all_rpc",
         "_rpc_id",
@@ -208,7 +209,7 @@ class RingProgressBar(BECConnector, QWidget):
             index(int): Index of the progress bar to remove.
         """
         ring = self._find_ring_by_index(index)
-        ring.cleanup()
+        ring.reset_connection()
         self._rings.remove(ring)
         self.config.rings.remove(ring.config)
         self.config.num_bars -= 1
@@ -622,9 +623,8 @@ class RingProgressBar(BECConnector, QWidget):
 
     def clear_all(self):
         for ring in self._rings:
-            ring.cleanup()
-            del ring
-            self._rings = []
+            ring.reset_connection()
+        self._rings.clear()
         self.update()
         self.initialize_bars()
 
@@ -633,6 +633,6 @@ class RingProgressBar(BECConnector, QWidget):
             self.on_scan_queue_status, MessageEndpoints.scan_queue_status()
         )
         for ring in self._rings:
-            ring.cleanup()
-            del ring
+            ring.reset_connection()
+        self._rings.clear()
         super().cleanup()

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/scan_control/scan_control.py

@@ -10,13 +10,13 @@ from qtpy.QtWidgets import (
     QWidget,
 )
 
-from bec_widgets.utils import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import apply_theme
 from bec_widgets.widgets.scan_control.scan_group_box import ScanGroupBox
 from bec_widgets.widgets.stop_button.stop_button import StopButton
 
 
-class ScanControl(BECConnector, QWidget):
+class ScanControl(BECWidget, QWidget):
 
     def __init__(
         self, parent=None, client=None, gui_id: str | None = None, allowed_scans: list | None = None
@@ -196,10 +196,6 @@ class ScanControl(BECConnector, QWidget):
                     widget.cleanup()
         super().cleanup()
 
-    def closeEvent(self, event):
-        self.cleanup()
-        return QWidget.closeEvent(self, event)
-
 
 # Application example
 if __name__ == "__main__":  # pragma: no cover

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/stop_button/stop_button.py

@@ -1,10 +1,10 @@
 from qtpy.QtCore import Slot
 from qtpy.QtWidgets import QPushButton
 
-from bec_widgets.utils import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 
 
-class StopButton(BECConnector, QPushButton):
+class StopButton(BECWidget, QPushButton):
     """A button that stops the current scan."""
 
     def __init__(self, parent=None, client=None, config=None, gui_id=None):

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/text_box/text_box.py

@@ -3,7 +3,8 @@ import re
 from pydantic import Field, field_validator
 from qtpy.QtWidgets import QTextEdit
 
-from bec_widgets.utils.bec_connector import BECConnector, ConnectionConfig
+from bec_widgets.utils.bec_connector import ConnectionConfig
+from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import Colors
 
 
@@ -27,7 +28,7 @@ class TextBoxConfig(ConnectionConfig):
     _validate_background_color = field_validator("background_color")(Colors.validate_color)
 
 
-class TextBox(BECConnector, QTextEdit):
+class TextBox(BECWidget, QTextEdit):
 
     USER_ACCESS = ["set_color", "set_text", "set_font_size"]
 

{bec_widgets-0.87.0 → bec_widgets-0.87.1}/bec_widgets/widgets/website/website.py

@@ -2,7 +2,7 @@ from qtpy.QtCore import QUrl, qInstallMessageHandler
 from qtpy.QtWebEngineWidgets import QWebEngineView
 from qtpy.QtWidgets import QApplication
 
-from bec_widgets.utils import BECConnector
+from bec_widgets.utils.bec_widget import BECWidget
 
 
 def suppress_qt_messages(type_, context, msg):
@@ -14,7 +14,7 @@ def suppress_qt_messages(type_, context, msg):
 qInstallMessageHandler(suppress_qt_messages)
 
 
-class WebsiteWidget(BECConnector, QWebEngineView):
+class WebsiteWidget(BECWidget, QWebEngineView):
     """
     A simple widget to display a website
     """