欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2024-01-23 16:18:10
windows环境下python2.7 脚本指定一个参数作为要检索的字符串 例如: >find.py ./ hello # coding=utf-8...

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])

以上这篇python检索特定内容的文本文件实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。