Python 遍历文件，搜索特定字符串，如果找到，则复制这些行的 rest 并合并到一个组合文件中

Question

I have a folder contains 500 text files. Python Iterate through files Search for certain string if found copy and consolidate to a combined file as "Output.txt". The string we are looking for in each of the file in the directory

import os

searchquery = 'No' #string we are looking for in each of the file in the directory
def my_function(fname):
    Output=[]
    with fname as f1:
      with Output as f2:

        Lines = f1.readlines()

        try:
          i = Lines.index(searchquery)
          for iline in range(i+1, i+18): # we need to copy rest of the 18 or less line after 'No' is found
            f2.write(Lines[iline])
        except:
          print(" ")
    return Output

for filename in os.listdir('C:\\Users\\XXX\\Desktop\\Tox\\tcm2'):
    M1=open(filename)
    M2=my_function(M1)
    opened_file = open(Output.txt, 'a')
    opened_file.write("%r\n" % M1)
    opened_file.close()

I am seeing the following error

    with Output as f2:
AttributeError: __enter__

Answer 1

you cannot do with Output as f2 , because Output is a list and it doesn't support that, and gives you AttributeError: __enter__ , another problem is the line where you did f2.write() again you cannot write to a list, use append() instead.

Here is the full working code, i tested it:

import os
searchquery = 'No'
path = 'C:\\Users\\XXX\\Desktop\\Tox\\tcm2\\'

def my_function(fname):
    Output=[]           
    Lines = fname.readlines()   
    found = False 
    for line in Lines :
        if (found == True):
            Output.append(line)
        if line.startswith(searchquery):
            found = True
    return Output

opened_file = open('Output.txt', 'a')
for filename in os.listdir(path):
    M1=open(path+filename)
    result=my_function(M1)        
    for e in result:
        opened_file.write(e)        
    M1.close()
opened_file.close()

Answer 2

Why not simply using a cmd line, go to the directory and run:

grep no -A18 * | egrep -v "no|--" > output.txt

In case you do not have egrep:

grep no -A18 * | grep -v no | grep -v "--" > output.txt