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

为了熟练使用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 os
import re

s = os.sep

def 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()