python 之遍历文件夹下的所有文件并查找特定内容

在使用python 编写自动化脚本时需要对特定数据进行处理，那么如何在某文件夹下查找特定数据呢？
代码如下：

import os
path = os.getcwd()
files = os.listdir(path)
print files
for file in files:

    logfile = open(file,"r")
    lines = logfile.readlines()
    for line in lines:
        if line.find("111") != -1:
            print "111 exist in file:" + file
        elif line.find("222") != -1:
            print "222 exist in file:" + file

运行结果：

可以看到，test.py 脚本中的数据也被读取了，这对于数据分析来说显然是不合理的，那么就需要对需处理的文件进行判断，即仅检查txt文件：

import os
path = os.getcwd()
files = os.listdir(path)
print files
for file in files:
    if file.split(".")[1] == "txt":
        logfile = open(file,"r")
        lines = logfile.readlines()
        for line in lines:
            if line.find("111") != -1:
                print "111 exist in file:" + file
            elif line.find("222") != -1:
                print "222 exist in file:" + file