可以使用 Python 的 os 和 shutil 模块来实现整理文件的功能。
使用 os.walk() 遍历整个电脑的所有文件夹,获取所有文件的路径。
对于每个文件,使用 os.path.splitext() 函数获取文件的扩展名,并建立字典存储扩展名和文件的对应关系。
使用 shutil.move() 将文件移动到不同的文件夹中,文件夹名称为文件的扩展名。
以下是示例代码:
import os
import shutil
rootdir = "C:\\" # 需要整理文件的根目录
ext_dict = {} # 存储扩展名和文件的对应关系
# 遍历文件夹
for dirpath, dirnames, filenames in os.walk(rootdir):
for filename in filenames:
file_path = os.path.join(dirpath, filename)
ext = os.path.splitext(file_path)[1] # 获取文件扩展名
ext = ext.lower() # 将扩展名转换为小写
if ext not in ext_dict:
ext_dict[ext] = []
ext_dict[ext].append(file_path)
# 创建文件夹并移动文件
for ext, file_list in ext_dict.items():
if ext == "": # 忽略没有扩展名的文件
continue
ext_folder = os.path.join(rootdir, ext) # 创建扩展名对应的文件夹
if os.path.exists(ext_folder):
user_input = input(f"{ext_folder} 已存在,是否继续? (y/n)")
if user_input.lower() != "y":
continue
else:
os.makedirs(ext_folder)
for file_path in file_list:
try:
shutil.move(file_path, ext_folder)
except Exception as e:
print(f"Failed to move {file_path}:{e}")
print("成功按扩展名组织所有文件!")
代码在执行过程中会询问用户是否继续,当文件移动失败时会给出错误信息,最后会给出提示信息,告知用户文件整理操作已经完成。
不过,程序会将文件夹也移动到扩展名相同的文件夹中,如果不希望这样做,可以在遍历文件夹时判断文件类型,将文件夹和文件分开处理。
另外,还可以考虑将文件移动到指定的目录下,而不是根目录下,以防止误操作。