Mercurial > cpython-lru_cache-weakref

annotate Lib/fileinput.py @ 22360:c8a6ea00a9e1 legacy-trunk

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
restructure a bit to not rely on test case execution ordering add test case for bug #482460
author Skip Montanaro <skip@pobox.com>
date Sun, 17 Mar 2002 23:03:42 +0000
parents edd9d0b3c47e
children 64eb119b816d 204d46fc54f6
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
1 """Helper class to quickly write a loop over all standard input files.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
2
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
3 Typical use is:
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
4
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
5 import fileinput
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
6 for line in fileinput.input():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
7 process(line)
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
8
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
9 This iterates over the lines of all files listed in sys.argv[1:],
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
10 defaulting to sys.stdin if the list is empty. If a filename is '-' it
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
11 is also replaced by sys.stdin. To specify an alternative list of
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
12 filenames, pass it as the argument to input(). A single file name is
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
13 also allowed.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
14
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
15 Functions filename(), lineno() return the filename and cumulative line
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
16 number of the line that has just been read; filelineno() returns its
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
17 line number in the current file; isfirstline() returns true iff the
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
18 line just read is the first line of its file; isstdin() returns true
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
19 iff the line was read from sys.stdin. Function nextfile() closes the
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
20 current file so that the next iteration will read the first line from
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
21 the next file (if any); lines not read from the file will not count
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
22 towards the cumulative line count; the filename is not changed until
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
23 after the first line of the next file has been read. Function close()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
24 closes the sequence.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
25
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
26 Before any lines have been read, filename() returns None and both line
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
27 numbers are zero; nextfile() has no effect. After all lines have been
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
28 read, filename() and the line number functions return the values
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
29 pertaining to the last line read; nextfile() has no effect.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
30
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
31 All files are opened in text mode. If an I/O error occurs during
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
32 opening or reading a file, the IOError exception is raised.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
33
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
34 If sys.stdin is used more than once, the second and further use will
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
35 return no lines, except perhaps for interactive use, or if it has been
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
36 explicitly reset (e.g. using sys.stdin.seek(0)).
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
37
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
38 Empty files are opened and immediately closed; the only time their
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
39 presence in the list of filenames is noticeable at all is when the
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
40 last file opened is empty.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
41
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
42 It is possible that the last line of a file doesn't end in a newline
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
43 character; otherwise lines are returned including the trailing
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
44 newline.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
45
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
46 Class FileInput is the implementation; its methods filename(),
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
47 lineno(), fileline(), isfirstline(), isstdin(), nextfile() and close()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
48 correspond to the functions in the module. In addition it has a
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
49 readline() method which returns the next input line, and a
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
50 __getitem__() method which implements the sequence behavior. The
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
51 sequence must be accessed in strictly sequential order; sequence
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
52 access and readline() cannot be mixed.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
53
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
54 Optional in-place filtering: if the keyword argument inplace=1 is
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
55 passed to input() or to the FileInput constructor, the file is moved
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
56 to a backup file and standard output is directed to the input file.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
57 This makes it possible to write a filter that rewrites its input file
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
58 in place. If the keyword argument backup=".<some extension>" is also
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
59 given, it specifies the extension for the backup file, and the backup
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
60 file remains around; by default, the extension is ".bak" and it is
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
61 deleted when the output file is closed. In-place filtering is
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
62 disabled when standard input is read. XXX The current implementation
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
63 does not work for MS-DOS 8+3 filesystems.
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
64
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
65 Performance: this module is unfortunately one of the slower ways of
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
66 processing large numbers of input lines. Nevertheless, a significant
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
67 speed-up has been obtained by using readlines(bufsize) instead of
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
68 readline(). A new keyword argument, bufsize=N, is present on the
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
69 input() function and the FileInput() class to override the default
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
70 buffer size.
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
71
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
72 XXX Possible additions:
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
73
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
74 - optional getopt argument processing
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
75 - specify open mode ('r' or 'rb')
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
76 - fileno()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
77 - isatty()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
78 - read(), read(size), even readlines()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
79
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
80 """
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
81
11387
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
82 import sys, os, stat
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
83
16072
bae889f44cd1 more __all__ updates
Skip Montanaro <skip@pobox.com>
parents: 15746
diff changeset
84 __all__ = ["input","close","nextfile","filename","lineno","filelineno",
bae889f44cd1 more __all__ updates
Skip Montanaro <skip@pobox.com>
parents: 15746
diff changeset
85 "isfirstline","isstdin","FileInput"]
bae889f44cd1 more __all__ updates
Skip Montanaro <skip@pobox.com>
parents: 15746
diff changeset
86
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
87 _state = None
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
88
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
89 DEFAULT_BUFSIZE = 8*1024
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
90
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
91 def input(files=None, inplace=0, backup="", bufsize=0):
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
92 global _state
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
93 if _state and _state._file:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
94 raise RuntimeError, "input() already active"
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
95 _state = FileInput(files, inplace, backup, bufsize)
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
96 return _state
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
97
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
98 def close():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
99 global _state
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
100 state = _state
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
101 _state = None
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
102 if state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
103 state.close()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
104
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
105 def nextfile():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
106 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
107 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
108 return _state.nextfile()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
109
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
110 def filename():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
111 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
112 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
113 return _state.filename()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
114
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
115 def lineno():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
116 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
117 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
118 return _state.lineno()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
119
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
120 def filelineno():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
121 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
122 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
123 return _state.filelineno()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
124
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
125 def isfirstline():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
126 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
127 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
128 return _state.isfirstline()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
129
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
130 def isstdin():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
131 if not _state:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
132 raise RuntimeError, "no active input()"
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
133 return _state.isstdin()
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
134
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
135 class FileInput:
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
136
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
137 def __init__(self, files=None, inplace=0, backup="", bufsize=0):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
138 if type(files) == type(''):
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
139 files = (files,)
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
140 else:
12340
e2620aab9f2d Implement suggestion from Lawrence Kesteloot in PR#280, to change the
Guido van Rossum <guido@python.org>
parents: 11387
diff changeset
141 if files is None:
e2620aab9f2d Implement suggestion from Lawrence Kesteloot in PR#280, to change the
Guido van Rossum <guido@python.org>
parents: 11387
diff changeset
142 files = sys.argv[1:]
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
143 if not files:
12340
e2620aab9f2d Implement suggestion from Lawrence Kesteloot in PR#280, to change the
Guido van Rossum <guido@python.org>
parents: 11387
diff changeset
144 files = ('-',)
e2620aab9f2d Implement suggestion from Lawrence Kesteloot in PR#280, to change the
Guido van Rossum <guido@python.org>
parents: 11387
diff changeset
145 else:
e2620aab9f2d Implement suggestion from Lawrence Kesteloot in PR#280, to change the
Guido van Rossum <guido@python.org>
parents: 11387
diff changeset
146 files = tuple(files)
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
147 self._files = files
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
148 self._inplace = inplace
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
149 self._backup = backup
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
150 self._bufsize = bufsize or DEFAULT_BUFSIZE
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
151 self._savestdout = None
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
152 self._output = None
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
153 self._filename = None
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
154 self._lineno = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
155 self._filelineno = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
156 self._file = None
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
157 self._isstdin = 0
8687
fd49ed65cd20 Add missing initializer for self._backupfilename. Due to Ralph Butler.
Guido van Rossum <guido@python.org>
parents: 7755
diff changeset
158 self._backupfilename = None
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
159 self._buffer = []
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
160 self._bufindex = 0
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
161
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
162 def __del__(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
163 self.close()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
164
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
165 def close(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
166 self.nextfile()
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
167 self._files = ()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
168
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
169 def __getitem__(self, i):
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
170 try:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
171 line = self._buffer[self._bufindex]
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
172 except IndexError:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
173 pass
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
174 else:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
175 self._bufindex += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
176 self._lineno += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
177 self._filelineno += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
178 return line
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
179 if i != self._lineno:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
180 raise RuntimeError, "accessing lines out of order"
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
181 line = self.readline()
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
182 if not line:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
183 raise IndexError, "end of input reached"
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
184 return line
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
185
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
186 def nextfile(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
187 savestdout = self._savestdout
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
188 self._savestdout = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
189 if savestdout:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
190 sys.stdout = savestdout
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
191
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
192 output = self._output
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
193 self._output = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
194 if output:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
195 output.close()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
196
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
197 file = self._file
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
198 self._file = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
199 if file and not self._isstdin:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
200 file.close()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
201
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
202 backupfilename = self._backupfilename
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
203 self._backupfilename = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
204 if backupfilename and not self._backup:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
205 try: os.unlink(backupfilename)
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
206 except: pass
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
207
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
208 self._isstdin = 0
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
209 self._buffer = []
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
210 self._bufindex = 0
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
211
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
212 def readline(self):
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
213 try:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
214 line = self._buffer[self._bufindex]
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
215 except IndexError:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
216 pass
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
217 else:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
218 self._bufindex += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
219 self._lineno += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
220 self._filelineno += 1
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
221 return line
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
222 if not self._file:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
223 if not self._files:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
224 return ""
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
225 self._filename = self._files[0]
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
226 self._files = self._files[1:]
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
227 self._filelineno = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
228 self._file = None
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
229 self._isstdin = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
230 self._backupfilename = 0
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
231 if self._filename == '-':
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
232 self._filename = '<stdin>'
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
233 self._file = sys.stdin
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
234 self._isstdin = 1
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
235 else:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
236 if self._inplace:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
237 self._backupfilename = (
20654
edd9d0b3c47e SF patch #474590 -- RISC OS support
Guido van Rossum <guido@python.org>
parents: 16072
diff changeset
238 self._filename + (self._backup or os.extsep+"bak"))
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
239 try: os.unlink(self._backupfilename)
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
240 except os.error: pass
11387
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
241 # The next few lines may raise IOError
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
242 os.rename(self._filename, self._backupfilename)
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
243 self._file = open(self._backupfilename, "r")
11387
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
244 try:
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
245 perm = os.fstat(self._file.fileno())[stat.ST_MODE]
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
246 except:
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
247 self._output = open(self._filename, "w")
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
248 else:
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
249 fd = os.open(self._filename,
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
250 os.O_CREAT | os.O_WRONLY | os.O_TRUNC,
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
251 perm)
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
252 self._output = os.fdopen(fd, "w")
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
253 try:
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
254 os.chmod(self._filename, perm)
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
255 except:
019cd21c8c9f Fix for PR#111: when using the inplace option, give the new file the
Guido van Rossum <guido@python.org>
parents: 8687
diff changeset
256 pass
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
257 self._savestdout = sys.stdout
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
258 sys.stdout = self._output
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
259 else:
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
260 # This may raise IOError
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
261 self._file = open(self._filename, "r")
15746
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
262 self._buffer = self._file.readlines(self._bufsize)
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
263 self._bufindex = 0
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
264 if not self._buffer:
4dbbf322a9df Speed it up by using readlines(sizehint). It's still slower than
Guido van Rossum <guido@python.org>
parents: 12340
diff changeset
265 self.nextfile()
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
266 # Recursive call
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
267 return self.readline()
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
268
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
269 def filename(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
270 return self._filename
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
271
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
272 def lineno(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
273 return self._lineno
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
274
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
275 def filelineno(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
276 return self._filelineno
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
277
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
278 def isfirstline(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
279 return self._filelineno == 1
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
280
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
281 def isstdin(self):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
282 return self._isstdin
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
283
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
284 def _test():
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
285 import getopt
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
286 inplace = 0
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
287 backup = 0
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
288 opts, args = getopt.getopt(sys.argv[1:], "ib:")
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
289 for o, a in opts:
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
290 if o == '-i': inplace = 1
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
291 if o == '-b': backup = a
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
292 for line in input(args, inplace=inplace, backup=backup):
7755
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
293 if line[-1:] == '\n': line = line[:-1]
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
294 if line[-1:] == '\r': line = line[:-1]
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
295 print "%d: %s[%d]%s %s" % (lineno(), filename(), filelineno(),
fb4b4d069cc7 Mass check-in after untabifying all files that need it.
Guido van Rossum <guido@python.org>
parents: 6620
diff changeset
296 isfirstline() and "*" or "", line)
6620
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
297 print "%d: %s[%d]" % (lineno(), filename(), filelineno())
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
298
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
299 if __name__ == '__main__':
00b1b77c4177 A new standard module, as discussed on comp.lang.python, to simplify
Guido van Rossum <guido@python.org>
parents:
diff changeset
300 _test()