【python实用小脚本-113】Python文件自动分类整理工具技术文档

项目概述

本项目是一个Python脚本，用于自动整理和分类计算机中的文件。该工具可以扫描当前目录下的所有文件，并根据文件扩展名将它们移动到相应的文件夹中。例如，所有图片文件将被移动到"Images"文件夹，所有文档文件将被移动到"Docs"文件夹，以此类推。对于无法识别的文件类型，工具会将它们移动到"Others"文件夹。

核心功能

1. 文件分类系统

脚本定义了一个名为DIRECTORIES的字典，其中包含了各种文件类型及其对应的文件夹名称。这个字典是整个分类系统的基础，它将文件扩展名映射到相应的目录。

DIRECTORIES = {
    "Webpages": [".html5", ".html", ".htm", ".xhtml", ".aspx", ".php"],
    "Images": [".jpg", ".jpeg", ".png", ".tiff", ".gif", ".bmp", ".bpg", ".svg",
               ".heif", ".psd"],
    "Videos": [".avi", ".mp4", ".flv", ".mkv", ".wmv", ".mov", ".webm", ".vob",
               ".3gp", ".mpeg", ".mpg", ".qt"],
    "Docs": [".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".epub",
             ".opus", ".txt", ".in", ".out", ".xml"],
    "Archive": [".rar", ".zip", ".7z"],
    "Audio": [".mp3", ".aac", ".ogg", ".m4a", ".wav", ".aa", ".dvf", ".m4b",
              ".m4p", ".msv", ".oga", ".raw", ".vox"]
}

2. 文件扩展名映射

为了更方便地进行文件分类，脚本创建了一个FILE_FORMATS字典，将每个文件扩展名映射到其对应的目录名称。这个字典是通过遍历DIRECTORIES字典自动生成的。

FILE_FORMATS = {file_format: directory
                for directory, file_formats in DIRECTORIES.items()
                for file_format in file_formats}

3. 文件整理功能

organize_junk()函数是脚本的核心功能，它执行以下操作：

1. 使用os.scandir()扫描当前目录下的所有条目。
2. 对于每个条目，检查它是否是文件（跳过目录）。
3. 获取文件的扩展名，并将其转换为小写以便于比较。
4. 根据FILE_FORMATS字典确定文件应该被移动到哪个目录。
5. 如果目标目录不存在，则创建它。
6. 将文件移动到相应的目录。
7. 尝试删除空目录（如果存在）。

def organize_junk():
    for entry in os.scandir():
        if entry.is_dir():
            continue

        file_path = Path(entry)
        file_format = file_path.suffix.lower()

        if file_format in FILE_FORMATS:
            directory_path = Path(FILE_FORMATS[file_format])
            directory_path.mkdir(exist_ok=True)
            file_path.rename(directory_path.joinpath(file_path))
        else:
            directory_path = Path('Others')
            directory_path.mkdir(exist_ok=True)
            file_path.rename(directory_path.joinpath(file_path))

        for dir in os.scandir():
            try:
                os.rmdir(dir)
            except Exception:
                pass

技术实现细节

1. 使用pathlib库

脚本使用pathlib库中的Path类来处理文件路径。pathlib提供了面向对象的路径操作方式，比传统的os.path更加直观和易用。

from pathlib import Path

file_path = Path(entry)
directory_path = Path(FILE_FORMATS[file_format])

2. 文件扩展名处理

脚本通过file_path.suffix.lower()获取文件扩展名，并将其转换为小写，以确保扩展名比较时不区分大小写。

file_format = file_path.suffix.lower()

3. 目录创建与文件移动

使用Path.mkdir(exist_ok=True)创建目录，如果目录已存在则不会抛出异常。使用Path.rename()方法将文件移动到目标目录。

directory_path.mkdir(exist_ok=True)
file_path.rename(directory_path.joinpath(file_path))

4. 空目录删除

在文件移动完成后，脚本尝试删除可能产生的空目录。这是通过遍历当前目录并尝试删除每个子目录实现的。

for dir in os.scandir():
    try:
        os.rmdir(dir)
    except Exception:
        pass

使用说明

1. 将脚本保存为.py文件，例如organize_files.py。
2. 将脚本放在需要整理的目录中。
3. 运行脚本：python organize_files.py。
4. 脚本将自动扫描当前目录下的所有文件，并根据文件扩展名将它们移动到相应的文件夹中。

扩展与定制

1. 添加新的文件类型

要添加新的文件类型，只需在DIRECTORIES字典中添加新的条目。例如，要添加"Markdown"文件类型：

DIRECTORIES = {
    # ...其他目录
    "Markdown": [".md", ".markdown"],
}

2. 修改现有分类

如果需要修改某个文件类型的分类，只需在DIRECTORIES字典中调整相应的条目。例如，将.txt文件从"Docs"分类移动到"Others"分类：

DIRECTORIES = {
    # ...其他目录
    "Docs": [".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".epub",
             ".opus", ".in", ".out", ".xml"],
    "Others": [".txt"],  # 添加这一行
}

3. 自定义目标目录名称

如果需要自定义目标目录的名称，只需修改DIRECTORIES字典中的键名。例如，将"Webpages"改为"Web Files"：

DIRECTORIES = {
    "Web Files": [".html5", ".html", ".htm", ".xhtml", ".aspx", ".php"],
    # ...其他目录
}

注意事项

1. 备份重要文件：在运行脚本之前，建议备份重要文件，以防意外情况发生。
2. 权限问题：确保脚本有足够的权限来移动文件和创建目录。
3. 文件冲突：如果目标目录中已存在同名文件，Path.rename()可能会覆盖现有文件或抛出异常。在实际使用中，可能需要添加额外的逻辑来处理文件冲突。

总结

本项目提供了一个简单而实用的Python脚本，用于自动整理和分类计算机中的文件。通过定义文件类型与目录的映射关系，脚本可以高效地将文件移动到相应的文件夹中，从而帮助用户保持文件系统的整洁和有序。该工具易于使用和扩展，可以根据个人需求进行定制，是一个非常实用的文件管理辅助工具。

源码获取

完整代码已开源，包含详细的注释文档：
🔗 [GitCode仓库] https://gitcode.com/laonong-1024/python-automation-scripts
📥 [备用下载] https://pan.quark.cn/s/654cf649e5a6 提取码：f5VG