python办公自动化1之文件自动分类（根据后缀名）[python+ pyside2图形化界面实现]

需求分析

选择文件夹下的所有文件，根据文件的后缀名自动将文件分类。

用到的库

import os
import  glob   #文件遍历库
import  shutil #文件操作

知识点

# 复制文件
shutil.copy(file,ext_dir)
#分离文件名与扩展名 list list[1]是扩展名
os.path.splitext(file)    

# pyside2图形化界面选择文件夹并获得选择的路径
directory = QtWidgets.QFileDialog.getExistingDirectory(None, "选取文件夹", "D:/")  
#pyside设置文本框文本
self.ui.label.setText("已选择文件夹："+directory)
#pyside2获取富文本文本
last_text =self.ui.textEdit.toPlainText()
#pyside弹窗
QMessageBox.warning(self.mainWindow,'错误', '请选择文件夹')

核心代码

from PySide2 import QtWidgets
from PySide2.QtWidgets import QMainWindow, QMessageBox
from ui_main import Ui_MainWindow
import os
import  shutil #文件操作

class service():
    def __init__(self):
        # 创建主窗口对象
        self.mainWindow = QMainWindow()
        # 创建UI对象
        self.ui = Ui_MainWindow()
        # 向主窗口上添加控件
        self.ui.setupUi(self.mainWindow)
        self.mainWindow.show()

        # 链接信号与槽
        self.ui.btnfile1.clicked.connect(self.btnClickFile1) # 选择分类文件夹
        self.ui.btnfile2.clicked.connect(self.btnClickFile2)  # 选择保存文件夹
        self.ui.btnRun.clicked.connect(self.runClassify)      #点击运行


    #弹出文件选择框1
    def btnClickFile1(self):
        directory = QtWidgets.QFileDialog.getExistingDirectory(None, "选取文件夹", "D:/")  # 起始路径
        self.old_dir=directory
        self.ui.label.setText("已选择文件夹："+directory)

    # 弹出文件选择框2
    def btnClickFile2(self):
        directory = QtWidgets.QFileDialog.getExistingDirectory(None, "选取文件夹", "D:/")  # 选择分类后保存路径
        self.new_dir = directory  # 分类后保存的文件夹
        self.ui.label_2.setText("已选择文件夹：" + directory)

    #执行分类
    def runClassify(self):
        try:
            self.ui.textEdit.setText("开始进行文件自动分类..")
            file_list =self.get_file_list(self.old_dir)
            for file in file_list :
                #判断是否是文件,如果是文件才执行文件操作
                if os.path.isfile(file):
                    #获取文件扩展名
                    ext=os.path.splitext(file)[-1]
                    #根据扩展名创建新目录
                    ext=ext[1:]
                    ext_dir =self.new_dir + "/"+ext
                    if not os.path.exists(ext_dir):
                        os.makedirs(self.new_dir + "/"+ext)
                    # 复制文件
                    shutil.copy(file,ext_dir)
            #添加已完成的信息
            last_text = self.ui.textEdit.toPlainText()
            self.ui.textEdit.setText(last_text+"\n已完成分类操作")
        except Exception as err:
            QMessageBox.warning(self.mainWindow, '错误', str(err))

    #递归遍历获取所有文件
    def  get_file_list(self,path, all_files = []):
        # 首先遍历当前目录所有文件及文件夹
        file_list = os.listdir(path)
        # 准备循环判断每个元素是否是文件夹还是文件，是文件的话，把名称传入list，是文件夹的话，递归
        for file in file_list:
            # 利用os.path.join()方法取得路径全名，并存入cur_path变量，否则每次只能遍历一层目录
            cur_path = os.path.join(path, file)
            # 判断是否是文件夹
            if os.path.isdir(cur_path):
                self.get_file_list(cur_path, all_files)
            else:
                all_files.append(path+"/"+file)
        return all_files

效果演示