Python 磁盘垃圾文件清理器

磁盘垃圾文件清理器

实验目的：

1. 熟练运用标准库 os 和 os.path 中的函数。

2. 理解 sys 库中 argv 成员用法。

3. 理解 Python 程序接收命令行参数的方式。

4. 理解递归遍历目录树的原理。

5. 了解从命令提示符环境运行 Python 程序的方式。

实验内容：

编写程序，实现磁盘垃圾文件清理功能。要求程序运行时，通过命令行参数指定要清理的文件夹，然后删除该文件夹及其子文件夹中所有扩展名为 tmp、log、obj、txt 以及大小为 0 的文件

实验解题：

创建测试文件

import os

os.chdir("cleartest")

f = open('zero.doc', "w")  # 创建空文件
f.close()

list_file = ['tmp', 'log', 'obj', 'txt']
for i in list_file:
    filename = "{0}file.{0}".format(i)
    f = open(filename, "w")
    f.write("写一些东西不至于0KB")
    f.close()

磁盘垃圾文件清理程序

import re,os

def find_file(directory, depth=-1, re_pattern=None, onerror=print) -> list:
    """
    查找文件，并进行筛选。
    - `directory`:目标目录。
    - `depth`: 查找文件树的最大深度。如果值为负数，则无限深度。
    - `re_pattern`: 根据正则表达式筛选文件名。
    - `onerror`: 一个可调用的参数，用于记录报错。
    """
    if not os.path.isdir(directory):
        raise ValueError("{} 文件夹路径不存在".format(directory))

    try:
        file_list = os.listdir(directory)
    except PermissionError as e:
        onerror("您没有打开{}的权限".format(e))
        return -1

    path_list = []
    for filename in file_list:
        path = os.path.join(directory, filename)
        # 发现零大小文件立刻删除
        if os.path.getsize(path) == 0:
            os.remove(path)
            print("已删除大小为 0 的{}".format(path))
        if depth != 0 and os.path.isdir(path):
            sub_list = find_file(path, depth - 1, re_pattern, onerror)
            if sub_list != -1:
                path_list.extend(sub_list)
            continue
        # 正则表达式匹配
        if re_pattern and not re.findall(re_pattern, filename):
            continue
        path_list.append(path)
    return path_list


# 使用正则表达式 .*.tmp|.*.obj|.*.txt|.*.log 找出拓展名为 tmp、log、obj、txt的文件
for i in find_file(directory=position, re_pattern='.*.tmp|.*.obj|.*.txt|.*.log'):
    os.remove(i)
    print("已删除{}类型文件{}".format(i[-3:],i))

命令行接收参数

import sys

position = sys.argv[1]  # 获取命令行参数

命令行运行

PS ***\exp04\T02> python clearru.py cleartest
已删除大小为 0 的cleartest\zero.doc
已删除log类型文件cleartest\logfile.log
已删除obj类型文件cleartest\objfile.obj
已删除tmp类型文件cleartest\tmpfile.tmp
已删除txt类型文件cleartest\txtfile.txt

程序运行截图