1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
# 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()
|
