python + pyqt5 QTreeWidget 实现文件资源管理

这是VSCode的文件资源管理或者叫资源文件导航，这是一个树形结构的，我们使用 QTreeWidget来实现一下，最终的结果如下：

 

我先把代码贴出来，UI界面的编程，不可能把每一行代码都讲清楚，我只讲一些思路和需要注意的细节吧：

 第一个文件（fileopen.py），UI设计的文件，是QT Designer设计好的ui文件直接转换为的Python文件：

# -*- coding: utf-8 -*- # Form implementation generated from reading ui file 'fileopen.ui' # # Created by: PyQt5 UI code generator 5.13.0 # # WARNING! All changes made in this file will be lost! from PyQt5 import QtCore, QtGui, QtWidgets class Ui_MainWindow(object): def setupUi(self, MainWindow): MainWindow.setObjectName("MainWindow") MainWindow.resize(800, 600) self.centralwidget = QtWidgets.QWidget(MainWindow) self.centralwidget.setObjectName("centralwidget") MainWindow.setCentralWidget(self.centralwidget) self.menubar = QtWidgets.QMenuBar(MainWindow) self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 23)) self.menubar.setObjectName("menubar") self.menufile = QtWidgets.QMenu(self.menubar) self.menufile.setObjectName("menufile") MainWindow.setMenuBar(self.menubar) self.statusbar = QtWidgets.QStatusBar(MainWindow) self.statusbar.setObjectName("statusbar") MainWindow.setStatusBar(self.statusbar) self.actionfileopen = QtWidgets.QAction(MainWindow) self.actionfileopen.setObjectName("actionfileopen") self.menufile.addAction(self.actionfileopen) self.menubar.addAction(self.menufile.menuAction()) self.retranslateUi(MainWindow) QtCore.QMetaObject.connectSlotsByName(MainWindow) def retranslateUi(self, MainWindow): _translate = QtCore.QCoreApplication.translate MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow")) self.menufile.setTitle(_translate("MainWindow", "File")) self.actionfileopen.setText(_translate("MainWindow", "Open Folder..."))

 

第二个文件（file_tree.py）， 逻辑文件，主要内容的实现，在这里：

import sys from PyQt5.QtWidgets import * from PyQt5 import Qt, QtGui from filename import file_name as fn from fileopen import Ui_MainWindow from PyQt5 import QtWidgets import threading import os class Tree(QMainWindow, Ui_MainWindow): def __init__(self, parent = None): super(Tree, self).__init__(parent) self.setupUi(self) self.setWindowTitle("File_Tree") self.actionfileopen.triggered.connect(self.Open_Folder) def Open_Folder(self): path = QFileDialog.getExistingDirectory(self, "选取文件夹", "./") self.tree = QTreeWidget() self.tree.setColumnCount(1) self.tree.setColumnWidth(0, 50) self.tree.setHeaderLabels(["EXPLORER"]) self.tree.setIconSize(Qt.QSize(25, 25)) self.tree.setSelectionMode(QAbstractItemView.ExtendedSelection) self.actionfileopen.triggered.connect(self.Open_Folder) dirs = fn(path) fileInfo = Qt.QFileInfo(path) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) root = QTreeWidgetItem(self.tree) root.setText(0,path.split('/')[-1]) root.setIcon(0,QtGui.QIcon(icon)) self.CreateTree(dirs, root, path) self.setCentralWidget(self.tree) QApplication.processEvents() def CreateTree(self, dirs, root, path): for i in dirs: path_new = path + '\\' + i if os.path.isdir(path_new): fileInfo = Qt.QFileInfo(path_new) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) child = QTreeWidgetItem(root) child.setText(0,i) child.setIcon(0,QtGui.QIcon(icon)) dirs_new = fn(path_new) self.CreateTree(dirs_new, child, path_new) else: fileInfo = Qt.QFileInfo(path_new) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) child = QTreeWidgetItem(root) child.setText(0,i) child.setIcon(0,QtGui.QIcon(icon)) if __name__ == "__main__": app = QApplication(sys.argv) win = Tree() win.show() sys.exit(app.exec_())

 

第三个文件（filename.py）我用于测试的文件，但是其中 file_name函数用于读取目录下的文件名，要使用到：

import os def file_name(path): return os.listdir(path) if __name__ == "__main__": path = 'C:\\Users\\...\\Desktop\\SeeTheData' dirs = file_name(path) for i in dirs: print(i)

 菜单选项的事件函数如下：

def Open_Folder(self): path = QFileDialog.getExistingDirectory(self, "选取文件夹", "./") self.tree = QTreeWidget() self.tree.setColumnCount(1) self.tree.setColumnWidth(0, 50) self.tree.setHeaderLabels(["EXPLORER"]) self.tree.setIconSize(Qt.QSize(25, 25)) self.tree.setSelectionMode(QAbstractItemView.ExtendedSelection) self.actionfileopen.triggered.connect(self.Open_Folder) dirs = fn(path) fileInfo = Qt.QFileInfo(path) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) root = QTreeWidgetItem(self.tree) root.setText(0,path.split('/')[-1]) root.setIcon(0,QtGui.QIcon(icon)) self.CreateTree(dirs, root, path) self.setCentralWidget(self.tree) QApplication.processEvents()

 生成文件选择对话框，并获得目录路径：

path = QFileDialog.getExistingDirectory(self, "选取文件夹", "./")

 创建QTreeWidget组件，并设置相关参数：

self.tree = QTreeWidget() self.tree.setColumnCount(1) self.tree.setColumnWidth(0, 50) self.tree.setHeaderLabels(["EXPLORER"]) self.tree.setIconSize(Qt.QSize(25, 25)) self.tree.setSelectionMode(QAbstractItemView.ExtendedSelection)

获取文件在系统下的图标：

fileInfo = Qt.QFileInfo(path) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo))

创建根节点：

root = QTreeWidgetItem(self.tree) root.setText(0,path.split('/')[-1]) root.setIcon(0,QtGui.QIcon(icon))

递归创建子节点的函数如下：

def CreateTree(self, dirs, root, path): for i in dirs: path_new = path + '\\' + i if os.path.isdir(path_new): fileInfo = Qt.QFileInfo(path_new) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) child = QTreeWidgetItem(root) child.setText(0,i) child.setIcon(0,QtGui.QIcon(icon)) dirs_new = fn(path_new) self.CreateTree(dirs_new, child, path_new) else: fileInfo = Qt.QFileInfo(path_new) fileIcon = Qt.QFileIconProvider() icon = QtGui.QIcon(fileIcon.icon(fileInfo)) child = QTreeWidgetItem(root) child.setText(0,i) child.setIcon(0,QtGui.QIcon(icon))

 os.path.isdir用于判断路径是文件还是文件夹。