python-递归删除空文件夹、遍历目录文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import os
import tkinter as tk
from tkinter.filedialog import (askopenfilename,   # 请求打开一个文件名，用来选择需要打开的文件
                                askdirectory,   # 请求一个目录，并返回文件名，用来选择需要的文件夹
                                asksaveasfilename,   # 请求将文件名保存为
                                asksaveasfile   # 选择保存文件的路径和自定义文件名及文件类型进行保存
                                )

class Operate_Dir:

    def __init__(self):
        self.root = tk.Tk()
        self.root.withdraw()  # 用来关闭root窗口
        self.dir_path = askdirectory()

    def Remove_emptyDir(self):
        '''
        递归删除空的目录文件，os.walk提供了从内到外的遍历目录树的方法，设置topdown=False后，
        可以由内到外判断当前目录树下是否有文件和文件夹，如果都没有说明当前目录树为空文件夹，
        使用os.rmdir删除。
        '''
        for path,dirs,files in os.walk(self.dir_path, topdown=False):
            if not files and not dirs:  # 用来递归删除所有空的文件目录
                os.rmdir(path)

    def traverse_dir(self,topdown=True):
        '''
        遍历选择文件中的所有文件夹目录
        '''
        for path, dirs, files in os.walk(self.dir_path, topdown):
            print(path)

     def traverse_files(self,find=None,topdown=True):
        '''
        遍历文件下所有的文件，可以通过遍历出的文件全路径做查找包含自己需要的文件进行返回其路径
        find的值为 1 时，选择关键字搜索文件目录
        '''
        if find == 1:
            keyword = input('输入需要查找的目录或文件名中所包含的相关文字：')
            for path, dirs, files in os.walk(self.dir_path, topdown):
                files = os.listdir(path)
                for file in files:
                    file_path = path + '/' + file
                    if (keyword in file_path) and (os.path.isfile(file_path)):
                        print(file_path)
        else:
            for path, dirs, files in os.walk(self.dir_path, topdown):
                files = os.listdir(path)
                for file in files:
                    file_path = path + '/' + file
                    if os.path.isfile(file_path):
                        print(file_path)

if __name__ == '__main__':

    # Operate_Dir().traverse_files(find=1)   # 在指定文件下查找文件
    Operate_Dir().traverse_files()  # 遍历指定目录下的所有文件