使用python将QTreeWidgetItem发送到函数

时间:2014-02-01 21:01:37

标签: python contextmenu pyqt4 qtreewidget

我正在尝试使用Python,Komodo和QT4制作程序。

我试图在用户用鼠标选择QTreeWidgetItem后将其发送给函数。我所能做的就是用鼠标移动所选点的X和Y的位置。谁能告诉我如何将QTreeWidgetItem发送给函数?

这是我的代码:

from PyQt4 import QtCore, QtGui

try:
    _fromUtf8 = QtCore.QString.fromUtf8
except AttributeError:
    def _fromUtf8(s):
        return s

try:
    _encoding = QtGui.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig)

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName(_fromUtf8("MainWindow"))
        MainWindow.resize(800, 600)
        self.centralwidget = QtGui.QWidget(MainWindow)
        self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
        self.treeWidget = QtGui.QTreeWidget(self.centralwidget)
        self.treeWidget.setGeometry(QtCore.QRect(155, 50, 481, 361))
        self.treeWidget.setObjectName(_fromUtf8("treeWidget"))
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtGui.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 25))
        self.menubar.setObjectName(_fromUtf8("menubar"))
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtGui.QStatusBar(MainWindow)
        self.statusbar.setObjectName(_fromUtf8("statusbar"))
        MainWindow.setStatusBar(self.statusbar)
        self.create_popup_menu()
        self.treeWidget.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.treeWidget.customContextMenuRequested.connect(self.on_context_menu)
        string="default,default:cluster1,default:cluster1:clusterA,default:cluster2,default:cluster2:clusterA"
        self.retranslateUi(MainWindow)
        self.buildingTree(string)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow", None))
        self.treeWidget.headerItem().setText(0, _translate("MainWindow", "Assignment1", None))
        __sortingEnabled = self.treeWidget.isSortingEnabled()

    def buildingTree(self, string):
        arrTree = []
        arrPath = []
        arrString = []
        arrString = string.split(",")
        root = QtGui.QTreeWidgetItem(self.treeWidget)
        root.setText(0,arrString[0])
        arrTree.append(root)
        arrString.pop(0)
        for path in arrString:
            arrPath = path.split(":")
            nameOfFather = arrPath[len(arrPath)-2]
            arrTree.reverse()
            for node in arrTree:
                nameOfNode = node.text(0)
                if nameOfNode == nameOfFather:
                    sonNode = QtGui.QTreeWidgetItem(node)
                    sonNode.setText(0,arrPath[len(arrPath)-1])
                    arrTree.insert(0,sonNode)
                    arrTree.reverse()
                    break
        self.treeWidget.setSortingEnabled(False)        

    def new_cluster(self):
        print "New Cluster"

    def rename_cluster(self):
        print "Rename cluster"

    def delete_cluster(self):
        print "Delete cluster"

    def create_popup_menu(self, parent=None):
        self.popup_menu = QtGui.QMenu(parent)
        self.popup_menu.addAction("New", self.new_cluster)
        self.popup_menu.addAction("Rename", self.rename_cluster)
        self.popup_menu.addSeparator()
        self.popup_menu.addAction("Delete", self.delete_cluster)

    def on_context_menu(self, pos):
        print "open menu"
        position = QtGui.QTreeWidgetItem(self.treeWidget)
        print position
        position = pos
        print position
        print "%%%%%%%%%%%%%%%"
        """
        node = self.treeWidget.itemAt(position)
        print node
        node = self.treeWidget.setCurrentItem(node)
        print node.text(0)
        self.popup_menu.exec_(self.treeWidget.mapToGlobal(pos))

        global_pos = self.mapToGlobal(pos)
        t = self.itemAt(pos)
        self.setCurrentItem(t)
        """

if __name__ == "__main__":
    import sys
    app = QtGui.QApplication(sys.argv)
    MainWindow = QtGui.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

1 个答案:

答案 0 :(得分:1)

您的代码非常不清楚,因为您正在编辑pyuic生成的文件,而不是将其导入主应用程序。

我重新构建了代码,并使用event filter来处理上下文菜单,这有助于简化代码。我还使用lambda函数来处理菜单操作。以下代码假定pyuic中的文件保存为mainwindow_ui.py(您显然需要重新生成)。

from PyQt4 import QtCore, QtGui
from mainwindow_ui import Ui_MainWindow

class MainWindow(QtGui.QMainWindow, Ui_MainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.setupUi(self)
        self.buildingTree(
            "default,default:cluster1,default:cluster1:clusterA,"
            "default:cluster2,default:cluster2:clusterA"
            )
        self.treeWidget.viewport().installEventFilter(self)

    def eventFilter(self, target, event):
        if (event.type() == QtCore.QEvent.ContextMenu and
            target is self.treeWidget.viewport()):
            item = self.treeWidget.itemAt(event.pos())
            if item is not None:
                menu = QtGui.QMenu()
                menu.addAction(
                    "New", lambda: self.new_cluster(item))
                menu.addAction(
                    "Rename", lambda: self.rename_cluster(item))
                menu.addSeparator()
                menu.addAction(
                    "Delete", lambda: self.delete_cluster(item))
                menu.exec_(event.globalPos())
            return True
        return super(MainWindow, self).eventFilter(target, event)

    def new_cluster(self, item):
        print "New Cluster", item.text(0)

    def rename_cluster(self, item):
        print "Rename cluster", item.text(0)

    def delete_cluster(self, item):
        print "Delete cluster", item.text(0)

    def buildingTree(self, string):
        arrTree = []
        arrPath = []
        arrString = []
        arrString = string.split(",")
        root = QtGui.QTreeWidgetItem(self.treeWidget)
        root.setText(0,arrString[0])
        arrTree.append(root)
        arrString.pop(0)
        for path in arrString:
            arrPath = path.split(":")
            nameOfFather = arrPath[len(arrPath)-2]
            arrTree.reverse()
            for node in arrTree:
                nameOfNode = node.text(0)
                if nameOfNode == nameOfFather:
                    sonNode = QtGui.QTreeWidgetItem(node)
                    sonNode.setText(0,arrPath[len(arrPath)-1])
                    arrTree.insert(0,sonNode)
                    arrTree.reverse()
                    break
        self.treeWidget.setSortingEnabled(False)

if __name__ == "__main__":

    import sys
    app = QtGui.QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())