撤消文件 readline() 操作，使文件指针回到原始状态

Question

我正在使用 file.readline() 以只读模式浏览文本文件的 Python 文件指针，查找特殊行。一旦我找到该行，我想将文件指针传递给一个方法，该方法期望文件指针位于该读取行的开始位置（而不是紧随其后）。

我如何基本上撤消一个 file.readline() 操作文件指针？

Answer 1

您必须通过在读取行之前调用 file.tell() 来记住位置，然后调用 file.seek() 来倒回。比如：

fp = open('myfile')
last_pos = fp.tell()
line = fp.readline()
while line != '':
  if line == 'SPECIAL':
    fp.seek(last_pos)
    other_function(fp)
    break
  last_pos = fp.tell()
  line = fp.readline()

我不记得在 for line in file 循环中调用 file.seek() 是否安全，所以我通常只写出 > while 循环。可能有一种更Pythonic 的方法可以做到这一点。

Answer 2

在调用 readline 之前，您可以使用 thefile.tell() 记录该行的起点，如果需要，可以使用 thefile 返回到该点.seek。

>>> with open('bah.txt', 'w') as f:
...   f.writelines('Hello %s\n' % i for i in range(5))
... 
>>> with open('bah.txt') as f:
...   f.readline()
...   x = f.tell()
...   f.readline()
...   f.seek(x)
...   f.readline()
... 
'Hello 0\n'
'Hello 1\n'
'Hello 1\n'
>>> 

如您所见，seek/tell“对”是“撤消”的，可以说，由 readline 执行的文件指针移动。当然，这只能对实际的可查找（即磁盘）文件起作用，而不能（例如）对使用套接字的 makefile 方法构建的类文件对象等起作用。

Answer 3

如果您的方法只是想迭代文件，那么您可以使用 itertools.chain 来创建适当的迭代器：

import itertools

# do something to the marker line and everything after
def process(it):
    for line in it:
        print line,
        
with open(filename,'r') as f:
    for line in f:
        if 'marker' in line:
            it=itertools.chain((line,),f)
            process(it)
            break

Answer 4

fin = open('myfile')
for l in fin:
    if l == 'myspecialline':
        # Move the pointer back to the beginning of this line
        fin.seek(fin.tell() - len(l))
        break
# now fin points to the start of your special line

Answer 5

如果您不知道最后一行，因为您没有访问过它，您可以向后阅读，直到看到换行符：

with open(logfile, 'r') as f:
    # go to EOF
    f.seek(0, os.SEEK_END)
    nlines = f.tell()
    i=0
    while True:
        f.seek(nlines-i)
        char = f.read(1)
        if char=='\n':
            break
        i+=1