-
Notifications
You must be signed in to change notification settings - Fork 159
/
fileinput.py
92 lines (79 loc) · 2.4 KB
/
fileinput.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
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)