python实现批量修改文件名

 1 import os
 2 def dele():
 3     # 设置一个计数器
 4     n=0
 5     st = input('请输入你要删除的字符：')
 6     for i in f:
 7         b = f[n]
 8 
 9         if st in b:
10             oldname= path+b
11             s = b.strip().split(st)
12             newname = path+s[0]+'.jpg'
13             if os.path.exists(newname):
14                 print('删除前' + newname)
15                 os.remove(newname)
16                 print('删除后'+newname)
17                 os.rename(oldname, newname)
18                 # break
19             else:
20                 os.rename(oldname,newname)
21                 print(oldname+'=========>'+newname+'修改成功！')
22         else:
23             print(st+"不存在")
24         n+=1
25 def add():
26     n = 0
27     st = input('请输入你要添加的后缀：')
28     for i in f:
29         b = f[n]
30         oldname = path + b
31         s = b.strip().split('.jpg')
32         newname = path + s[0] +st+'.jpg'
33         if os.path.exists(newname):
34             print('发现新重复货号' + newname + oldname)
35             os.remove(newname) #删除文件
36 
37             os.rename(oldname, newname)
38             # break
39         else:
40             os.rename(oldname, newname)#用新的文件名替换掉老的的文件名
41             print(oldname + '=========>' + newname + '修改成功！')
42         n += 1
43 
44 
45 if __name__ == '__main__':
46     path = input('输入文件路径：')
47     path = path + '\\'
48     # 获取该目录下的所有文件，存入列表中
49     f = os.listdir(path)
50     # 执行删除函数
51     r = input('请选择你要进行的任务：\n1、添加字符\n2、删除字符')
52     if r == '1':
53         add()
54     elif r == '2':
55         dele()
56     else:
57         print('输入错误！')