整理目录下的文件并进行分类-python

[rungod]

代码： 全选

import os
import shutil

def organize_files(directory):
    # 定义文件类型的分类
    file_types = {
        "Images": [".jpeg", ".jpg", ".png", ".gif", ".bmp"],
        "Documents": [".doc", ".docx", ".pdf", ".txt", ".xls", ".xlsx", ".ppt", ".pptx"],
        "Videos": [".mp4", ".avi", ".mkv", ".mov"],
        "Music": [".mp3", ".wav", ".flac", ".aac"],
        "Rar":[".rar",".zip"]
        # 可以添加更多类型
    }
    
    # 遍历目标目录中的所有文件和文件夹
    for filename in os.listdir(directory):
        file_ext = os.path.splitext(filename)[1].lower()
        
        # 检查文件扩展名是否属于定义的类型之一
        for folder, extensions in file_types.items():
            if file_ext in extensions:
                # 创建对应的文件夹（如果不存在）
                folder_path = os.path.join(directory, folder)
                if not os.path.exists(folder_path):
                    os.makedirs(folder_path)
                
                # 移动文件到对应的文件夹
                src_file = os.path.join(directory, filename)
                dest_file = os.path.join(folder_path, filename)
                shutil.move(src_file, dest_file)
                print(f"Moved {filename} to {folder}")
                break
                
if __name__ == "__main__":
    target_directory = input("请输入要整理的目录路径: ")
    organize_files(target_directory)