import os
import shutil
import tkinter.messagebox as tm
from tkinter import Tk

Tk().withdraw()
src = r'C:\Users\MTC\Desktop' #改為自己的桌面文件地址
os.chdir(src)

if not (os.path.exists('Word')):
    os.makedirs("Word")

if not (os.path.exists('Pdf')):
    os.makedirs("Pdf")

if not (os.path.exists('Excel')):
    os.makedirs("Excel")

if not (os.path.exists('PPT')):
    os.makedirs("PPT")

if not (os.path.exists('Picture')):
    os.makedirs("Picture")

if not (os.path.exists('CNKI')):
    os.makedirs("CNKI")

if not (os.path.exists('Others')):
    os.makedirs("Others")


for i in os.listdir("."):
    file_type = os.path.splitext(i)[1]
    if file_type=='.docx' or file_type=='.doc':
        shutil.move(os.path.join(src, i), os.path.join((src + '\\Word'), i))
    elif file_type=='.pdf':
        shutil.move(os.path.join(src, i), os.path.join((src + '\\Pdf'), i))
    elif file_type=='.xlsx' or file_type=='.xls':
        shutil.move(os.path.join(src, i), os.path.join((src + '\\Excel'), i))
    elif file_type=='.ppt' or file_type=='.pptx':
        shutil.move(os.path.join(src, i), os.path.join((src + '\\PPT'), i))
    elif file_type in ['.jpg','.png','tiff','.bmp','.gif']:
        shutil.move(os.path.join(src, i), os.path.join((src + '\\Picture'), i))
    elif file_type in ['.caj']:
        shutil.move(os.path.join(src, i), os.path.join((src + '\\CNKI'), i))
    elif file_type !='':
        shutil.move(os.path.join(src, i), os.path.join((src + '\\Others'), i))
    else:
        pass

tm.showinfo("狀態(tài)","已經(jīng)結(jié)束")