查找相关脚本

zhaohang3031

在指定路径中查找所有文本文件里的关键内容：

#!/usr/bin/env python
import os, sys
listonly = False
skipexts = ['.gif', '.exe', '.pyc', '.o', '.a','.dll','.lib','.pdb','.mdb']
def visitfile(fname, searchKey):
    global fcount, vcount
    try:
        if not listonly:
            if os.path.splitext(fname)[1] in skipexts:
                pass
            elif open(fname).read().find(searchKey) != -1:
                print'%s has %s' % (fname, searchKey)
                fcount += 1
    except: pass
    vcount += 1
def visitor(args, directoryName,filesInDirectory):
    for fname in filesInDirectory:
        fpath = os.path.join(directoryName, fname)
        if not os.path.isdir(fpath):
            visitfile(fpath,args)
def searcher(startdir, searchkey):
    global fcount, vcount
    fcount = vcount = 0
    os.path.walk(startdir, visitor, searchkey)
if __name__ == '__main__':
    root=raw_input("type root directory:")
    key=raw_input("type key:")
    searcher(root,key)
    print 'Found in %d files, visited %d' % (fcount, vcount)

在指定路径中查找固定扩展名的文件：

#!/usr/bin/env python
import os
def anyTrue(predicate, sequence):
    return True in map(predicate, sequence)
def filterFiles(folder, exts):
    for fileName in os.listdir(folder):
        if os.path.isdir(folder + '/' + fileName):
            filterFiles(folder + '/' + fileName, exts)
        elif anyTrue(fileName.endswith, exts):
            print fileName
exts = ['.rmvb', '.avi', '.pmp','.tar.gz']
filterFiles('/root', exts)



本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u3/93470/showart_1871923.html