消除缺失的编号

消除缺失的编号

编写一个程序，在一个文件夹中，找到所有带指定前缀的文件，诸如 spam001.txt,spam002.txt 等，并定位缺失的编号（例如存在 spam001.txt 和 spam003.txt，但不存在 spam002.txt）。让该程序对所有后面的文件改名，消除缺失的编号。作为附加的挑战，编写另一个程序，在一些连续编号的文件中，空出一些编号，以便加入新的文件。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2021-02-20 11:41:35
# @Author  : Yang chunyu (ycyyangchunyu@qq.com)
# @Link    : https://blog.csdn.net/weixin_49374896
# @Version : $Id$

import os,shutil

folder='D:\\python_test\\code'
#首先编写一个程序，生成在一些连续编号的文件中，空出一些编号。
for i in range(10):
    #编号ch0 ch4 ch8空缺
    if i % 4!=0:
        file_name='ch'+str(i)+'.txt'
        name=os.path.join(folder,file_name)
        afile=open(name,'w')
        afile.close()

#消除缺失的编号

#number是完整的递增的编号
number=1
for folder_name,sub_floders,file_names in os.walk(folder):
    for file_name in file_names:
        #文件绝对路径
        name=os.path.join(folder_name,file_name)
        #file_name[-4:]是'.txt'后缀名
        #print(file_name[-5:-4])
        #n是编号
        n=int(file_name[-5:-4])
        #print(n)
        #如果有缺失的编号
        if n!=number:
            #通过shutil.move重命名文件
            new_file_name='ch'+str(number)+'.txt'
            new_name=os.path.join(folder_name,new_file_name)
            shutil.move(name,new_name)
        number=number+1