27ccc3bfedfd7a5b2584258b86228f4df2f89c78
"""Helper class to quickly write a loop over all standard input files.
for line in fileinput.input():
This iterates over the lines of all files listed in sys.argv[1:],
defaulting to sys.stdin if the list is empty. If a filename is '-' it
is also replaced by sys.stdin. To specify an alternative list of
filenames, pass it as the argument to input(). A single file name is
Functions filename(), lineno() return the filename and cumulative line
number of the line that has just been read; filelineno() returns its
line number in the current file; isfirstline() returns true iff the
line just read is the first line of its file; isstdin() returns true
iff the line was read from sys.stdin. Function nextfile() closes the
current file so that the next iteration will read the first line from
the next file (if any); lines not read from the file will not count
towards the cumulative line count; the filename is not changed until
after the first line of the next file has been read. Function close()
Before any lines have been read, filename() returns None and both line
numbers are zero; nextfile() has no effect. After all lines have been
read, filename() and the line number functions return the values
pertaining to the last line read; nextfile() has no effect.
All files are opened in text mode. If an I/O error occurs during
opening or reading a file, the IOError exception is raised.
If sys.stdin is used more than once, the second and further use will
return no lines, except perhaps for interactive use, or if it has been
explicitly reset (e.g. using sys.stdin.seek(0)).
Empty files are opened and immediately closed; the only time their
presence in the list of filenames is noticeable at all is when the
last file opened is empty.
It is possible that the last line of a file doesn't end in a newline
character; otherwise lines are returned including the trailing
Class FileInput is the implementation; its methods filename(),
lineno(), fileline(), isfirstline(), isstdin(), nextfile() and close()
correspond to the functions in the module. In addition it has a
readline() method which returns the next input line, and a
__getitem__() method which implements the sequence behavior. The
sequence must be accessed in strictly sequential order; sequence
access and readline() cannot be mixed.
Optional in-place filtering: if the keyword argument inplace=1 is
passed to input() or to the FileInput constructor, the file is moved
to a backup file and standard output is directed to the input file.
This makes it possible to write a filter that rewrites its input file
in place. If the keyword argument backup=".<some extension>" is also
given, it specifies the extension for the backup file, and the backup
file remains around; by default, the extension is ".bak" and it is
deleted when the output file is closed. In-place filtering is
disabled when standard input is read. XXX The current implementation
does not work for MS-DOS 8+3 filesystems.
Performance: this module is unfortunately one of the slower ways of
processing large numbers of input lines. Nevertheless, a significant
speed-up has been obtained by using readlines(bufsize) instead of
readline(). A new keyword argument, bufsize=N, is present on the
input() function and the FileInput() class to override the default
- optional getopt argument processing
- specify open mode ('r' or 'rb')
- read(), read(size), even readlines()
__all__
= ["input","close","nextfile","filename","lineno","filelineno",
"isfirstline","isstdin","FileInput"]
def input(files
=None, inplace
=0, backup
="", bufsize
=0):
"""input([files[, inplace[, backup]]])
Create an instance of the FileInput class. The instance will be used
as global state for the functions of this module, and is also returned
to use during iteration. The parameters to this function will be passed
along to the constructor of the FileInput class.
if _state
and _state
._file
:
raise RuntimeError, "input() already active"
_state
= FileInput(files
, inplace
, backup
, bufsize
)
"""Close the sequence."""
Close the current file so that the next iteration will read the first
line from the next file (if any); lines not read from the file will
not count towards the cumulative line count. The filename is not
changed until after the first line of the next file has been read.
Before the first line has been read, this function has no effect;
it cannot be used to skip the first file. After the last line of the
last file has been read, this function has no effect.
raise RuntimeError, "no active input()"
Return the name of the file currently being read.
Before the first line has been read, returns None.
raise RuntimeError, "no active input()"
Return the cumulative line number of the line that has just been read.
Before the first line has been read, returns 0. After the last line
of the last file has been read, returns the line number of that line.
raise RuntimeError, "no active input()"
Return the line number in the current file. Before the first line
has been read, returns 0. After the last line of the last file has
been read, returns the line number of that line within the file.
raise RuntimeError, "no active input()"
return _state
.filelineno()
Returns true the line just read is the first line of its file,
raise RuntimeError, "no active input()"
return _state
.isfirstline()
Returns true if the last line was read from sys.stdin,
raise RuntimeError, "no active input()"
"""class FileInput([files[, inplace[, backup]]])
Class FileInput is the implementation of the module; its methods
filename(), lineno(), fileline(), isfirstline(), isstdin(), nextfile()
and close() correspond to the functions of the same name in the module.
In addition it has a readline() method which returns the next
input line, and a __getitem__() method which implements the
sequence behavior. The sequence must be accessed in strictly
sequential order; random access and readline() cannot be mixed.
def __init__(self
, files
=None, inplace
=0, backup
="", bufsize
=0):
if type(files
) == type(''):
self
._bufsize
= bufsize
or DEFAULT_BUFSIZE
self
._backupfilename
= None
line
= self
._buffer
[self
._bufindex
]
def __getitem__(self
, i
):
raise RuntimeError, "accessing lines out of order"
raise IndexError, "end of input reached"
savestdout
= self
._savestdout
if file and not self
._isstdin
:
backupfilename
= self
._backupfilename
if backupfilename
and not self
._backup
:
try: os
.unlink(backupfilename
)
line
= self
._buffer
[self
._bufindex
]
self
._filename
= self
._files
[0]
self
._files
= self
._files
[1:]
if self
._filename
== '-':
self
._filename
= '<stdin>'
self
._filename
+ (self
._backup
or os
.extsep
+"bak"))
try: os
.unlink(self
._backupfilename
)
# The next few lines may raise IOError
os
.rename(self
._filename
, self
._backupfilename
)
self
._file
= open(self
._backupfilename
, "r")
perm
= os
.fstat(self
._file
.fileno()).st_mode
self
._output
= open(self
._filename
, "w")
fd
= os
.open(self
._filename
,
os
.O_CREAT | os
.O_WRONLY | os
.O_TRUNC
,
self
._output
= os
.fdopen(fd
, "w")
os
.chmod(self
._filename
, perm
)
self
._savestdout
= sys
.stdout
sys
.stdout
= self
._output
self
._file
= open(self
._filename
, "r")
self
._buffer
= self
._file
.readlines(self
._bufsize
)
return self
._filelineno
== 1
opts
, args
= getopt
.getopt(sys
.argv
[1:], "ib:")
if o
== '-i': inplace
= 1
for line
in input(args
, inplace
=inplace
, backup
=backup
):
if line
[-1:] == '\n': line
= line
[:-1]
if line
[-1:] == '\r': line
= line
[:-1]
print "%d: %s[%d]%s %s" % (lineno(), filename(), filelineno(),
isfirstline() and "*" or "", line
)
print "%d: %s[%d]" % (lineno(), filename(), filelineno())
if __name__
== '__main__':