Varia's website
https://varia.zone
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
288 lines
12 KiB
288 lines
12 KiB
4 weeks ago
|
"""text_file
|
||
|
|
|
||
|
|
provides the TextFile class, which gives an interface to text files
|
||
|
|
that (optionally) takes care of stripping comments, ignoring blank
|
||
|
|
lines, and joining lines with backslashes."""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
class TextFile:
|
||
|
|
"""Provides a file-like object that takes care of all the things you
|
||
|
|
commonly want to do when processing a text file that has some
|
||
|
|
line-by-line syntax: strip comments (as long as "#" is your
|
||
|
|
comment character), skip blank lines, join adjacent lines by
|
||
|
|
escaping the newline (ie. backslash at end of line), strip
|
||
|
|
leading and/or trailing whitespace. All of these are optional
|
||
|
|
and independently controllable.
|
||
|
|
|
||
|
|
Provides a 'warn()' method so you can generate warning messages that
|
||
|
|
report physical line number, even if the logical line in question
|
||
|
|
spans multiple physical lines. Also provides 'unreadline()' for
|
||
|
|
implementing line-at-a-time lookahead.
|
||
|
|
|
||
|
|
Constructor is called as:
|
||
|
|
|
||
|
|
TextFile (filename=None, file=None, **options)
|
||
|
|
|
||
|
|
It bombs (RuntimeError) if both 'filename' and 'file' are None;
|
||
|
|
'filename' should be a string, and 'file' a file object (or
|
||
|
|
something that provides 'readline()' and 'close()' methods). It is
|
||
|
|
recommended that you supply at least 'filename', so that TextFile
|
||
|
|
can include it in warning messages. If 'file' is not supplied,
|
||
|
|
TextFile creates its own using 'io.open()'.
|
||
|
|
|
||
|
|
The options are all boolean, and affect the value returned by
|
||
|
|
'readline()':
|
||
|
|
strip_comments [default: true]
|
||
|
|
strip from "#" to end-of-line, as well as any whitespace
|
||
|
|
leading up to the "#" -- unless it is escaped by a backslash
|
||
|
|
lstrip_ws [default: false]
|
||
|
|
strip leading whitespace from each line before returning it
|
||
|
|
rstrip_ws [default: true]
|
||
|
|
strip trailing whitespace (including line terminator!) from
|
||
|
|
each line before returning it
|
||
|
|
skip_blanks [default: true}
|
||
|
|
skip lines that are empty *after* stripping comments and
|
||
|
|
whitespace. (If both lstrip_ws and rstrip_ws are false,
|
||
|
|
then some lines may consist of solely whitespace: these will
|
||
|
|
*not* be skipped, even if 'skip_blanks' is true.)
|
||
|
|
join_lines [default: false]
|
||
|
|
if a backslash is the last non-newline character on a line
|
||
|
|
after stripping comments and whitespace, join the following line
|
||
|
|
to it to form one "logical line"; if N consecutive lines end
|
||
|
|
with a backslash, then N+1 physical lines will be joined to
|
||
|
|
form one logical line.
|
||
|
|
collapse_join [default: false]
|
||
|
|
strip leading whitespace from lines that are joined to their
|
||
|
|
predecessor; only matters if (join_lines and not lstrip_ws)
|
||
|
|
errors [default: 'strict']
|
||
|
|
error handler used to decode the file content
|
||
|
|
|
||
|
|
Note that since 'rstrip_ws' can strip the trailing newline, the
|
||
|
|
semantics of 'readline()' must differ from those of the builtin file
|
||
|
|
object's 'readline()' method! In particular, 'readline()' returns
|
||
|
|
None for end-of-file: an empty string might just be a blank line (or
|
||
|
|
an all-whitespace line), if 'rstrip_ws' is true but 'skip_blanks' is
|
||
|
|
not."""
|
||
|
|
|
||
|
|
default_options = {
|
||
|
|
'strip_comments': 1,
|
||
|
|
'skip_blanks': 1,
|
||
|
|
'lstrip_ws': 0,
|
||
|
|
'rstrip_ws': 1,
|
||
|
|
'join_lines': 0,
|
||
|
|
'collapse_join': 0,
|
||
|
|
'errors': 'strict',
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, filename=None, file=None, **options):
|
||
|
|
"""Construct a new TextFile object. At least one of 'filename'
|
||
|
|
(a string) and 'file' (a file-like object) must be supplied.
|
||
|
|
They keyword argument options are described above and affect
|
||
|
|
the values returned by 'readline()'."""
|
||
|
|
if filename is None and file is None:
|
||
|
|
raise RuntimeError(
|
||
|
|
"you must supply either or both of 'filename' and 'file'"
|
||
|
|
)
|
||
|
|
|
||
|
|
# set values for all options -- either from client option hash
|
||
|
|
# or fallback to default_options
|
||
|
|
for opt in self.default_options.keys():
|
||
|
|
if opt in options:
|
||
|
|
setattr(self, opt, options[opt])
|
||
|
|
else:
|
||
|
|
setattr(self, opt, self.default_options[opt])
|
||
|
|
|
||
|
|
# sanity check client option hash
|
||
|
|
for opt in options.keys():
|
||
|
|
if opt not in self.default_options:
|
||
|
|
raise KeyError("invalid TextFile option '%s'" % opt)
|
||
|
|
|
||
|
|
if file is None:
|
||
|
|
self.open(filename)
|
||
|
|
else:
|
||
|
|
self.filename = filename
|
||
|
|
self.file = file
|
||
|
|
self.current_line = 0 # assuming that file is at BOF!
|
||
|
|
|
||
|
|
# 'linebuf' is a stack of lines that will be emptied before we
|
||
|
|
# actually read from the file; it's only populated by an
|
||
|
|
# 'unreadline()' operation
|
||
|
|
self.linebuf = []
|
||
|
|
|
||
|
|
def open(self, filename):
|
||
|
|
"""Open a new file named 'filename'. This overrides both the
|
||
|
|
'filename' and 'file' arguments to the constructor."""
|
||
|
|
self.filename = filename
|
||
|
|
self.file = open(self.filename, errors=self.errors)
|
||
|
|
self.current_line = 0
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
"""Close the current file and forget everything we know about it
|
||
|
|
(filename, current line number)."""
|
||
|
|
file = self.file
|
||
|
|
self.file = None
|
||
|
|
self.filename = None
|
||
|
|
self.current_line = None
|
||
|
|
file.close()
|
||
|
|
|
||
|
|
def gen_error(self, msg, line=None):
|
||
|
|
outmsg = []
|
||
|
|
if line is None:
|
||
|
|
line = self.current_line
|
||
|
|
outmsg.append(self.filename + ", ")
|
||
|
|
if isinstance(line, (list, tuple)):
|
||
|
|
outmsg.append("lines %d-%d: " % tuple(line))
|
||
|
|
else:
|
||
|
|
outmsg.append("line %d: " % line)
|
||
|
|
outmsg.append(str(msg))
|
||
|
|
return "".join(outmsg)
|
||
|
|
|
||
|
|
def error(self, msg, line=None):
|
||
|
|
raise ValueError("error: " + self.gen_error(msg, line))
|
||
|
|
|
||
|
|
def warn(self, msg, line=None):
|
||
|
|
"""Print (to stderr) a warning message tied to the current logical
|
||
|
|
line in the current file. If the current logical line in the
|
||
|
|
file spans multiple physical lines, the warning refers to the
|
||
|
|
whole range, eg. "lines 3-5". If 'line' supplied, it overrides
|
||
|
|
the current line number; it may be a list or tuple to indicate a
|
||
|
|
range of physical lines, or an integer for a single physical
|
||
|
|
line."""
|
||
|
|
sys.stderr.write("warning: " + self.gen_error(msg, line) + "\n")
|
||
|
|
|
||
|
|
def readline(self): # noqa: C901
|
||
|
|
"""Read and return a single logical line from the current file (or
|
||
|
|
from an internal buffer if lines have previously been "unread"
|
||
|
|
with 'unreadline()'). If the 'join_lines' option is true, this
|
||
|
|
may involve reading multiple physical lines concatenated into a
|
||
|
|
single string. Updates the current line number, so calling
|
||
|
|
'warn()' after 'readline()' emits a warning about the physical
|
||
|
|
line(s) just read. Returns None on end-of-file, since the empty
|
||
|
|
string can occur if 'rstrip_ws' is true but 'strip_blanks' is
|
||
|
|
not."""
|
||
|
|
# If any "unread" lines waiting in 'linebuf', return the top
|
||
|
|
# one. (We don't actually buffer read-ahead data -- lines only
|
||
|
|
# get put in 'linebuf' if the client explicitly does an
|
||
|
|
# 'unreadline()'.
|
||
|
|
if self.linebuf:
|
||
|
|
line = self.linebuf[-1]
|
||
|
|
del self.linebuf[-1]
|
||
|
|
return line
|
||
|
|
|
||
|
|
buildup_line = ''
|
||
|
|
|
||
|
|
while True:
|
||
|
|
# read the line, make it None if EOF
|
||
|
|
line = self.file.readline()
|
||
|
|
if line == '':
|
||
|
|
line = None
|
||
|
|
|
||
|
|
if self.strip_comments and line:
|
||
|
|
|
||
|
|
# Look for the first "#" in the line. If none, never
|
||
|
|
# mind. If we find one and it's the first character, or
|
||
|
|
# is not preceded by "\", then it starts a comment --
|
||
|
|
# strip the comment, strip whitespace before it, and
|
||
|
|
# carry on. Otherwise, it's just an escaped "#", so
|
||
|
|
# unescape it (and any other escaped "#"'s that might be
|
||
|
|
# lurking in there) and otherwise leave the line alone.
|
||
|
|
|
||
|
|
pos = line.find("#")
|
||
|
|
if pos == -1: # no "#" -- no comments
|
||
|
|
pass
|
||
|
|
|
||
|
|
# It's definitely a comment -- either "#" is the first
|
||
|
|
# character, or it's elsewhere and unescaped.
|
||
|
|
elif pos == 0 or line[pos - 1] != "\\":
|
||
|
|
# Have to preserve the trailing newline, because it's
|
||
|
|
# the job of a later step (rstrip_ws) to remove it --
|
||
|
|
# and if rstrip_ws is false, we'd better preserve it!
|
||
|
|
# (NB. this means that if the final line is all comment
|
||
|
|
# and has no trailing newline, we will think that it's
|
||
|
|
# EOF; I think that's OK.)
|
||
|
|
eol = (line[-1] == '\n') and '\n' or ''
|
||
|
|
line = line[0:pos] + eol
|
||
|
|
|
||
|
|
# If all that's left is whitespace, then skip line
|
||
|
|
# *now*, before we try to join it to 'buildup_line' --
|
||
|
|
# that way constructs like
|
||
|
|
# hello \\
|
||
|
|
# # comment that should be ignored
|
||
|
|
# there
|
||
|
|
# result in "hello there".
|
||
|
|
if line.strip() == "":
|
||
|
|
continue
|
||
|
|
else: # it's an escaped "#"
|
||
|
|
line = line.replace("\\#", "#")
|
||
|
|
|
||
|
|
# did previous line end with a backslash? then accumulate
|
||
|
|
if self.join_lines and buildup_line:
|
||
|
|
# oops: end of file
|
||
|
|
if line is None:
|
||
|
|
self.warn("continuation line immediately precedes " "end-of-file")
|
||
|
|
return buildup_line
|
||
|
|
|
||
|
|
if self.collapse_join:
|
||
|
|
line = line.lstrip()
|
||
|
|
line = buildup_line + line
|
||
|
|
|
||
|
|
# careful: pay attention to line number when incrementing it
|
||
|
|
if isinstance(self.current_line, list):
|
||
|
|
self.current_line[1] = self.current_line[1] + 1
|
||
|
|
else:
|
||
|
|
self.current_line = [self.current_line, self.current_line + 1]
|
||
|
|
# just an ordinary line, read it as usual
|
||
|
|
else:
|
||
|
|
if line is None: # eof
|
||
|
|
return None
|
||
|
|
|
||
|
|
# still have to be careful about incrementing the line number!
|
||
|
|
if isinstance(self.current_line, list):
|
||
|
|
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)
|
||
|
|
if self.lstrip_ws and self.rstrip_ws:
|
||
|
|
line = line.strip()
|
||
|
|
elif self.lstrip_ws:
|
||
|
|
line = line.lstrip()
|
||
|
|
elif self.rstrip_ws:
|
||
|
|
line = line.rstrip()
|
||
|
|
|
||
|
|
# blank line (whether we rstrip'ed or not)? skip to next line
|
||
|
|
# if appropriate
|
||
|
|
if (line == '' or line == '\n') and self.skip_blanks:
|
||
|
|
continue
|
||
|
|
|
||
|
|
if self.join_lines:
|
||
|
|
if line[-1] == '\\':
|
||
|
|
buildup_line = line[:-1]
|
||
|
|
continue
|
||
|
|
|
||
|
|
if line[-2:] == '\\\n':
|
||
|
|
buildup_line = line[0:-2] + '\n'
|
||
|
|
continue
|
||
|
|
|
||
|
|
# well, I guess there's some actual content there: return it
|
||
|
|
return line
|
||
|
|
|
||
|
|
def readlines(self):
|
||
|
|
"""Read and return the list of all logical lines remaining in the
|
||
|
|
current file."""
|
||
|
|
lines = []
|
||
|
|
while True:
|
||
|
|
line = self.readline()
|
||
|
|
if line is None:
|
||
|
|
return lines
|
||
|
|
lines.append(line)
|
||
|
|
|
||
|
|
def unreadline(self, line):
|
||
|
|
"""Push 'line' (a string) onto an internal buffer that will be
|
||
|
|
checked by future 'readline()' calls. Handy for implementing
|
||
|
|
a parser with line-at-a-time lookahead."""
|
||
|
|
self.linebuf.append(line)
|