快速整理文档的python脚本

import os
import shutil

# 下面这个map决定了该结尾的文件夹会去哪个目录，只会移动map有的文件后缀，其他文件不做操作
fileSuffixDict={'xlsx': 'excel', 'csv': 'excel',
                'jpg':'picture','jpeg':'picture','picture':'picture',
                "zip":"zip","dmg":'dmg','pdf':'pdf','txt':'txt',
                'js':'js','docx':'docx','sql':'sql',
                'py':'py','go':'go'}

src_folder = input('输入你想整理的文件夹的绝对路径：')   #'/Users/name/Downloads/' 必须末尾有斜杠哈
des_folder = input('输入整理后文件放置的文件夹绝对位置：')  #'/Users/name/Downloads/'


def sort_file(name):
    files = os.listdir(name)
    print('文件整理中...',name)
    for file in files:
        src_path = name + file
        if os.path.isfile(src_path):
            fileSuffix = file.split('.')[-1]
            if fileSuffix in fileSuffixDict:
                fileSuffix = fileSuffixDict[fileSuffix]
            else:
                continue
            des_path = des_folder + fileSuffix
            if not os.path.exists(des_path):
                os.makedirs(des_path)
            try:
                shutil.move(src_path, des_path)
                # shutil.copy(src_path, des_path)  #如果担心可以先测试copy一份到另一个地址下面
            except:
                if os.path.exists(des_path):
                    print("文件复制移动失败 fail,已经存在重名的文件",file.title(),src_path,des_path)
                else:
                    print("文件复制移动失败 fail,其他错误", file.title(), src_path, des_path)


        else:
            print("子文件整理开始")
            sort_file(src_path+'/')
            print("子文件整理完毕")

    print('文件整理完毕！')

sort_file(src_folder)