python学习笔记之os.walk()实现遍历文件搜索字符串

来源:互联网 发布:everyman睡眠 知乎 编辑:程序博客网 时间:2024/05/07 18:01

为了熟练使用python,打算有事没事用python写一些实用的小脚本
我刚刚进入计算机专业一年,一些专业用语不是很清楚,有说错的地方欢迎大家批评指正

—————————–分割线———————————————————————-

今天学习了os库中的os.walk(),并用它实现了一个小程序(功能如题)

函数原型:

 os.walk(top, topdown=True, onerror=None, followlinks=False)

参数解释:
top是你要遍历的目录
topdown是遍历顺序(True为从上往下)
onerror是一个函数,在遍历文件夹出现错误时调用(默认为None,即忽略错误,继续walk,当然这里可以写一个异常处理函数)
这里注意的是:onerror只是抛出在读取文件名时的异常,如果在打开文件(如用file的open方法打开.mp3 文件,还是会报错,所以还是用try比较好
followlinks确定文件链接方式(硬链接软连接啥的不是很明白。。反正现在用不到先不管)
返回值解释:
os.walk的返回值是一个三元组(dirpath,dirname,filename)
dirpath:根目录下的文件夹路径列表(这个列表包括整个目录树的每个文件所在文件夹的路径)
dirname:当前路径中的文件夹(不是文件)的名字列表
filename:当前路径中的文件(不是文件夹)的文件名列表

根据这个3元组用个嵌套的for循环就可以实现功能了
贴个代码:

import osimport res = os.sepdef walkFiles(root,string):    for dirpath,dirname,filename in os.walk(root):        for fn in filename:            try:                name = dirpath + s + fn                print "Searcing file '" + name  + "'..."                flag = 0                fp = open(name,'r')                count  = 0                for line in fp.readlines():                    count += 1                    if string in line:                        flag = 1                        print "Your string is in file '" + name + "' line " + str(count)                 if flag == 0 :                    print "Not Found"            except:                print "File '" + name + "'can't be read"def main():    print "Pls input the root you want to search:"    root = raw_input()    print "Pls input your string:"    string = raw_input()    walkFiles(root,string)    print "Done"if __name__ == "__main__":    main()
0 0
原创粉丝点击