Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
3543085
commit 44c49a0
Showing
1 changed file
with
92 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,92 @@ | ||
""" | ||
This is a simplistic implementation of fileinput module | ||
""" | ||
import sys | ||
|
||
class FileInput(object): | ||
def __init__(self, files=None): | ||
if not files: | ||
self._files = sys.argv[1:] | ||
if not self._files: | ||
self._files = ("_", ) | ||
elif isinstance(files, list): | ||
self._files = tuple(files) | ||
elif isinstance(files, basestring): | ||
self._files = (files, ) | ||
else: | ||
raise ValueError("Invalid input") | ||
|
||
self._file = None | ||
self._fileindex = 0 | ||
self._lineindex = 0 | ||
self._buffer = None | ||
self._filename = None | ||
self._is_stdin = False | ||
|
||
def __iter__(self): | ||
return self | ||
|
||
def next(self): | ||
line = self.get_next_line() | ||
if line: | ||
return line | ||
else: | ||
raise StopIteration | ||
|
||
def get_next_line(self): | ||
if not self._buffer: | ||
if not self.load_next_file(): | ||
return None | ||
try: | ||
line = self._buffer[self._lineindex] | ||
except IndexError: | ||
if self.load_next_file(): | ||
return self.get_next_line() | ||
else: | ||
return None | ||
else: | ||
self._lineindex += 1 | ||
return line | ||
|
||
def load_next_file(self): | ||
if not self._files: | ||
return False | ||
_filename = self._files[0] | ||
self._files = self._files[1:] | ||
self._fileindex += 1 | ||
|
||
if _filename == "_": | ||
self._filename = "<std input>" | ||
self._file = sys.stdin | ||
self._is_stdin = True | ||
else: | ||
self._filename = _filename | ||
if self._file and not self._is_stdin: | ||
self._file.close() | ||
try: | ||
self._file = open(_filename, "r") | ||
except: | ||
return self.load_next_file() | ||
self._lineindex = 0 | ||
self._buffer = self._file.readlines() | ||
if not self._buffer: | ||
return load_next_file() | ||
return True | ||
|
||
def lineno(self): | ||
return self._lineindex | ||
|
||
def fileno(self): | ||
return self._fileindex | ||
|
||
def filename(self): | ||
return self._filename | ||
|
||
if __name__ == "__main__": | ||
f = FileInput() | ||
f = FileInput("../test.js") | ||
f = FileInput(["../test.js", "../weixin.txt", "foobar.txt"]) | ||
for line in f: | ||
print "%s(%d):%d\t%s"%(f.filename(), f.fileno(), f.lineno(), line) | ||
|
||
|