题目:
编写一个程序,在当前文件夹中,找到所有带指定前缀的文件,诸如 spam001.txt, spam002.txt 等,并定位缺失的编号(例如存在 spam001.txt 和 spam003.txt,但不存 在 spam002.txt)。让该程序对所有后面的文件改名,消除缺失的编号。
代码:
#! python3
# SortFileName.py - 找到所有带指定前缀的文件,诸如 spam001.txt, spam002.txt 等,并定位缺失的编号(例如存在 spam001.txt 和 spam003.txt,但不存在 spam002.txt)。让该程序对所有后面的文件改名,消除缺失的编号。
import re,os,shutil
#构建一个文件名的正则表达式
FilenameRegex=re.compile(r'^spam\d{3}.*(\.\w*)$')
#构建两个列表,放原来的文件名、放新的文件名
RawNameList=[]
NewNameList=[]
#遍历当前文件夹
fileList=os.listdir('.')
num=1
#获取旧文件名,生成新文件名
for filename in fileList:
nameMatch=FilenameRegex.search(filename)
if nameMatch != None:
RawNameList.append(filename)
NewName='spam'+'%03d'%num + nameMatch.group(1)
NewNameList.append(NewName)
num += 1
#重命名文件
for raw, new in zip(RawNameList, NewNameList):
print('Rename %s to %s'%(raw, new))
shutil.move(raw, new)