本人编写的第一个脚本，用来删除所有旧版本文件和空文件夹，恳请高手给出意见。

ssfjhh

为什么CU发代码的这个<>会吃代码？

比如正则表达式我明明写的是<'\.\d+$'>

发出来以后却变成了<'\.\d+>，<$'>被吃掉了，

导致贴出的代码无法正常运行。

我经常要用proe工作，这个软件在保存的时候每次都会生产一个带有新版本号的文件，形如
abc.asm.13', 'abc.asm.14', 'abc.asm.2', 'abc.asm.3', 'abc.asm.4', 'abc.asm.5', 'abc.asm.6', 'abc.log.8', 'abc.log.9', 'abc.prt.1', 'abc.prt.10', 'abc.prt.11', 'abc.prt.12', 'abc.prt.13', 'abc.prt.14', 'abc.prt.2', 'abc.prt.3', 'abc.prt.4', 'abc.prt.5', 'abc.prt.6', 'abc.prt.7', 'abc.prt.8', 'abc.prt.9',

但是这目录下还会有其他文件夹和一些不带版本号的文件，如 'abc0', 'abc1', 'abc1.doc', 'abc1.dwg', 'abc1.dxf',

我写的这个脚本用来删除脚本所在目录以及子目录下所有旧版本的文件以及空目录，并将最新版的的文件的版本号改为1，不带版本号的文件不做处理
但是总感觉在判断旧版本的地方写得很不好，又想不出更好的办法，恳请高手给出意见，算法上的和代码编写习惯上的，只要你能给出的意见都可以因为是第一次写代码，所以脚本中尽可能多的用上了自己所掌握的技能。

代码如下：

1. def name(filename):
   
2.     """输入一个文件名，如果带版本号就去掉版本号，如果不带版本号，原样返回"""
   
3.     import re
   
