python 内置的os模块将文件夹下所有图片名称加上指定字符

# eg:将文件夹下所有图片名称加上'M'
def change_name(path_img):
global i
if  os.path.isfile(path_img):
file_path=os.path.split(path_img) # 分割出目录与文件
#print(file_path)
lists=file_path[1].split(separator) # 分割出文件与文件扩展名
file_ext=lists[-1]      # 取出后缀名(列表切片操作)
#print(file_ext)
img_ext=['bmp','jpeg','gif','psd','png','jpg']
if file_ext in img_ext:
os.rename(path_img,img_dir+'\\'+ lists[0]+add+separator+file_ext)
i=i+1

def init():
global img_dir,separator,add
img_dir=input('The directory of files to be processed:')
img_dir=os.path.abspath(img_dir)
print(img_dir)
separator=input('the separator to be operated:')
print(separator)
add=input('the symbols of you want to add:')
print(add)

if __name__ == '__main__':
i=0
init() # 初始化条件
img_file=os.listdir(img_dir) # 显示要操作的所有文件
for y in img_file:
print(y)
#print('切换目录前：',os.getcwd())
os.chdir(img_dir) # 切换到要处理的工作目录
print('切换到工作目录：',os.getcwd())  # 当前的工作目录

start=time.time()
for  img in img_file:
full_path=os.path.abspath(img) # 获得绝对路径
change_name(full_path)
count=time.time()-start
img_file=os.listdir(img_dir)
for c in img_file:
print(c)
print('程序运行耗时：%0.2f'%(count))
print('总共处理了%s张照片'%(i))