项目要求:编写一个程序, 在一个文件夹中, 找到所有带指定前缀的文件, 诸如 spam001.txt,spam002.txt 等,并定位缺失的编号(例如存在 spam001.txt 和 spam003.txt,但不存在spam002.txt)。让该程序对所有后面的文件改名,消除缺失的编号。
代码如下:
#!/usr/local/bin/python3.6
# -*- coding: utf-8 -*-
import os
spamList = []
fullList = []
for file in os.listdir('.'):
if file.startswith('spam') and file.endswith('txt'):
spamList.append(file)
#把列表从小到大排序,这个列表是不完整的,缺失了一些元素。
spamList.sort()
#找到列表最后一个元素对应的数值,以这个数值为长度,创建一个完整的列表。
finalNum = int(spamList[-1][5:-4])
for i in range(1, finalNum+1):
fileNum = str(i).zfill(3)
#zfill的作用是给1、2、3前面补上0。
fullList.append('spam' + fileNum + '.txt')
#寻找spamList不存在,而fullList中存在的元素。
diffList = list(set(fullList).difference(set(spamList)))
#spamList倒序,从后往前,改为diffList的名字。
for i in range((len(diffList) * -1), 0):
os.rename(spamList[i], diffList[i])
思路:
- 先根据已有的文件,做成列表。
- 再根据最大的一个文件名,做成完整列表。
- 对比这两个列表,寻找缺失的元素,并生成缺失文件列表。
- 根据缺失文件列表,把最原先的列表,从后往前,改为diffList的名字。
作为附加的挑战,编写另一个程序,在一些连续编号的文件中,空出一些编号, 以便加入新的文件。
import os
import random
import send2trash
spamList = []
for file in os.listdir('.'):
if file.startswith('spam') and file.endswith('txt'):
spamList.append(file)
for delFile in random.sample(spamList,3):
send2trash.send2trash(delFile)
这个就比较简单了。
- 先遍历所有形式为spamXXX.txt的文件。
- 用random.sample函数随机选择三个文件。
- 用send2trash删除这些文件。