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('输入错误!')