# 不同文件組成的嵌套字典
file_dict = {
            '圖片': ['jpg','png','gif','webp'],
            '視頻': ['rmvb','mp4','avi','mkv','flv'],
            "音頻": ['cd','wave','aiff','mpeg','mp3','mpeg-4'],
            '文檔': ['xls','xlsx','csv','doc','docx','ppt','pptx','pdf','txt'],
            '壓縮文件': ['7z','ace','bz','jar','rar','tar','zip','gz'],
            '常用格式': ['json','xml','md','ximd'],
            '程序腳本': ['py','java','html','sql','r','css','cpp','c','sas','js','go'], 
            '可執(zhí)行程序': ['exe','bat','lnk','sys','com'],
            '字體文件': ['eot','otf','fon','font','ttf','ttc','woff','woff2']
        }

# 導(dǎo)入相關(guān)庫(kù)
import os
import glob
import shutil

# 采用input()函數(shù)，動(dòng)態(tài)輸入要處理的文件路徑。
path = input("請(qǐng)輸入要清理的文件路徑：")

# 定義一個(gè)文件字典，不同的文件類型，屬于不同的文件夾，一共9個(gè)大類。
file_dict = {
            '圖片': ['jpg','png','gif','webp'],
            '視頻': ['rmvb','mp4','avi','mkv','flv'],
            "音頻": ['cd','wave','aiff','mpeg','mp3','mpeg-4'],
            '文檔': ['xls','xlsx','csv','doc','docx','ppt','pptx','pdf','txt'],
            '壓縮文件': ['7z','ace','bz','jar','rar','tar','zip','gz'],
            '常用格式': ['json','xml','md','ximd'],
            '程序腳本': ['py','java','html','sql','r','css','cpp','c','sas','js','go'], 
            '可執(zhí)行程序': ['exe','bat','lnk','sys','com'],
            '字體文件': ['eot','otf','fon','font','ttf','ttc','woff','woff2']
        }

# 定義一個(gè)函數(shù)，傳入每個(gè)文件對(duì)應(yīng)的后綴。判斷文件是否存在于字典file_dict中；
# 如果存在，返回對(duì)應(yīng)的文件夾名；如果不存在，將該文件夾命名為"未知分類"；
def func(suffix):
    for name, type_list in file_dict.items():
        if suffix.lower() in type_list:
            return name
    return "未知分類"

# 遞歸獲取 "待處理文件路徑" 下的所有文件和文件夾。
for file in glob.glob(f"{path}/**/*",recursive=True):
 # 由于我們是對(duì)文件分類，這里需要挑選出文件來(lái)。
    if os.path.isfile(file):
     # 由于isfile()函數(shù)，獲取的是每個(gè)文件的全路徑。這里再調(diào)用basename()函數(shù)，直接獲取文件名；
        file_name = os.path.basename(file)
        suffix = file_name.split(".")[-1]
        # 判斷 "文件名" 是否在字典中。
        name = func(suffix)
        #print(func(suffix))
        # 根據(jù)每個(gè)文件分類，創(chuàng)建各自對(duì)應(yīng)的文件夾。
        if not os.path.exists(f"{path}\\{name}"):
            os.mkdir(f"{path}\\{name}")
        # 將文件復(fù)制到各自對(duì)應(yīng)的文件夾中。
        shutil.copy(file,f"{path}\\{name}")