在日常开发或文件管理时,经常遇到大量文件需要统一命名或按类型归档的场景。Python 内置的 os 和 shutil 模块提供了文件操作接口,可以快速编写脚本完成批量处理。下面结合具体函数参数和逻辑,实现两个实用功能:批量重命名(支持统一前缀编号和关键词替换)以及智能文件分类。
## 批量重命名
### 1. 统一命名规则
以下函数将指定目录下的所有文件按修改时间排序后,统一重命名为“前缀_编号.扩展名”格式。
- import os
- def batch_rename(directory, prefix="file", start=1, digits=3):
- """
- 统一命名:prefix_001.ext, prefix_002.ext ...
- directory: 文件夹路径
- prefix: 文件名前缀
- start: 起始编号
- digits: 编号位数(不足补零)
- """
- files = [f for f in os.listdir(directory)
- if os.path.isfile(os.path.join(directory, f))]
- # 按修改时间排序,保持顺序一致性
- files.sort(key=lambda x: os.path.getmtime(
- os.path.join(directory, x)))
- for i, filename in enumerate(files):
- name, ext = os.path.splitext(filename)
- new_name = f"{prefix}_{str(start + i).zfill(digits)}{ext}"
- old_path = os.path.join(directory, filename)
- new_path = os.path.join(directory, new_name)
- os.rename(old_path, new_path)
- print(f"{filename} → {new_name}")
复制代码
参数说明:
- `directory`:目标文件夹路径,需要绝对路径或相对当前工作目录的相对路径。
- `prefix`:新文件名的前缀字符串,默认 "file"。
- `start`:起始序号,默认从 1 开始。
- `digits`:序号部分的位数,例如 digits=3 会生成 001、002 ...,不足位时自动补零。
调用示例:`batch_rename("/path/to/files", prefix="photo", start=1, digits=4)` 会将文件重命名为 photo_0001.jpg、photo_0002.jpg 等。
### 2. 替换文件名中的关键词
如果只是想批量替换文件名中的一段文字(例如将“草稿”改为“正式版”),可使用以下函数:
- def replace_text(directory, old_text, new_text):
- """批量替换文件名中的关键词"""
- count = 0
- for f in os.listdir(directory):
- fp = os.path.join(directory, f)
- if os.path.isfile(fp) and old_text in f:
- new_name = f.replace(old_text, new_text)
- os.rename(fp, os.path.join(directory, new_name))
- count += 1
- print(f"共替换 {count} 个文件")
复制代码
该函数会遍历目录下所有普通文件(跳过子目录),检测文件名是否包含 `old_text`,若包含则将其替换为 `new_text`。注意替换操作会原地修改文件名,不会改变扩展名。
## 智能文件归档
智能文件分类的目标是将不同扩展名的文件移动到对应的分类目录中。首先定义一个扩展名到分类名称的映射字典:
- EXTENSION_MAP = {
- ".jpg": "图片", ".jpeg": "图片", ".png": "图片", ".gif": "图片",
- ".doc": "文档", ".docx": "文档", ".pdf": "文档", ".txt": "文档",
- ".xls": "表格", ".xlsx": "表格",
- ".zip": "压缩包", ".rar": "压缩包", ".7z": "压缩包",
- ".mp4": "视频", ".avi": "视频",
- ".mp3": "音乐", ".wav": "音乐",
- ".py": "代码", ".java": "代码", ".js": "代码", ".html": "代码",
- }
复制代码
然后通过 `auto_sort` 函数完成实际移动操作:
- import shutil
- def auto_sort(directory):
- """按文件类型自动归档到子文件夹"""
- for f in os.listdir(directory):
- fp = os.path.join(directory, f)
- if not os.path.isfile(fp):
- continue
- _, ext = os.path.splitext(f)
- folder = EXTENSION_MAP.get(ext.lower(), "其他")
- target = os.path.join(directory, folder)
- os.makedirs(target, exist_ok=True)
- shutil.move(fp, os.path.join(target, f))
复制代码
关键点:
- 使用 `os.path.splitext` 获取扩展名,并转为小写以匹配字典。
- 若扩展名不在映射字典中,则归类到“其他”文件夹。
- `os.makedirs(target, exist_ok=True)` 确保目标文件夹存在,如果已存在不会报错。
- `shutil.move` 将文件移动到对应分类子目录,原文件被移除。
调用 `auto_sort("/path/to/messy_folder")` 后,文件夹中所有文件会按类型整理到图片、文档、表格、压缩包、视频、音乐、代码、其他等子目录中。
## 注意事项
- 重命名和移动操作不可逆,建议先在小范围内测试。
- 如果文件名包含特殊字符或路径包含空格,`os.rename` 和 `shutil.move` 均能正常处理(Windows 下需注意路径长度限制)。
- 统一重命名时按修改时间排序,适合将按时间顺序的文件编号,若需要按文件名排序可修改 `sort` 的 key。
以上两个脚本可以独立使用,也可以组合调用:先重命名再归档,或者归档后再对分类结果重命名。Python 的 os 和 shutil 模块足够完成大部分日常文件整理需求,修改映射字典即可适配更多文件类型。 |