python - grep 和 Python

Question

我需要一种通过 Unix 命令行中的正则表达式使用 grep 搜索文件的方法。例如当我在命令行中输入:

python pythonfile.py 'RE' 'file-to-be-searched'

我需要在文件中搜索正则表达式 'RE' 并打印出匹配的行。

这是我的代码:

import re
import sys

search_term = sys.argv[1]
f = sys.argv[2]

for line in open(f, 'r'):
    if re.search(search_term, line):
        print line,
        if line == None:
            print 'no matches found'

但是当我输入一个不存在的单词时，no matches found 不会打印

Answer 1

自然的问题是为什么不直接使用 grep？!但假设你不能......

import re
import sys

file = open(sys.argv[2], "r")

for line in file:
     if re.search(sys.argv[1], line):
         print line,

注意事项:

• search 而不是 match 来查找字符串中的任意位置
• print 后的逗号 (,) 删除回车(行会有一个)
• argv包含python文件名，所以变量需要从1开始

这不处理多个参数(像 grep 那样)或扩展通配符(像 Unix shell 那样)。如果您想要此功能，您可以使用以下方法获得它:

import re
import sys
import glob

for arg in sys.argv[2:]:
    for file in glob.iglob(arg):
        for line in open(file, 'r'):
            if re.search(sys.argv[1], line):
                print line,