diff options
Diffstat (limited to 'python/examples/triage/imports.py')
| -rw-r--r-- | python/examples/triage/imports.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/python/examples/triage/imports.py b/python/examples/triage/imports.py index 24de5327..e1a85a7a 100644 --- a/python/examples/triage/imports.py +++ b/python/examples/triage/imports.py @@ -2,7 +2,7 @@ from PySide2.QtWidgets import QTreeView, QVBoxLayout, QWidget from PySide2.QtCore import Qt, QAbstractItemModel, QModelIndex, QSize from binaryninja.enums import SymbolType import binaryninjaui -from binaryninjaui import ViewFrame, FilterTarget, FilteredView, UIContext +from binaryninjaui import ViewFrame, FilterTarget, FilteredView, UIContext, UIActionHandler class GenericImportsModel(QAbstractItemModel): @@ -132,6 +132,11 @@ class ImportsTreeView(QTreeView, FilterTarget): self.parent = parent self.view = view + # Allow view-specific shortcuts when imports are focused + self.actionHandler = UIActionHandler() + self.actionHandler.setupActionHandler(self) + self.actionHandler.setActionContext(lambda: self.view.actionContext()) + self.model = GenericImportsModel(self.data) self.setModel(self.model) self.setRootIsDecorated(False) |