Python:撤消文件readline()操作,使文件指针恢复原始状态

Question

我正在使用file.readline()查找特殊行,以只读模式浏览文本文件的Python文件指针.一旦找到该行,我想将文件指针传递给一个方法,该方法期望文件指针位于该readline的START位置(不在它之后).

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

Answer 1

您必须通过file.tell()在readline之前调用然后调用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()

如果是安全调用我不记得file.seek()一个内for line in file循环,所以我通常只是写出来的while循环.这样做可能有更多的pythonic方式.

Answer 2

thefile.tell()在调用之前记录行的readline起始点,如果需要,则返回到该点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'
>>> 

如你所见,搜索/告诉"对"是"撤消",可以说,文件指针移动由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