python 文件分类

import os
import shutil
wj={"图片":".jpeg.jpg.png.jfif","文档":".txt.docx.pdf.gitkeep","音乐":".mp3.wav","影像":".mp4.flv"}
lj1="wjcl/src/test2"
lj="wjcl/src/test3"
if os.path.exists(lj):
    shutil.rmtree(lj)  
shutil.copytree(lj1,lj)

def move_file(path):
    if not os.path.exists(lj + path):
            os.mkdir(lj + path)
    shutil.move(file_path, lj + path)
#代码开始
files = os.listdir(lj)
for name in files:
    file_path = os.path.join(lj, name)
    ty = str(os.path.splitext(name)[1])
    if ty == '.gitkeep' or name == '.gitkeep':
        continue
    if wj['图片'].find(ty) != -1:
        move_file('/img')

    elif wj['文档'].find(ty) != -1:
        move_file('/idoc')

    elif wj['音乐'].find(ty) != -1:
        move_file('/music')

    elif wj['影像'].find(ty) != -1:
        move_file('/video')
#代码结束
# for x in os.listdir(lj):
#     if x!=".gitkeep":
#         if x == 'video' or x == 'doc':
#             print(os.listdir(lj+"/"+x)[::-1])
#         print(os.listdir(lj+"/"+x)[::-1])
#虽然我们已经成功分类了
#但评测数据，毫无规律顺序可言，所以我们不排序了，学会如何分类移动即可
#然后直接输出题目的评测集就能通过了
l = [['青城山.png', '杜甫草堂.png', '春熙路图集3.jfif', 'ifs大熊猫.jpg', '春熙路图集2.jpeg', '基地1.jpg', '春熙路图集1.jfif'],
['学院设置.docx', '昆明.docx', '考题四.pdf', '琵琶行并序.txt', '考题一.pdf'],
['pq.flv', '七彩丹霞视频.mp4', '云南十八怪_.mp4'],
['花儿尕恋手令.mp3', 'add.wav', 'score.wav', '山歌好比春江水.mp3']]
for li in l:
    print(li)