一个 python 小脚本 —— 判断文件是否被 slackware 包管理

grissiom

将文件在 PATH 下保存成 slackfiles.py 并 chmod +x 就好~ 其实这个 python 脚本功能很单一，关键是要和 shell 脚本联动~
这里给几个常用的用法：
# 在 "/var/tmp/slackfile/db.pkl" 建立数据库
tail -n+19 /var/log/packages/* | slackfiles.py --creat-db
# 在 ~/db 建立数据库
tail -n+19 /var/log/packages/* | slackfiles.py --creat-db ~/db
# 根据数据库 "/var/tmp/slackfile/db.pkl" 在 /usr/lib64 下面找没有被包管理的文件
cd /
find usr/lib64 -type f | slackfiles.py
# 根据数据库 ~/db 在 /usr/lib64 下面找没有被包管理的文件
cd /
find usr/lib64 -type f | slackfiles.py --use-db ~/db
# 以上方法都没有考虑链接

欢迎提意见，拍砖~;)

源码：

1. 
   
2. #!/usr/bin/env python
   
3. import sys, cPickle
   
4. import os
   
5. 
   
6. def_db = "/var/tmp/slackfile/db.pkl"
   
7. 
   
8. def print_help(outfile=sys.stdout):
   
9.         outfile.write(
   
10. '''slackfiles.py [--use-db <db-name> | [--creat-db | --add-db] [db-name]]
    
11. 
    
12. If no arguments, it eqauls to "slackfile.py --use-db /var/tmp/slackfile/db.pkl"
    
13. Default database name is "/var/tmp/slackfile/db.pkl"
    
14. ''')
    
15. 
    
16. def check_dir(name):
    
17.         if not os.path.isdir(name):
    
18.                 try:
    
19.                         os.makedirs(name)
    
20.                 except:
    
21.                         sys.stderr.write('Could not make dir: ' + name + '\n')
    
22. 
    
23. def load_set(name):
    
24.         try:
    
25.                 return cPickle.load(open(name, 'rb'))
    
26.         except IOError:
    
27.                 sys.stderr.write(name + 'does not exist or \
    
28.                                 you don\'t have a read permission.\n\
    
29.                                 Please creat the datebase first\n')
    
30.                 sys.exit(1)
    
31. 
    
32. 
    
33. def build_set(se, infile):
    
34.         for i in infile:
    
35.                 se.add(i)
    
36. 
    
37. def build_db(filename, instream=sys.stdin, add=False):
    
38.         check_dir(os.path.split(filename)[0])
    
39. 
    
40.         if add == False:
    
41.                 se = set()
    
42.         else:
    
43.                 se = load_set(filename)
    
44.         build_set(se, instream)
    
45. 
    
46.         try:
    
47.                 dbfile = open(filename, 'wb')
    
48.         except IOError:
    
49.                 sys.stderr.write('Could not write to ' + filename + '\n')
    
50.                 sys.exit(1)
    
51.         cPickle.dump(se, dbfile, -1)
    
52.         dbfile.close()
    
53. 
    
54. def search_set(se, infile):
    
55.         for i in infile:
    
56.                 if i not in se:
    
57.                         print i[:-1] # strip one trailing \n manually.
    
58. 
    
59. def search_db(filename, instream=sys.stdin):
    
60.         se = load_set(filename)
    
61.         search_set(se, instream)
    
62. 
    
63. def main(args):
    
64.         i = 1    # skip the first arg(program name)
    
65.         while i < len(args):
    
66.                 if args[i] == '--use-db':
    
67.                         if i + 1 > len(args):
    
68.                                 print_help()
    
69.                                 sys.stderr.write('\nPlease specify the database name.\n')
    
70.                                 sys.exit(2)
    
71.                         search_db(args[i + 1])
    
72.                         sys.exit(0)
    
73.                 elif args[i] == '--creat-db':
    
74.                         if i + 1 < len(args):
    
75.                                 name = args[i + 1]
    
76.                         else:
    
77.                                 name = def_db
    
78.                         build_db(name)
    
79.                         sys.exit(0)
    
80.                 elif args[i] == '--add-db':
    
81.                         if i + 1 < len(args):
    
82.                                 name = args[i + 1]
    
83.                         else:
    
84.                                 name = def_db
    
85.                         build_db(name, add=True)
    
86.                         sys.exit(0)
    
87.                 i += 1
    
88.         search_db(def_db)
    
89. 
    
90. if __name__ == '__main__':
    
91.         main(sys.argv)
    

复制代码

不过需要注意的是根据创建数据库的方法不同，有的已经被包管理的文件可能会被误认为是没有被包管理，也可能被其他文件用到。删除文件时需慎重！！