Python脚本,用于按扩展名组织文件并将其移动到不同的文件夹

可以使用 Python 的 os 和 shutil 模块来实现整理文件的功能。

  1. 使用 os.walk() 遍历整个电脑的所有文件夹,获取所有文件的路径。

  2. 对于每个文件,使用 os.path.splitext() 函数获取文件的扩展名,并建立字典存储扩展名和文件的对应关系。

  3. 使用 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("成功按扩展名组织所有文件!")

代码在执行过程中会询问用户是否继续,当文件移动失败时会给出错误信息,最后会给出提示信息,告知用户文件整理操作已经完成。

不过,程序会将文件夹也移动到扩展名相同的文件夹中,如果不希望这样做,可以在遍历文件夹时判断文件类型,将文件夹和文件分开处理。

另外,还可以考虑将文件移动到指定的目录下,而不是根目录下,以防止误操作。

你可能感兴趣的:(python,开发语言)