如何使用Python中的函数搜索文件中的特定行并将它们写入另一个文件

Question

我的目标是构建一个日志解析器，它将复制我想要的关键字之间的选定行并将其写入文件。由于我必须在单个文件中的多个关键字之间进行搜索，因此我想编写一个函数并在我的脚本中多次使用它。

但是我无法通过以下脚本实现这一点并收到错误：

import re

def myfunc (infile ,outfile, search1 , search2):

    fi =  infile.readlines()
    fo =  open(outfile, 'w')

    write1 = False
    for line in fi:
     if re.findall('search1' , str(line)):
        write1 = True
     elif re.findall('search2', str(line)):
        write1 = False
     elif write1:
        fo.write(line)

    fo.close()
    fi.close()

    return;

text_file = open(input("name of inputfile : "))
resultfile =  input("name of outputfile : ")

search1 = "teen"
search2 = "eight"
myfunc (text_file , resultfile , search1 , search2)

我收到以下错误：

Traceback (most recent call last):
  File "C:/Users/zoro/PycharmProjects/text-parsing/write selected test 2 sets.py", line 38, in <module>
    myfunc (text_file , resultfile , search1 , search2)
  File "C:/Users/zoro/PycharmProjects/text-parsing/write selected test 2 sets.py", line 28, in myfunc
    fi.close()
AttributeError: 'list' object has no attribute 'close'

Answer 1

fi = infile.readlines()

这使得fi 成为文件infile 中的行列表。因此，当您稍后调用fi.close() 时，您正在尝试关闭一个列表，这当然不起作用。

相反，您需要关闭文件，即infile:

infile.close()

一般来说，以某种方式更改变量名称是个好主意，这样它们所包含的内容就很明显了。 infile 是您从中读取的文件对象，所以没关系。 outfile 是您要写入的文件的文件名，因此您应该将其命名为 outFileName 或其他名称。 fi 是infile 中的行列表，所以你应该称它为inFileLines。

您还应该避免手动关闭文件对象；相反，使用with 语句确保它们自动关闭：

with open(outfile, 'w') as fo:
    fo.write('stuff')
    # no need to manually close it

最后，您的代码还有另一个问题：re.findall('search1' , str(line)) 这将在行中搜索字符串'search1'；它不会尊重传递给函数并存储在search1（和search2）变量中的值。所以你需要删除那里的引号：re.findall(search1, line)（你也不需要将行转换为字符串）。

另外，如果你只评估它的真值，使用re.findall() 并不是最好的方法。相反，使用re.search，它只返回第一个结果（所以对于非常长的行，如果你已经找到了结果，你就不会继续搜索）。如果search1 和search2 不包含实际的正则表达式，而只是您想在该行中找到的字符串，那么您也应该只使用in 运算符：

if search1 in line:
    write1 = True

---

最后一点：文件句柄应始终从打开它们的同一级别关闭。因此，如果您在函数中打开文件句柄，那么 that 函数也应该关闭它。如果您在函数外部打开文件，则函数不应关闭它。关闭文件是打开者的责任，而其他情况下关闭文件可能会导致错误的行为，所以你不应该这样做（除非它被明确记录，例如函数doSomethingAndClose可能会关闭文件）。

使用with 语句通常可以避免这种情况，因为您从不手动调用file.close()，而with 语句已经确保文件已正确关闭。

如果你想多次使用一个文件，那么你必须seek to the beginning 才能再次读取它。在您的情况下，由于您使用infile.readlines() 将整个文件读入内存，因此最好从文件中读取行一次，然后将其重用于多个函数调用：

text_file = input("name of inputfile : ")
with open(text_file) as infile:
    fi = infile.readlines() # read the lines *once*

    myfunc(fi, …)
    myfunc(fi, …)
    myfunc(fi, …)