Python：删除用户home目录下的PNG文件的小程序。-CSDN博客

这是一款用于删除根目录下的PNG文件的小程序。由于工作中经常用shutter截图，把图片保存到根目录下，然后执行这个命令定时删除。

 1 #!/usr/bin/env python
 2 """
 3 version: 2.0
 4 function: to del png file under ~/
 5 update: using os.path.join() instead of '+'
 6 update: add output infos
 7 update: using os.getenv('HOME') instead of '/home/nathanh'
 8 update: import sys, using sys.exit(2) for error to remove a dir
 9 time: 2012-02-24
10 """
11 import os
12 import re
13 import sys
14 
15 def main():
16     """ to del png file under os.getenv('HOME')"""
17     home_dir=os.getenv('HOME')
18     x = 0
19     for file_name in os.listdir(home_dir):
20         if re.search('\.[pP][nN][gG]$', file_name):
21             file_path = os.path.join(home_dir, file_name)
22             try:
23                 os.remove(file_path)
24                 print 'del: ', file_path
25             except OSError, err:
26                 print 'del: ', err
27                 sys.exit(2)
28             x = x + 1
29 
30     if not x:
31         print "No PNG files under %s" % home_dir 
32 
33 if __name__ == '__main__':
34     main()