使用 Python 过滤 Linux 日志文件

Question

我想过滤一个日志文件以保持所有行匹配特定模式。我想用 Python 做到这一点。

这是我的第一次尝试：

#!/usr/bin/env python

from sys import argv 

script, filename = argv
with open(filename) as f:
    for line in f:
        try:
            e = line.index("some_term_I_want_to_match")
        except: 
            pass
        else:
            print(line)

我该如何改进：

• 将结果保存到名称相似的新文件中（即不同的扩展名）
• 使用正则表达式使其更加灵活/强大。

（我只是在学习 Python。这个问题既是关于学习 Python，也是关于完成这个特定的结果。）

好的，这就是我到目前为止的想法......但是你如何做相当于在下面的行中添加r

re.compile(r"\s*")

其中的字符串不是字符串文字，如下一行所示？

re.compile(a_string_variable)

除此之外，我认为这个更新版本可以完成这项工作：

#!/usr/bin/env python

from sys import argv 
import re
import os
import argparse #requires Python 2.7 or above

parser = argparse.ArgumentParser(description='filters a text file on the search phrase')
parser.add_argument('-s','--search', help='search phrase or keyword to match',required=True)
parser.add_argument('-f','--filename', help='input file name',required=True)
parser.add_argument('-v','--verbose', help='display output to the screen too', required=False, action="store_true")
args = parser.parse_args()

keyword = args.search
original_file = args.filename
verbose = args.verbose

base_file, ext = os.path.splitext(original_file)
new_file = base_file + ".filtered" + ext

regex_c = re.compile(keyword)

with open(original_file) as fi:
    with open(new_file, 'w') as fo:
        for line in fi:
            result = regex_c.search(line)
            if(result):
                fo.write(line)
                if(verbose):
                    print(line)

这可以轻松改进吗？

Answer 1

嗯，你知道，你自己已经回答了大部分问题:)

对于正则表达式匹配，请使用re module（文档中有很好的解释性示例）。

您已经使用open() 函数打开文件。对打开的文件使用相同的函数进行写入，只需提供相应的mode 参数（如果需要，“w”或“a”与“+”组合，请参阅 Python 交互式 shell 中的help(open)）。就是这样。