import os
import shutil

# 下面這個(gè)map決定了該結(jié)尾的文件夾會(huì)去哪個(gè)目錄，只會(huì)移動(dòng)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('輸入你想整理的文件夾的絕對(duì)路徑：')   #'/Users/name/Downloads/' 必須末尾有斜杠哈
des_folder = input('輸入整理后文件放置的文件夾絕對(duì)位置：')  #'/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)  #如果擔(dān)心可以先測(cè)試copy一份到另一個(gè)地址下面
            except:
                if os.path.exists(des_path):
                    print("文件復(fù)制移動(dòng)失敗 fail,已經(jīng)存在重名的文件",file.title(),src_path,des_path)
                else:
                    print("文件復(fù)制移動(dòng)失敗 fail,其他錯(cuò)誤", file.title(), src_path, des_path)


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

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

sort_file(src_folder)