如何使用python删除测试文件中的后半部分内容？

Question

Just as the picture below, there is a line ENDMDL . How can I use the readline() function and some basic loop to delete all the content after this line?

Answer 1

Here you go:

file_path = 'your/file/path'

with open(file_path) as inf, open('outfile', 'w') as outf:
    for i in inf:
        if i.strip() == 'ENDMDL':
            break
        else:
            outf.write(i)

Answer 2

Use a tempfile and use shutil.move to replace the original:

from tempfile import NamedTemporaryFile

from shutil import move

with open("your_file") as f, NamedTemporaryFile("w", dir=".",delete=False) as tmp:
    for line in f:
        tmp.write(line)
        if line.rstrip() == "ENDMDL":
            break
move(tmp.name, "your_file")