1 :mod:`fileinput` --- Iterate over lines from multiple input streams 2 =================================================================== 3 4 .. module:: fileinput 5 :synopsis: Loop over standard input or a list of files. 6 7 .. moduleauthor:: Guido van Rossum <guido (a] python.org> 8 .. sectionauthor:: Fred L. Drake, Jr. <fdrake (a] acm.org> 9 10 **Source code:** :source:`Lib/fileinput.py` 11 12 -------------- 13 14 This module implements a helper class and functions to quickly write a 15 loop over standard input or a list of files. If you just want to read or 16 write one file see :func:`open`. 17 18 The typical use is:: 19 20 import fileinput 21 for line in fileinput.input(): 22 process(line) 23 24 This iterates over the lines of all files listed in ``sys.argv[1:]``, defaulting 25 to ``sys.stdin`` if the list is empty. If a filename is ``'-'``, it is also 26 replaced by ``sys.stdin``. To specify an alternative list of filenames, pass it 27 as the first argument to :func:`.input`. A single file name is also allowed. 28 29 All files are opened in text mode by default, but you can override this by 30 specifying the *mode* parameter in the call to :func:`.input` or 31 :class:`FileInput`. If an I/O error occurs during opening or reading a file, 32 :exc:`OSError` is raised. 33 34 .. versionchanged:: 3.3 35 :exc:`IOError` used to be raised; it is now an alias of :exc:`OSError`. 36 37 If ``sys.stdin`` is used more than once, the second and further use will return 38 no lines, except perhaps for interactive use, or if it has been explicitly reset 39 (e.g. using ``sys.stdin.seek(0)``). 40 41 Empty files are opened and immediately closed; the only time their presence in 42 the list of filenames is noticeable at all is when the last file opened is 43 empty. 44 45 Lines are returned with any newlines intact, which means that the last line in 46 a file may not have one. 47 48 You can control how files are opened by providing an opening hook via the 49 *openhook* parameter to :func:`fileinput.input` or :class:`FileInput()`. The 50 hook must be a function that takes two arguments, *filename* and *mode*, and 51 returns an accordingly opened file-like object. Two useful hooks are already 52 provided by this module. 53 54 The following function is the primary interface of this module: 55 56 57 .. function:: input(files=None, inplace=False, backup='', bufsize=0, mode='r', openhook=None) 58 59 Create an instance of the :class:`FileInput` class. The instance will be used 60 as global state for the functions of this module, and is also returned to use 61 during iteration. The parameters to this function will be passed along to the 62 constructor of the :class:`FileInput` class. 63 64 The :class:`FileInput` instance can be used as a context manager in the 65 :keyword:`with` statement. In this example, *input* is closed after the 66 :keyword:`with` statement is exited, even if an exception occurs:: 67 68 with fileinput.input(files=('spam.txt', 'eggs.txt')) as f: 69 for line in f: 70 process(line) 71 72 .. versionchanged:: 3.2 73 Can be used as a context manager. 74 75 .. deprecated-removed:: 3.6 3.8 76 The *bufsize* parameter. 77 78 The following functions use the global state created by :func:`fileinput.input`; 79 if there is no active state, :exc:`RuntimeError` is raised. 80 81 82 .. function:: filename() 83 84 Return the name of the file currently being read. Before the first line has 85 been read, returns ``None``. 86 87 88 .. function:: fileno() 89 90 Return the integer "file descriptor" for the current file. When no file is 91 opened (before the first line and between files), returns ``-1``. 92 93 94 .. function:: lineno() 95 96 Return the cumulative line number of the line that has just been read. Before 97 the first line has been read, returns ``0``. After the last line of the last 98 file has been read, returns the line number of that line. 99 100 101 .. function:: filelineno() 102 103 Return the line number in the current file. Before the first line has been 104 read, returns ``0``. After the last line of the last file has been read, 105 returns the line number of that line within the file. 106 107 108 .. function:: isfirstline() 109 110 Returns true if the line just read is the first line of its file, otherwise 111 returns false. 112 113 114 .. function:: isstdin() 115 116 Returns true if the last line was read from ``sys.stdin``, otherwise returns 117 false. 118 119 120 .. function:: nextfile() 121 122 Close the current file so that the next iteration will read the first line from 123 the next file (if any); lines not read from the file will not count towards the 124 cumulative line count. The filename is not changed until after the first line 125 of the next file has been read. Before the first line has been read, this 126 function has no effect; it cannot be used to skip the first file. After the 127 last line of the last file has been read, this function has no effect. 128 129 130 .. function:: close() 131 132 Close the sequence. 133 134 The class which implements the sequence behavior provided by the module is 135 available for subclassing as well: 136 137 138 .. class:: FileInput(files=None, inplace=False, backup='', bufsize=0, mode='r', openhook=None) 139 140 Class :class:`FileInput` is the implementation; its methods :meth:`filename`, 141 :meth:`fileno`, :meth:`lineno`, :meth:`filelineno`, :meth:`isfirstline`, 142 :meth:`isstdin`, :meth:`nextfile` and :meth:`close` correspond to the 143 functions of the same name in the module. In addition it has a 144 :meth:`~io.TextIOBase.readline` method which returns the next input line, 145 and a :meth:`__getitem__` method which implements the sequence behavior. 146 The sequence must be accessed in strictly sequential order; random access 147 and :meth:`~io.TextIOBase.readline` cannot be mixed. 148 149 With *mode* you can specify which file mode will be passed to :func:`open`. It 150 must be one of ``'r'``, ``'rU'``, ``'U'`` and ``'rb'``. 151 152 The *openhook*, when given, must be a function that takes two arguments, 153 *filename* and *mode*, and returns an accordingly opened file-like object. You 154 cannot use *inplace* and *openhook* together. 155 156 A :class:`FileInput` instance can be used as a context manager in the 157 :keyword:`with` statement. In this example, *input* is closed after the 158 :keyword:`with` statement is exited, even if an exception occurs:: 159 160 with FileInput(files=('spam.txt', 'eggs.txt')) as input: 161 process(input) 162 163 .. versionchanged:: 3.2 164 Can be used as a context manager. 165 166 .. deprecated:: 3.4 167 The ``'rU'`` and ``'U'`` modes. 168 169 .. deprecated-removed:: 3.6 3.8 170 The *bufsize* parameter. 171 172 173 **Optional in-place filtering:** if the keyword argument ``inplace=True`` is 174 passed to :func:`fileinput.input` or to the :class:`FileInput` constructor, the 175 file is moved to a backup file and standard output is directed to the input file 176 (if a file of the same name as the backup file already exists, it will be 177 replaced silently). This makes it possible to write a filter that rewrites its 178 input file in place. If the *backup* parameter is given (typically as 179 ``backup='.<some extension>'``), it specifies the extension for the backup file, 180 and the backup file remains around; by default, the extension is ``'.bak'`` and 181 it is deleted when the output file is closed. In-place filtering is disabled 182 when standard input is read. 183 184 185 The two following opening hooks are provided by this module: 186 187 .. function:: hook_compressed(filename, mode) 188 189 Transparently opens files compressed with gzip and bzip2 (recognized by the 190 extensions ``'.gz'`` and ``'.bz2'``) using the :mod:`gzip` and :mod:`bz2` 191 modules. If the filename extension is not ``'.gz'`` or ``'.bz2'``, the file is 192 opened normally (ie, using :func:`open` without any decompression). 193 194 Usage example: ``fi = fileinput.FileInput(openhook=fileinput.hook_compressed)`` 195 196 197 .. function:: hook_encoded(encoding, errors=None) 198 199 Returns a hook which opens each file with :func:`open`, using the given 200 *encoding* and *errors* to read the file. 201 202 Usage example: ``fi = 203 fileinput.FileInput(openhook=fileinput.hook_encoded("utf-8", 204 "surrogateescape"))`` 205 206 .. versionchanged:: 3.6 207 Added the optional *errors* parameter. 208