截断文本文件不会更改文件

Question

当一个新手（像我一样）要求在 python 中读取/处理一个文本文件时，他经常得到如下答案：

with open("input.txt", 'r') as f:
    for line in f:
        #do your stuff

现在我想在特殊行之后截断正在读取的文件中的所有内容。修改上面的示例后，我使用：

with open("input.txt", 'r+') as file:
    for line in file:
        print line.rstrip("\n\r") #for debug
        if line.rstrip("\n\r")=="CC":
           print "truncating!"  #for debug
           file.truncate();
           break;

并期望它在看到第一个“CC”之后扔掉所有东西。在 input.txt 上运行此代码：

AA
CC
DD

在控制台上打印以下内容（如预期的那样）：

AA
CC
truncating!

但文件“input.txt”保持不变！？！？

怎么可能？我做错了什么？

编辑：操作后我希望文件包含：

AA
CC

Answer 1

看起来您正在成为 Python 内部使用的预读缓冲区的受害者。来自documentation for the file.next() method：

文件对象是它自己的迭代器，例如iter(f) 返回 f（除非 f 被关闭）。当文件用作迭代器时，通常在for 循环中（例如，for line in f: print line.strip()），next() 方法会被重复调用。此方法返回下一个输入行，或在文件打开以供读取时遇到 EOF 时引发StopIteration（当文件打开以供写入时，行为未定义）。为了使for 循环成为循环文件行的最有效方法（一种非常常见的操作），next() 方法使用隐藏的预读缓冲区。作为使用预读缓冲区的结果，将next() 与其他文件方法（如readline()）组合起来无法正常工作。但是，使用seek() 将文件重新定位到绝对位置会刷新预读缓冲区。

结果是文件的位置不在您截断时所期望的位置。解决此问题的一种方法是使用 readline 循环文件，而不是迭代器：

line = file.readline()
while line:
    ...
    line = file.readline()

Answer 2

除了 glibdud 的回答之外，truncate() 还需要删除内容的大小。您可以通过tell() 命令获取文件中的当前位置。正如他所提到的，通过使用 for 循环，next() 禁止了诸如 tell 之类的命令。但是在建议的 while 循环中，您可以在当前的 tell() 位置截断。所以完整的代码应该是这样的：

Python 3：

with open("test.txt", 'r+') as file:
line = file.readline()
while line:
    print(line.strip())
    if line.strip() == "CC":
        print("truncating")
        file.truncate(file.tell())
        break
    line = file.readline()