python实现批量修改文件名

 import os
 def dele():
     # 设置一个计数器
     n=0
     st = input('请输入你要删除的字符：')
     for i in f:
         b = f[n]

         if st in b:
             oldname= path+b
             s = b.strip().split(st)
             newname = path+s[0]+'.jpg'
             if os.path.exists(newname):
                 print('删除前' + newname)
                 os.remove(newname)
                 print('删除后'+newname)
                 os.rename(oldname, newname)
                 # break
             else:
                 os.rename(oldname,newname)
                 print(oldname+'=========>'+newname+'修改成功！')
         else:
             print(st+"不存在")
         n+=1
 def add():
     n = 0
     st = input('请输入你要添加的后缀：')
     for i in f:
         b = f[n]
         oldname = path + b
         s = b.strip().split('.jpg')
         newname = path + s[0] +st+'.jpg'
         if os.path.exists(newname):
             print('发现新重复货号' + newname + oldname)
             os.remove(newname) #删除文件

             os.rename(oldname, newname)
             # break
         else:
             os.rename(oldname, newname)#用新的文件名替换掉老的的文件名
             print(oldname + '=========>' + newname + '修改成功！')
         n += 1

 if __name__ == '__main__':
     path = input('输入文件路径：')
     path = path + '\\'
     # 获取该目录下的所有文件，存入列表中
     f = os.listdir(path)
     # 执行删除函数
     r = input('请选择你要进行的任务：\n1、添加字符\n2、删除字符')
     if r == '':
         add()
     elif r == '':
         dele()
     else:
         print('输入错误！')