Python学习 —— 桌面文件分类

import os
import shutil

# 找到桌面路径
desktop = os.path.join(os.path.expanduser("~"),'Desktop')
print(desktop)

# 如果文件不存在，新建文件夹
name = input('输入文件夹的名字:')
new_folder = os.path.join(desktop,name)
print(new_folder)

if os.path.exists(new_folder) == False:
    os.mkdir(new_folder)
else:
    print("文件夹已存在")

# 获取桌面所有文件
name_list = os.listdir(desktop)    
for file in name_list:
    # 判断是不是文件，如果是文件则进行分割
    filepath = os.path.join(desktop,file)
    if not os.path.isfile(filepath):
        continue
    elif os.path.isfile(filepath):
        # 分割
        fileExpand = os.path.splitext(file)[1]
        fileExpand = fileExpand[1:] 
        # 以文件后缀，新建文件夹
        expand_file_name = os.path.join(new_folder,fileExpand)
        if not os.path.exists(expand_file_name):
            os.mkdir(expand_file_name)
        
        # 移动文件
        shutil.move(filepath,expand_file_name)