mirror of
https://github.com/bec-project/bec_widgets.git
synced 2026-05-18 20:36:57 +02:00
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
from abc import ABC, abstractmethod
|
|
from collections import defaultdict
|
|
|
|
# pylint: disable=no-name-in-module
|
|
from qtpy.QtCore import QSize
|
|
from qtpy.QtWidgets import QToolBar, QToolButton, QWidget
|
|
|
|
|
|
class ToolBarAction(ABC):
|
|
@abstractmethod
|
|
def add_to_toolbar(self, toolbar: QToolBar, target: QWidget):
|
|
"""Adds an action or widget to a toolbar.
|
|
|
|
Args:
|
|
toolbar (QToolBar): The toolbar to add the action or widget to.
|
|
target (QWidget): The target widget for the action.
|
|
"""
|
|
|
|
|
|
class SeparatorAction(ToolBarAction):
|
|
"""Separator action for the toolbar."""
|
|
|
|
def add_to_toolbar(self, toolbar: QToolBar, target: QWidget):
|
|
self.separator = QToolButton()
|
|
self.separator.setFixedSize(2, 22)
|
|
toolbar.addWidget(self.separator)
|
|
|
|
|
|
class ModularToolBar(QToolBar):
|
|
"""Modular toolbar with optional automatic initialization.
|
|
Args:
|
|
parent (QWidget, optional): The parent widget of the toolbar. Defaults to None.
|
|
actions (list[ToolBarAction], optional): A list of action creators to populate the toolbar. Defaults to None.
|
|
target_widget (QWidget, optional): The widget that the actions will target. Defaults to None.
|
|
color (str, optional): The background color of the toolbar. Defaults to "black".
|
|
"""
|
|
|
|
def __init__(
|
|
self, parent=None, actions=None, target_widget=None, color: str = "rgba(255, 255, 255, 0)"
|
|
):
|
|
super().__init__(parent)
|
|
|
|
self.widgets = defaultdict(dict)
|
|
self.set_background_color(color)
|
|
|
|
if actions is not None and target_widget is not None:
|
|
self.populate_toolbar(actions, target_widget)
|
|
|
|
def populate_toolbar(self, actions: dict, target_widget):
|
|
"""Populates the toolbar with a set of actions.
|
|
|
|
Args:
|
|
actions (list[ToolBarAction]): A list of action creators to populate the toolbar.
|
|
target_widget (QWidget): The widget that the actions will target.
|
|
"""
|
|
self.clear()
|
|
for action_id, action in actions.items():
|
|
action.add_to_toolbar(self, target_widget)
|
|
self.widgets[action_id] = action
|
|
|
|
def set_background_color(self, color: str):
|
|
self.setStyleSheet(f"QToolBar {{ background: {color}; }}")
|
|
self.setIconSize(QSize(20, 20))
|
|
self.setMovable(False)
|
|
self.setFloatable(False)
|
|
self.setContentsMargins(0, 0, 0, 0)
|