9.8.3 消除缺失的编号

编写一个程序,在一个文件夹中,找到所有带指定前缀的文件,诸如 spam001.txt, spam002.txt 等,
并定位缺失的编号(例如存在 spam001.txt 和 spam003.txt,但不存 在 spam002.txt)。
让该程序对所有后面的文件改名,消除缺失的编号。

import os,re,shutil#导入模块
path=input('请输入查找路径:')
prefixname=input('请输入前缀:')#如spam,不包括编号
rg=re.compile(r'%s\d\d\d.*' %prefixname)#创建匹配特定前缀的正则表达式
rg2=re.compile(r'\d\d\d')#创建匹配编号的正则表达式
L=[]#储存文件名
for filename in os.listdir(path):#找出所有特定前缀的文件,并存放在L里
    if os.path.isfile(os.path.join(path,filename)) and rg.search(filename):
        L.append(filename)
print(L)
for i in range(len(L)):#迭代L
    if i==0:
        continue
    if (int(rg2.search(L[i]).group())-int(rg2.search(L[i-1]).group()))>1:#判断是否有编号缺失
        num=int(rg2.search(L[i-1]).group())+1#修改后的文件编号
        exname=os.path.splitext(os.path.join(path,L[i]))[1]#文件后缀名
        filename=r'%s%03d%s' %(prefixname,num,exname)#新文件名
        shutil.move(os.path.join(path,L[i]),os.path.join(path,filename))#则重命名文件
        L[i] = filename#替换L中原有文件名,以便下一次比较

print(L)

猜你喜欢

转载自blog.csdn.net/qq_40012965/article/details/81283597