方便办公—文件整理

1.需求

整理电脑文件的时候，发现某个文件夹内右很多文件，各种类型的都有，逐个分类，易错且麻烦。该程序，可以将文件夹内的文件根据自己设定的匹配规则，一键实现分类。

2.代码实现

import os
import shutil

# 假设 source_dir 是通过某种方式获得的目录路径
source_dir = "D:\\test"  # 这里替换为你的源目录路径

# 子目录信息，这里将使用 source_dir 作为基础路径
# 自己定义匹配规则
sub_dirs = {
    ".txt": "文档",
    ".docx": "文档",
    ".xlsx": "表格",
    ".pptx": "演示文稿",
    ".jpg": "图片",
    ".mp3": "音频",
    ".mp4": "视频",
    ".png": "图片",
    ".wav": "视频"
}

# 确保所有子目录都存在
for subdir in sub_dirs.values():
    os.makedirs(os.path.join(source_dir, subdir), exist_ok=True)

# 遍历源目录中的所有文件
for filename in os.listdir(source_dir):
    file_path = os.path.join(source_dir, filename)
    if os.path.isfile(file_path):
        # 获取文件扩展名
        _, ext = os.path.splitext(filename)
        if ext in sub_