PYQGIS——图层目录树右键菜单

自定义MenuProvider

初始化时传入图层目录树（layerTreeView）和地图画布（mapCanvas），以便右键菜单操作目录树和画布。 重写createContextMenu方法，判断右键菜单的位置、当前节点、当前图层、图层类型等。 其中使用到了actionAddGroup、actionZoomToLayer两个action，更多action可以查看QgsLayerTreeViewDefaultActions的帮助文档

from PyQt5.QtWidgets import QMenu, QAction from qgis.core import QgsLayerTreeNode, QgsLayerTree, QgsMapLayerType from qgis.gui import QgsLayerTreeViewMenuProvider, QgsLayerTreeView, QgsLayerTreeViewDefaultActions, QgsMapCanvas class CustomMenuProvider(QgsLayerTreeViewMenuProvider): def __init__(self, layerTreeView: QgsLayerTreeView, mapCanvas: QgsMapCanvas, *args, **kwargs): super().__init__(*args, **kwargs) self.layerTreeView = layerTreeView self.mapCanvas = mapCanvas def createContextMenu(self): menu = QMenu() actions: QgsLayerTreeViewDefaultActions = self.layerTreeView.defaultActions() if not self.layerTreeView.currentIndex().isValid(): # 不在图层上右键 self.actionAddGroup = actions.actionAddGroup(menu) menu.addAction(self.actionAddGroup) menu.addAction('Expand All', self.layerTreeView.expandAll) menu.addAction('Collapse All', self.layerTreeView.collapseAll) return menu node: QgsLayerTreeNode = self.layerTreeView.currentNode() if QgsLayerTree.isGroup(node): # 图组操作 print('group') pass elif QgsLayerTree.isLayer(node): print('layer') self.actionZoomToLayer = actions.actionZoomToLayer(self.mapCanvas, menu) menu.addAction(self.actionZoomToLayer) # 图层操作 layer = self.layerTreeView.currentLayer() if layer.type() == QgsMapLayerType.VectorLayer: # 矢量图层 openAttributeDialog = QAction('open Attribute Table', menu) openAttributeDialog.triggered.connect(lambda: print(111)) menu.addAction(openAttributeDialog) else: # 栅格图层 pass else: print('node type is none') return menu

设置自定义MenuProvider

self.customMenuProvider = CustomMenuProvider(self.layerTreeView, self.mapCanvas) self.layerTreeView.setMenuProvider(self.customMenuProvider)

效果图

本文源码地址：点这里