setoolsgui/widgets/views/treewidget.py - main - githubmirror/setools

setoolsgui/widgets/views/treewidget.py
79 строк3 KB
Chris PeBenito
DirectedGraphResultTab: Add tree browser.
20 мар 2024, 15:19
20 мар 2024, 15:1905776e1
# Copyright 2016, Chris PeBenito <pebenito@ieee.org>
#
# SPDX-License-Identifier: LGPL-2.1-only
#
#
from PyQt6 import QtCore, QtGui, QtWidgets
import setools

from .. import details, models

__all__ = ("SEToolsTreeWidget",)


class SEToolsTreeWidget(QtWidgets.QTreeWidget):

    """QTreeWidget class extended for SETools use."""

    def contextMenuEvent(self, event: QtGui.QContextMenuEvent) -> None:  # type: ignore[override]
        """Handle the context menu event."""
        menu = QtWidgets.QMenu(self)
        menu.setAttribute(QtCore.Qt.WidgetAttribute.WA_DeleteOnClose)

        # Add any actions provided by the item. This works differently than the
        # table and list views.  In those, each time the context menu data is
        # retrieved from the model, a new action is generated.  Since this is
        # standard model, the item data contains a list of callables that
        # generate actions.
        index = self.indexAt(event.pos())
        if index.isValid():
            # For now, keep this simple; only types are used in trees at this time.
            type_: setools.Type = index.data(models.ModelRoles.PolicyObjRole)
            item_info = details.type_detail_action(type_, self)
            menu.addAction(item_info)

            # for action_setup in index.data(models.ModelRoles.PolicyObjRole):
            #    assert callable(action_setup), \
            #        "SEToolsTreeWidget ContextMenuRole must be callable. This is an SETools bug."
            #    action = action_setup(self)
            #    menu.addAction(action)
            menu.addSeparator()

        copy_tree_action = QtGui.QAction("Copy Tree...", self)
        copy_tree_action.triggered.connect(self.copy)
        menu.addAction(copy_tree_action)
        menu.exec(event.globalPos())
        return

    def copy(self) -> None:
        """Copy the tree to the clipboard."""
        items = list[str]()
        it = QtWidgets.QTreeWidgetItemIterator(self)
        prev_depth = 0
        while it.value():
            depth = 0
            item = it.value()
            assert item, "No item available, this is an SETools bug"  # type narrowing
            parent = item.parent()
            while parent:
                depth += 1
                parent = parent.parent()

            if depth < prev_depth:
                items.extend(["  |" * depth, "\n"])

            if depth:
                items.extend(["  |" * depth, "--", item.text(0), "\n"])
            else:
                items.extend([item.text(0), "\n"])

            prev_depth = depth
            it += 1

        cb = QtWidgets.QApplication.clipboard()
        assert cb, "No clipboard available, this is an SETools bug"  # type narrowing
        cb.setText("".join(items))

    def cut(self) -> None:
        """Copy the tree to the clipboard. Cut from the widget is not available."""
        self.copy()