文件夹下图片批量重新命名

import os
class BatchRename():
    """
    批量重命名文件夹中的图片文件
    """
    def __init__(self,path):
        self.path = path
    def rename(self):
        filelist = os.listdir(self.path)
        total_num = len(filelist)
        i = 1
        for item in filelist:
            Suffix_name = ['.png', '.jpg', '.jpeg', '.tif','.bmp']
            if item.endswith(tuple(Suffix_name)):
                n = 6 - len(str(i))
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), str(0) * n + str(i) + '.jpg')  #TR代表训练集train,NW代表no water
                try:
                    os.rename(src, dst)
                    print('converting %s--to-->%s' % (src, dst))
                    i = i + 1
                except:
                    continue
        print('total %d to rename & converted %d jpgs' % (total_num, i))

if __name__ == '__main__':
    #path = "D:/Program Files/caffe/caffe-master/caffe-master/examples/transfer-water/data/mydata/addwater/"        # 图片文件夹路径
    path = "D:/picture/test1/test/num/"
    #savePath = "D:/picture/samplesDatasets/water/nowater/"
    demo = BatchRename(path).rename()