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('请选择你要进行的任务：
1、添加字符
2、删除字符')
    if r == '1':
        add()
    elif r == '2':
        dele()
    else:
        print('输入错误！')