python文件夹下多个子文件夹,对图片批量修改格式和名称

之前写了一篇在一个文件夹下处理图片的博客,点这里.
发现数据集太多了,处理起来非常耗时麻烦,继续学习了一下多个子文件夹的图片处理
如图

原来的数据集甚至有gif格式,而且命名重复(比如1.jpg 和1.jpeg 和 1.png)
这些都会导致出bug
我的做法是:
首先一个个文件夹打开,按类型查看,
1.看到gif,删除
2.看到jpeg, png, 全选这些类型的图片,随便乱起一个名字
(这步是为了防止代码生成的名字出现重复
比如,1.jpeg->1.jpg, 若此前已经存在了1.jpg就会报错)

然后套用一下代码

# -*- coding: utf-8 -*-
"""
Created on Mon May 18 17:42:17 2020

@author: 吴
"""


#批量修改文件名
#批量修改图片文件名
import os
import re
import sys
from PIL import Image
import string

PATH = r"C:/Users/吴/Desktop/学习快速通道/可回收垃圾"
def renameall(root):
    
    #待修改文件夹
    fileList = os.listdir(root)
    #输出文件夹中包含的文件        
    print("修改前："+str(fileList))     
    #得到进程当前工作目录
    currentpath = os.getcwd()   
    #将当前工作目录修改为待修改文件夹的位置    
    os.chdir(root)
    num=1       #名称变量
    for fileName in fileList:       #遍历文件夹中所有文件
        pat=".+\.(jpg|png|pgm|jpeg|JPG)"     #匹配文件名正则表达式
        pattern = re.findall(pat,fileName)      #进行匹配
        os.rename(fileName,(str(num)+'.'+pattern[0]))       #文件重新命名
        num = num+1     #改变编号，继续下一项
    print("---------------------------------------------------")
    os.chdir(currentpath)       #改回程序运行前的工作目录
    sys.stdin.flush()       #刷新
    print("修改后："+str(os.listdir(root)))       #输出修改后文件夹中包含的文件


for fileName in os.listdir(PATH):
    # print(filename)
    dirName =  PATH + '/'+ fileName + '//'
    # print(dirName)
    li=os.listdir(dirName)
    for filename in li:
        newname = filename
        newname = newname.split(".")
        if newname[-1]=="jpeg":
            newname[-1]="jpg"
            newname = str.join(".",newname)  
            filename = dirName+filename
            newname = dirName+newname
            os.rename(filename,newname)
        elif newname[-1] == 'png':
            newname[-1]="jpg"
            newname = str.join(".",newname)  
            filename = dirName+filename
            newname = dirName+newname
            os.rename(filename,newname)
    
        # print(newname,"updated successfully")
    root = PATH + '/'+ fileName
    print(root)
    renameall(root)