4.     return re.sub('\.\d+,'', filename)
   
5. 
   
6. 
   
7. def version(filename):
   
8.     """返回给出带版本号的文件名的版本号"""
   
9.     import re
   
10.     return re.sub('^.+\..+\.(?=\d+$)', '', filename)
    
11. 
    
12. 
    
13. def chverto1(currentdir, filename):
    
14.     """将输入的版本号不为1的文件名的版本号改为1"""
    
15.     import os
    
16.     if version(filename) == '1':
    
17.         print("文件\"{}\"的最新版本为1，不需要处理".format(os.path.join(currentdir,filename)))
    
18.     else:
    
19.         oldpathname = os.path.join(currentdir, filename)
    
20.         newpathname = os.path.join(currentdir,  name(filename) + '.1')
    
21.         print("文件\"{}\"的版本号被改为1".format(oldpathname))
    
22.         if os.path.exists(newpathname):
    
23.             print('正在删除旧版文件' + "\"" +newpathname +"\"")
    
24.             os.remove(newpathname)
    
25.         os.rename(oldpathname, newpathname)
    
26. 
    
27. 
    
28. def isversionfile(filename):
    
29.     """判断给出的文件名是否带版本号"""
    
30.     import re
    
31.     if re.search('\.\d+,filename) != None:
    
32.         return True
    
33.     else:
    
34.         return False
    
35. 
    
36. 
    
37. def sortfile(currentdir):
    
38.     """分类给出的目录，返回一个二维列表，第一个是子目录列表，第二个是带版本号文件名列表"""
    
39.     import os
    
40.     dirlist = []
    
41.     versionfilelist = []
    
42.     allfile = [dirlist, versionfilelist]
    
43.     unsortedfilelist = os.listdir(currentdir)
    
44.     print('开始分类文件夹\"' + currentdir + "\"")
    
45.     for i in unsortedfilelist:
    
46.         if os.path.isdir(os.path.join(currentdir,i)):
    
47.             dirlist.append(i)
    
48.         elif isversionfile(i):
    
49.             versionfilelist.append(i)
    
50.         else:
    
51.             print("文件\"{}\"不是带版号的文件，不需要处理".format(os.path.join(currentdir, i)))
    
52.     else:
    
53.         return allfile
    
54. 
    
55. 
    
56. def delfiles(currentdir, versionfilelist):
    
57.     """输入一个目录和一个带版本号的文件名列表，删除所有旧版本，并将最新版的版本号改为1"""
    
58.     import os
    
59.     filecutverset = {name(i) for i in versionfilelist}
    
60.     for i in filecutverset:
    
61.         allversionofafile = []
    
62.         for j in versionfilelist:
    
63.             if i == name(j):
    
64.                 allversionofafile.append(j)
    
65.         #判断不带版本号部分的文件名部分是否与已知的去掉版本部分是否一直得到一个文件的所有版本
    
66.         else:
    
67.             allversionofafile.sort()
    
68.             latestversionname = allversionofafile.pop()
    
69.         #排序同文件的不同版本，最新的版本版本号最大,处于序列的最末位置，
    
70.         #使用pop()取出最新版本文件名备用，列表中剩下的全为旧版本，应该删除
    
71.         #使用else语句，可以保证循环部分没有错误才会执行排序
    
72.         for k in allversionofafile:
    
73.             print('正在删除旧版文件' + "\"" +os.path.join(currentdir,k) +"\"")
    
74.             os.remove(os.path.join(currentdir,k))
    
75.         else:
    
76.             chverto1(currentdir, latestversionname)
    
77.         #删除列表中的所有旧文件，使用else语句，保证删除过程没有错误才会执行重命操作
    
78. 
    
79. 
    
80. 
    
81. def rmof(currentdir):
    
82.     """看了这么多函数，都是为这个函数服务滴"""
    
83.     import os
    
84.     print("准备清理文件夹\"" + currentdir + "\"下的旧版本文件")
    
85. 
    
86.     if os.listdir(currentdir) == []:
    
87.         os.rmdir(currentdir)
    
88.         print("文件夹\"{}\"是空文件夹，删除".format(currentdir))
    
89.         return None
    
90.     else:
    
91.         dirlist, versionfilelist = sortfile(currentdir)
    
92.     #首先判断目录是否为空，避免进行不必要的操作，空目录留着何用呢，亮不犹豫的，删掉
    
93.    
    
94.     if versionfilelist == []:
    
95.         pass
    
96.     else:
    
97.         delfiles(currentdir, versionfilelist)
    
98.     #判断文件夹下是否有带版本号的文件名，有才需要进一步操作撒，没有直接pass
    
99.    
    
100.     for i in dirlist:
     
101.         print("进入文件夹\"{}\"下的子文件夹\"{}\"".format(currentdir,i))
     
102.         rmof(os.path.join(currentdir,i))
     
103.     #for循环本身就有判断作用，不需要现多作用if做判断了
     
104.     #如果存在子目录，递规调用自身，删除目录下的所有文件不费力，递规真好用呀
     
105.         
     
106.     if os.listdir(currentdir) == []:
     
107.         os.rmdir(currentdir)
     
108.         print("文件夹\"{}\"是空文件夹，删除".format(currentdir))
     
109.     #再做一次判断，避免当前目录下只有子目录，没有子文件，
     
110.     #而子目录又都是空目录，递规删除子目录后，父目录也变为空的极少情况发生，谁让咱有洁癖呢
     
111.         
     
112. 
     
113. def countdown(t):
     
114.     import time
     
115.     while t:
     
116.             print(str(t) + ' ', end = '')
     
117.             t= t-1
     
118.             time.sleep(1)
     
119.     #计个数，谁能告诉我怎么实现数字在一个地方跳动的效果，貌似这个\b不行呀
     
120. 
     
121. 
     
122. def comfirm():
     
123.     """一次要删这么多文件，本着负任的态度，必须得问问清楚用户是否真的要这么干
     
124. 
     
125.     好吧，我承认，我是怕我自己误操作才写这么一段函数的"""
     
126.     import os,sys,time
     
127.     currentdir = sys.path[0]
     
128.     #脚本有可能被转成exe，如果被转成exe，上边这句该怎么写
     
129.     #我保证，你告诉我之后我就能写出脚本和exe都能正确执行的语句
     
130.     """\
     
131. 确定要清理文件夹\"{}\"吗？
     
132. 本脚本将会删除所有该文件夹下的旧版本文件以及空文件夹，删除操作不可恢复!!!
     
133. """
     
134.     print("确定要清理文件夹\"{}\"吗？\n本脚本将会删除所有该文件夹下的旧版本文件以及空文件夹，删除操作不可恢复!!!".format(currentdir))
     
135.     trytimes = 3
     
136.     while trytimes:
     
137.         trytimes = trytimes -1
     
138.         getinput = input("确认操作请输入‘yes', 取消请输入'no':")
     
139.         if getinput == 'yes':
     
140.             rmof(currentdir)
     
141.             print("所有旧友版文件都已被删除，请检查！")
     
142.             input('按回车键退出')
     
143.             break
     
144.         elif getinput == 'no':
     
145.             print("操作已取消,5秒种后退出", end = '')
     
146.             countdown(5)
     
147.             break
     
148.         elif trytimes >= 1:
     
149.             print('yes or no, please!')
     
150.             print("还有{}次机会".format(trytimes,))
     
151.     else:
     
152.         print("错误次数太多，操作取消，10秒种后退出",end = '')
     
153.         countdown(10)
     
154.             
     
155. 
     
156. comfirm()
     

复制代码

ssfjhh

怎样才能插入代码，我看到别人都能插入比较漂亮的代码，如下所示

cdtits

回复 2# ssfjhh


    选中所有代码，点击回复框上面的”<>“按钮即可

ssfjhh

回复 3# cdtits

3q，已经改好了。

   

cdtits

1. 
   
2. def CountDown(t):
   
3.     import time
   
4.     mlen = len(str(t)) + 1
   
5.     blank = '\b \b' * mlen + '%d'
   
6.     print t,
   
7.     while t:
   
8.         time.sleep(1)
   
9.         t -= 1
   
10.         print blank % t,
    
11. 
    
12. if __name__ == '__main__':
    
13. 
    
14.     CountDown(100)

复制代码

xtdumpling

python 2.6的版本，3.0以上需要把raw_input 改为input

1. #-*- coding:utf-8 -*-
   
2. 
   
3. import os
   
4. import sys
   
5. import re
   
6. 
   
7. class delmass(object):
   
8.     def __init__(self, workdir = os.curdir):
   
9.         self.workdir = os.path.abspath(workdir)
   
10.         self.flist = [f for f in os.listdir(self.workdir) if os.path.isfile(f)]
    
11.         self.dlist = [d for d in os.listdir(self.workdir) if os.path.isdir(d)]
    
12. 
    
13.     def delfiles(self):
    
14.         # filename end with dot+digital, such as abcd.efgx.1234
    
15.         pattern = '\w+\.\w+\.\d+'
    
16.         for eachfile in self.flist:
    
17.             if re.match(pattern, eachfile):
    
18.                 print 'Remove file:', eachfile
    
19.                 os.unlink(eachfile)
    
20. 
    
21.     def deldirs(self):
    
22.         for eachdir in self.dlist:
    
23.             if os.listdir(eachdir) == []:
    
24.                 print 'Remove dir:', eachdir
    
25.                 os.rmdir(eachdir)
    
26. 
    
27. def main():
    
28.     '''删除目标文件夹里面带版本号的文件和空文件夹，目标文件夹可以作为参数或者默认是当前文件夹'''
    
29.     if len(sys.argv) > 1:
    
30.         dm = delmass(sys.argv[1])
    
31.     else:
    
32.         dm = delmass()
    
33. 
    
34.     print u"确认操作请输入‘yes', 取消请输入'no':",
    
35.     if raw_input().strip() == 'yes':
    
36.         dm.delfiles()
    
37.         dm.deldirs()
    
38.     else:
    
39.         print u'操作已取消'
    
40. 
    
41. if __name__ == '__main__':
    
42.     main()

复制代码

ssfjhh

回复 6# xtdumpling

1. def delfiles(self):
   
2.         # filename end with dot+digital, such as abcd.efgx.1234
   
3.         pattern = '\w+\.\w+\.\d+'
   
4.         for eachfile in self.flist:
   
5.             if re.match(pattern, eachfile):
   
6.                 print 'Remove file:', eachfile
   
7.                 os.unlink(eachfile)

复制代码

你这段代码的意思应该是删掉所有带版本号的文件，这个是不符合我的要求的。

我需要的功能是，假如文件夹下有abcd.efgx.1234，abcd.efgx.1235，abcd.efgx.1246三个文件，删掉abcd.efgx.1234，abcd.efgx.1235，并将abcd.efgx.1246改为abcd.efgx.1
而且要实现递规操作，删除，所有子目录中的旧版文件。

不过，你的代码比我的简洁太多了，用到了类，我还不会用。

非常感谢。

xtdumpling

我也是刚刚开始学习，把书上的知识尝试一下。
改进一下：

1. #!/usr/bin/env python
   
2. #-*- coding:utf-8 -*-
   
3. 
   
4. import os
   
5. import sys
   
6. import re
   
7. 
   
8. class delmass(object):
   
9.     def __init__(self, workdir = os.getcwd()):
   
10.         self.orgdir = os.getcwd()
    
11.         self.workdir = os.path.abspath(workdir)
    
12.         os.chdir(self.workdir)
    
13.         self.lastfiles = {}
    
14. 
    
15.     def _getlatestfiles(self, fname, fversion):
    
16.         #get all last file name
    
17.         if fname in self.lastfiles:
    
18.             if int(fversion) > int(self.lastfiles[fname]):
    
19.                 print 'Remove file: %s%s' %(fname, self.lastfiles[fname])
    
20.                 os.unlink('%s%s' %(fname,self.lastfiles[fname]))
    
21.                 self.lastfiles[fname] = fversion
    
22.             else:
    
23.                 print 'Remove file: %s%s' %(fname, fversion)
    
24.                 os.unlink('%s%s' %(fname,fversion))
    
25.         else:
    
26.             self.lastfiles[fname] = fversion
    
27. 
    
28.     def _delfiles(self):
    
29.         # handle filename end with dot+digital, such as abcd.efgx.1234
    
30.         self.flist = [f for f in os.listdir(self.workdir) if os.path.isfile(f)]
    
31.         pattern = '(\w+\.\w+\.)(\d+)'
    
32.         for eachfile in self.flist:
    
33.             m = re.match(pattern, eachfile)
    
34.             if m:
    
35.                 print 'found file: %s%s' %(m.group(1), m.group(2))
    
36.                 self._getlatestfiles(m.group(1), m.group(2))
    
37.         else:
    
38.             # change all last files' version to 1
    
39.             for fname in self.lastfiles:
    
40.                 os.rename('%s%s' %(fname,self.lastfiles[fname]), \
    
41.                           '%s%s' %(fname,'1'))
    
42. 
    
43.     def _deldirs(self):
    
44.         # remove all empty dirs
    
45.         self.dlist = [d for d in os.listdir(self.workdir) if os.path.isdir(d)]
    
46.         for eachdir in self.dlist:
    
47.             if os.listdir(os.path.abspath(eachdir)) == []:
    
48.                 print 'Remove dir:', os.path.abspath(eachdir)
    
49.                 os.rmdir(eachdir)
    
50.             else:
    
51.                 # self class recursion for sub dir
    
52.                 print 'handle dir:', os.path.abspath(eachdir)
    
53.                 delmass(eachdir).delall()
    
54. 
    
55.     def delall(self):
    
56.         self._delfiles()
    
57.         self._deldirs()
    
58. 
    
59.     def __del__(self):
    
60.         # restore to original dir
    
61.         os.chdir(os.path.abspath(self.orgdir))
    
62. 
    
63. def main():
    
64.     '''删除目标文件夹里面带版本号的文件和空文件夹，目标文件夹可以作为参数或者默认是当前文件夹'''
    
65.     if len(sys.argv) > 1:
    
66.         dm = delmass(sys.argv[1])
    
67.     else:
    
68.         dm = delmass(os.curdir);
    
69. 
    
70.     print u"确认操作请输入‘yes', 取消请输入'no':",
    
71.     if raw_input().strip() == 'yes':
    
72.         dm.delall()
    
73.     else:
    
74.         print u'操作已取消'
    
75. 
    
76. if __name__ == '__main__':
    
77.     main()

复制代码

ssfjhh

感谢，比我的代码精简多了，

def _getlatestfiles(self, fname, fversion):这部分里边出现了
os.unlink('%s%s' %(fname,self.lastfiles[fname]))
os.unlink('%s%s' %(fname,fversion))
都是不带路径的文件名，能正确运行吗？

xtdumpling

删除的文件都是在当前目录，我测试过可以的。