Line data    Source code

       1              : # SPDX-FileCopyrightText: 2025 PairInteraction Developers
       2              : # SPDX-License-Identifier: LGPL-3.0-or-later
       3            1 : from __future__ import annotations
       4              : 
       5            1 : import logging
       6            1 : from typing import TYPE_CHECKING, Generic, TypeVar
       7              : 
       8            1 : from PySide6.QtWidgets import QStackedWidget, QWidget
       9              : 
      10              : if TYPE_CHECKING:
      11              :     from collections.abc import ItemsView
      12              : 
      13            1 : logger = logging.getLogger(__name__)
      14              : 
      15              : 
      16            1 : WidgetType = TypeVar("WidgetType", bound=QWidget)
      17              : 
      18              : 
      19            1 : class NamedStackedWidget(QStackedWidget, Generic[WidgetType]):
      20            1 :     def __init__(self, parent: QWidget | None = None) -> None:
      21            1 :         super().__init__(parent)
      22            1 :         self._widgets: dict[str, WidgetType] = {}
      23              : 
      24            1 :     def addNamedWidget(self, widget: WidgetType, name: str) -> None:
      25            1 :         widget.setObjectName(name)
      26            1 :         self.addWidget(widget)
      27            1 :         self._widgets[name] = widget
      28              : 
      29            1 :     def setCurrentNamedWidget(self, name: str) -> None:
      30            1 :         widget = self.getNamedWidget(name)
      31            1 :         current_widget = self.currentWidget()
      32            1 :         if widget == current_widget:
      33            1 :             return
      34            1 :         self.setCurrentWidget(widget)
      35            1 :         logger.debug("Switched NamedStackedWidget to %s", name)
      36              : 
      37            1 :     def getNamedWidget(self, name: str) -> WidgetType:
      38            1 :         return self._widgets[name]
      39              : 
      40            1 :     def items(self) -> ItemsView[str, WidgetType]:
      41            1 :         return self._widgets.items()
      42              : 
      43            1 :     def currentWidget(self) -> WidgetType:
      44            1 :         return super().currentWidget()  # type: ignore [return-value] # explicitly override type hints
      45              : 
      46            1 :     def addWidget(self, widget: WidgetType) -> int:  # type: ignore [override] # explicitly override type hints
      47            1 :         return super().addWidget(widget)