python文件分类归档

import os

# 将 Python 移至目标目录
os.chdir('/Users/zhangsan/Desktop/资料')

# 构造扩展名和类型的对应字典
genre = {
    '文档':['.doc', '.docx', '.xls', '.xlsx', '.txt', '.pdf'],
    '图片':['.jpg', '.png', '.gif'],
    '音频':['.mp3', '.aac', '.wav'],
    '压缩包':['.zip', '.rar' ,'.7z']
}

# 构造函数，通过扩展名查类型
def find_genre(extension):
    for key, value in genre.items():
        if extension in value:
            return key

# 为四种类型创建四个目录
for i in genre.keys():
    os.mkdir(i)

files = os.listdir()

# 对当前目录的每个文件进行操作
for file in files:
    # 提取出扩展名
    extension = os.path.splitext(file)[1]
    # 判断是否成功提取出扩展名
    if extension != '':
        # 判断当前文件的类型，确定移动的目标目录
        target_folder = find_genre(extension)
        # 根据文件类型，构造文件的新路径
        new_file_path = os.path.join(target_folder, file)
        # 移动文件
        os.rename(file, new_file_path)