Python 文件操作

import os
import time
def batch_rename(path):
    global img_num
    if not os.path.isdir(path) and not os.path.isfile(path):
        return False
    if os.path.isfile(path):
        file_path = os.path.split(path)
        file_info_lists = file_path[1].split('.')
        file_ext = file_info_lists[-1]
        img_ext = ['bmp', 'jpg', 'jpeg', 'png', 'gif']
        if file_ext in img_ext:
            new_name = file_path[0] + '/' + file_info_lists[0] 

                + 'tst-' + '.' + file_ext
            os.rename(path, new_name)
            img_num += 1
            print(f'rename:{path},to{new_name}')
    elif os.path.isdir(path):
        for item in os.listdir(path):
            batch_rename(os.path.join(path, item))
if name == 'main':
    img_dir = '/home/zhu/img/'
    img_num = 0
    for i in range(100):
        with open(img_dir + str(i) + '.jpg', 'w') as f:
            f.writelines('')
    print('-'  10 + 'Start' + str(time.time()) + '-'  10)
    batch_rename(img_dir)
    print(f'已处理{img_num}张图片')
    print('-'  10 + 'end' + str(time.time()) + '-'  10)