Python模拟Linux/Mac OS工具grep和Windows工具findstr

在Linux和Mac OS系统中有个工具grep可以用来查找哪些文件中包含特定的字符串，Windows系统中也有类似的工具findstr，本文代码模拟了这两个工具的工作原理。

from os import listdir

from sys import argv

def prepare(fileNames):

    for item in fileNames:

        # 把类似于*.txt之类的文件名替换成真实文件名

        # 不考虑类似于abc*.py的文件名

        if '*' in item:

            index = item.index('.')

            fileNames.remove(item)

            temp = [fn for fn in listdir()\

                    if fn.endswith(item[index:])]

            fileNames.extend(temp)

def main(desStr, fileNames):

    result = []

    for filename in fileNames:

        try:

            # 使用UTF8编码打开文本文件

            # 可以根据需要进行改写或扩充

            with open(filename, encoding='utf8') as fp:

                for line in fp:

                    if desStr in line:

                        result.append(filename)

                        break

        except:

            pass

    return result

# 要查找的字符串

desStr = argv[1]

# 要查找的全部文件

fileNames = argv[2:]

# 预处理

prepare(fileNames)

# 进行查找并输出结果

result = main(desStr, fileNames)

for item in result:

    print(item)

本程序需要在命令提示符环境中运行，某次运行截图如下：