python检索特定内容的文本文件实例_Python

python检索特定内容的文本文件实例

2021-03-01 00:11wedvefv Python

今天小编就为大家分享一篇python检索特定内容的文本文件实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

windows环境下python2.7

脚本指定一个参数作为要检索的字符串

例如： >find.py ./ hello

									# coding=utf-8

									import os

									import sys

									# 找到当前目录下的所有文本文件

									def findFile(path):

									 f = []

									 d = []

									 l = os.listdir(path)

									 for x in l:

									 if os.path.isfile(os.path.join(os.getcwd() + "\\", x)):

									  f.append(x)

									 else:

									  d.append(x)

									 return f, d # 返回文件和目录的列表

									# print x, "\n", y

									# 统计一个文本内字符串的个数

									def findstrCount(file, strToFind):

									 count = 0

									 thefile = open(file, 'rb')

									 while True:

									 buffer = thefile.read()

									 if not buffer:

									  break

									 count += buffer.count(strToFind)

									 thefile.close()

									 return count

									# 遍历文件列表中，包含特定字符串的文件

									def findstr(file, str):

									 # f = open(file, "r+")

									 # if f.read().find(str) != -1:

									 # s = os.getcwd() + "\\" + file

									 # else:

									 # s = "None"

									 # f.close()

									 i = 1

									 global s

									 for line in open(file):

									  # return is index of the str start position.

									 if line.find(str) != -1:

									  s = os.getcwd() + "\\" + file + "------>line:%d" % (i)

									  print s

									 i = i + 1

									 return s

									L = [] # 全局变量，存放找到的目标文件

									def find(p, str):

									 try:

									 f, d = findFile(p)

									 for x in f:

									  Ret = findstr(x, str)

									  if Ret:

									  L.append(Ret)

									 if d:

									  for x in d:

									  os.chdir(x)

									  find("./", str)

									  os.chdir('../')

									 except Exception, e:

									 print e

									 finally:

									 pass

									if __name__ == '__main__':

									 s = 0

									 find(sys.argv[1], sys.argv[2])