mirror of
https://github.com/bec-project/bec_widgets.git
synced 2026-03-13 04:08:01 +01:00
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from qtpy.QtCore import Slot
|
|
from qtpy.QtWidgets import QApplication, QWidget
|
|
|
|
from bec_widgets.utils.bec_connector import BECConnector, ConnectionConfig
|
|
from bec_widgets.utils.colors import set_theme
|
|
|
|
|
|
class BECWidget(BECConnector):
|
|
"""Mixin class for all BEC widgets, to handle cleanup"""
|
|
|
|
# The icon name is the name of the icon in the icon theme, typically a name taken
|
|
# from fonts.google.com/icons. Override this in subclasses to set the icon name.
|
|
ICON_NAME = "widgets"
|
|
|
|
def __init__(
|
|
self,
|
|
client=None,
|
|
config: ConnectionConfig = None,
|
|
gui_id: str = None,
|
|
theme_update: bool = False,
|
|
):
|
|
if not isinstance(self, QWidget):
|
|
raise RuntimeError(f"{repr(self)} is not a subclass of QWidget")
|
|
super().__init__(client, config, gui_id)
|
|
|
|
# Set the theme to auto if it is not set yet
|
|
app = QApplication.instance()
|
|
if not hasattr(app, "theme"):
|
|
set_theme("auto")
|
|
|
|
if theme_update:
|
|
self._connect_to_theme_change()
|
|
|
|
def _connect_to_theme_change(self):
|
|
"""Connect to the theme change signal."""
|
|
qapp = QApplication.instance()
|
|
if hasattr(qapp, "theme_signal"):
|
|
qapp.theme_signal.theme_updated.connect(self._update_theme)
|
|
|
|
def _update_theme(self, theme: str):
|
|
"""Update the theme."""
|
|
if theme is None:
|
|
qapp = QApplication.instance()
|
|
if hasattr(qapp, "theme"):
|
|
theme = qapp.theme["theme"]
|
|
else:
|
|
theme = "dark"
|
|
self.apply_theme(theme)
|
|
|
|
@Slot(str)
|
|
def apply_theme(self, theme: str):
|
|
"""
|
|
Apply the theme to the plot widget.
|
|
|
|
Args:
|
|
theme(str, optional): The theme to be applied.
|
|
"""
|
|
|
|
def cleanup(self):
|
|
"""Cleanup the widget."""
|
|
|
|
def closeEvent(self, event):
|
|
self.rpc_register.remove_rpc(self)
|
|
try:
|
|
self.cleanup()
|
|
finally:
|
|
super().closeEvent(event)
|