从零写一个Python文件批量整理器：自动按类型归档桌面文件

从零写一个Python文件批量整理器：自动按类型归档桌面文件

专栏：《从零写一个小工具》 | 语言：Python | 依赖：纯标准库

一、场景介绍

公司发的报表、浏览器下的安装包、临时截图、各种格式的文档……几个月不收拾，桌面和下载文件夹就能堆成山。手动拖文件分类？太费时间，而且拖不了几次就放弃了。

这篇文章带你从零写一个小工具，把指定文件夹里的文件按类型自动归类，图片进图片夹，文档进文档夹，可执行文件单独放，整理完还给你输出一份统计报告。代码不到100行，拷走就能跑。

二、需求说明

这个小工具要解决这几个实际问题：

1. 指定一个待整理目录（比如桌面或下载文件夹）
2. 按文件扩展名自动分类：.jpg/.png归图片，.docx/.pdf归文档，.exe/.msi归安装包，等等
3. 如果目标分类文件夹不存在，自动创建
4. 遇到同名文件不要覆盖，自动重命名保留
5. 整理完打印一份统计报告：每类多少文件，总共多少

不需要装任何第三方库，只用 Python 自带的os、shutil、pathlib。

三、环境准备

Python 3.7+ 就行，不用装任何东西。打开终端确认一下：

python--version

四、逐段代码编写

4.1 定义文件类型映射

先把扩展名和分类文件夹对应起来。这是整个工具的核心规则，你可以按自己习惯改：

CATEGORY_MAP={"图片":[".jpg",".jpeg",".png",".gif",".bmp",".webp",".svg"],"文档":[".doc",".docx",".pdf",".txt",".md",".xls",".xlsx",".ppt",".pptx",".csv"],"视频":[".mp4",".avi",".mov",".mkv",".flv",".wmv"],"音频":[".mp3",".wav",".flac",".aac",".ogg",".m4a"],"压缩包":[".zip",".rar",".7z",".tar",".gz",".bz2"],"安装包":[".exe",".msi",".dmg",".pkg",".deb",".rpm"],"代码":[".py",".js",".html",".css",".java",".c",".cpp",".go",".rs",".json",".xml",".yaml",".yml"],}

这里有个设计细节：用一个扩展名到分类的反向字典，后续查找更快，不用每次遍历：

EXT_TO_CATEGORY={}forcategory,extsinCATEGORY_MAP.items():forextinexts:EXT_TO_CATEGORY[ext.lower()]=category

为什么要转小写？Windows 上扩展名可能是.PDF或.pdf，统一转小写再匹配，避免漏网之鱼。

4.2 获取分类文件夹名称

传入一个文件路径，返回它该去哪个分类文件夹：

frompathlibimportPathdefget_category(file_path:Path)->str:"""根据扩展名返回分类文件夹名称"""ext=file_path.suffix.lower()# 获取扩展名并转小写returnEXT_TO_CATEGORY.get(ext,"其他")# 未知类型丢进"其他"

Path.suffix是pathlib的方法，直接拿到带点的扩展名，比如.jpg，比用字符串分割干净多了。

4.3 处理重名文件

同一个分类里可能已经有一个报告.pdf，再移一个报告.pdf过来，默认会覆盖。我们要保留两份：

defsafe_target_path(target_dir:Path,original_name:str)