1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added 'linestart' array and 'unreadline()' method (makes parsing a lot easier).

Browse Source
This commit is contained in:
Greg Ward 1999-03-26 21:48:59 +00:00
parent 447b4a0652
commit 787451b65f
1 changed files with 30 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Lib/distutils/text_file.py
View File

@ -52,6 +52,11 @@ class TextFile:
self.filename = filename
self.file = file
self.current_line = 0 # assuming that file is at BOF!
# 'linestart' stores the file offset of the start of each logical
# line; it is used to back up the file pointer when the caller
# wants to "unread" a line
self.linestart = []
def open (self, filename):
@ -81,6 +86,11 @@ class TextFile:
buildup_line = ''
while 1:
# record current file position; this will be appended to
# the linestart array *unless* we're accumulating a
# continued logical line
current_pos = self.file.tell()
# read the line, optionally strip comments
line = self.file.readline()
if self.strip_comments and line:
@ -101,6 +111,11 @@ class TextFile:
self.current_line[1] = self.current_line[1] + 1
else:
self.current_line = [self.current_line, self.current_line+1]
# Forget current position: don't want to save it in the
# middle of a logical line
current_pos = None
# just an ordinary line, read it as usual
else:
if not line:
@ -111,7 +126,7 @@ class TextFile:
self.current_line = self.current_line[1] + 1
else:
self.current_line = self.current_line + 1
# strip whitespace however the client wants (leading and
# trailing, or one or the other, or neither)
@ -128,6 +143,12 @@ class TextFile:
if line == '' or line == '\n' and self.skip_blanks:
continue
# if we're still here and have kept the current position,
# then this physical line starts a logical line; record its
# starting offset
if current_pos is not None:
self.linestart.append (current_pos)
if self.join_lines:
if line[-1] == '\\':
buildup_line = line[:-1]
@ -147,6 +168,14 @@ class TextFile:
# end readline
def unreadline (self):
if not self.linestart:
raise IOError, "at beginning of file -- can't unreadline"
pos = self.linestart[-1]
del self.linestart[-1]
self.file.seek (pos)
def readlines (self):
lines = []
while 1: