python：文件管理

Question

文件分类:
    假设某个路径下的文件杂乱不堪，那我们现在要把这些个文件按照文件的类型进行分类。

    那我们要把文档的这些个文件放到一个文件夹中
    创建"文档"文件夹，收录这些内容:excel文件，doc文件，ppt文件，txt文件
    创建"图片"文件夹，收录图片文件：png,jpg,gif
    创建"音频"文件夹，收录音频文件：mp3,wma,acc
    创建"视频"文件夹，收录视频文件：avi,mp4,mkv 
    创建"其他"文件夹，收录除此以外的其他文件

Answer 1

import os
import shutil

dirname = r'C:\Users\Administrator\Desktop\test'

def getClass(ext):
    res = [k for k, v in d.items() if ext in v]
    if res:
        return res[0]
    else:
        return '其他'
    
d = {'文档': ['xls', 'xlsx', 'doc', 'docx', 'ppt', 'pptx', 'txt'],
     '图片': ['png', 'jpg', 'jpeg', 'gif'],
     '音频': ['mp3', 'wma', 'acc'],
     '视频': ['avi', 'mp4', 'mkv'],
     }

for i in os.listdir(dirname):
    dir_file_name = os.path.join(dirname, i)
    if not os.path.isdir(dir_file_name):
        file_name, file_ext = os.path.splitext(i)
        dir_name = getClass(file_ext[1:])
        dir_path = os.path.join(dirname , dir_name)
        if not os.path.exists(dir_path) :
            os.mkdir(dir_path)
        shutil.move(dir_file_name,os.path.join(dir_path,i))