blob: 54de5040ae9e363ab902a28735a52f3738658fc6 [file] [log] [blame]
Christian Heimes1a6387e2008-03-26 12:49:49 +00001"""
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00002The io module provides the Python interfaces to stream handling. The
3builtin open function is defined in this module.
4
5At the top of the I/O hierarchy is the abstract base class IOBase. It
6defines the basic interface to a stream. Note, however, that there is no
Jesus Cea585ad8a2009-07-02 15:37:21 +00007separation between reading and writing to streams; implementations are
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00008allowed to throw an IOError if they do not support a given operation.
9
10Extending IOBase is RawIOBase which deals simply with the reading and
11writing of raw bytes to a stream. FileIO subclasses RawIOBase to provide
12an interface to OS files.
13
14BufferedIOBase deals with buffering on a raw byte stream (RawIOBase). Its
15subclasses, BufferedWriter, BufferedReader, and BufferedRWPair buffer
16streams that are readable, writable, and both respectively.
17BufferedRandom provides a buffered interface to random access
18streams. BytesIO is a simple stream of in-memory bytes.
19
20Another IOBase subclass, TextIOBase, deals with the encoding and decoding
21of streams into text. TextIOWrapper, which extends it, is a buffered text
22interface to a buffered raw stream (`BufferedIOBase`). Finally, StringIO
23is a in-memory stream for text.
24
25Argument names are not part of the specification, and only the arguments
26of open() are intended to be used as keyword arguments.
27
28data:
29
30DEFAULT_BUFFER_SIZE
31
32 An int containing the default buffer size used by the module's buffered
33 I/O classes. open() uses the file's blksize (as obtained by os.stat) if
34 possible.
35"""
36# New I/O library conforming to PEP 3116.
37
38# This is a prototype; hopefully eventually some of this will be
39# reimplemented in C.
40
41# XXX edge cases when switching between reading/writing
42# XXX need to support 1 meaning line-buffered
43# XXX whenever an argument is None, use the default value
44# XXX read/write ops should check readable/writable
45# XXX buffered readinto should work with arbitrary buffer objects
46# XXX use incremental encoder for text output, at least for UTF-16 and UTF-8-SIG
47# XXX check writable, readable and seekable in appropriate places
Christian Heimes3784c6b2008-03-26 23:13:59 +000048from __future__ import print_function
49from __future__ import unicode_literals
Christian Heimes1a6387e2008-03-26 12:49:49 +000050
51__author__ = ("Guido van Rossum <guido@python.org>, "
52 "Mike Verdone <mike.verdone@gmail.com>, "
53 "Mark Russell <mark.russell@zen.co.uk>")
54
55__all__ = ["BlockingIOError", "open", "IOBase", "RawIOBase", "FileIO",
56 "BytesIO", "StringIO", "BufferedIOBase",
57 "BufferedReader", "BufferedWriter", "BufferedRWPair",
58 "BufferedRandom", "TextIOBase", "TextIOWrapper"]
59
60import os
61import abc
Christian Heimes1a6387e2008-03-26 12:49:49 +000062import codecs
63import _fileio
Antoine Pitrou11ec65d2008-08-14 21:04:30 +000064import threading
Christian Heimes1a6387e2008-03-26 12:49:49 +000065
66# open() uses st_blksize whenever we can
67DEFAULT_BUFFER_SIZE = 8 * 1024 # bytes
68
69# py3k has only new style classes
70__metaclass__ = type
71
72class BlockingIOError(IOError):
73
74 """Exception raised when I/O would block on a non-blocking I/O stream."""
75
76 def __init__(self, errno, strerror, characters_written=0):
77 IOError.__init__(self, errno, strerror)
78 self.characters_written = characters_written
79
80
81def open(file, mode="r", buffering=None, encoding=None, errors=None,
82 newline=None, closefd=True):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +000083 r"""Open file and return a stream. If the file cannot be opened, an IOError is
84 raised.
Christian Heimes1a6387e2008-03-26 12:49:49 +000085
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +000086 file is either a string giving the name (and the path if the file
87 isn't in the current working directory) of the file to be opened or an
88 integer file descriptor of the file to be wrapped. (If a file
89 descriptor is given, it is closed when the returned I/O object is
90 closed, unless closefd is set to False.)
Christian Heimes1a6387e2008-03-26 12:49:49 +000091
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +000092 mode is an optional string that specifies the mode in which the file
93 is opened. It defaults to 'r' which means open for reading in text
94 mode. Other common values are 'w' for writing (truncating the file if
95 it already exists), and 'a' for appending (which on some Unix systems,
96 means that all writes append to the end of the file regardless of the
97 current seek position). In text mode, if encoding is not specified the
98 encoding used is platform dependent. (For reading and writing raw
99 bytes use binary mode and leave encoding unspecified.) The available
100 modes are:
Christian Heimes1a6387e2008-03-26 12:49:49 +0000101
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000102 ========= ===============================================================
103 Character Meaning
104 --------- ---------------------------------------------------------------
105 'r' open for reading (default)
106 'w' open for writing, truncating the file first
107 'a' open for writing, appending to the end of the file if it exists
108 'b' binary mode
109 't' text mode (default)
110 '+' open a disk file for updating (reading and writing)
111 'U' universal newline mode (for backwards compatibility; unneeded
112 for new code)
113 ========= ===============================================================
Christian Heimes1a6387e2008-03-26 12:49:49 +0000114
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000115 The default mode is 'rt' (open for reading text). For binary random
116 access, the mode 'w+b' opens and truncates the file to 0 bytes, while
117 'r+b' opens the file without truncation.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000118
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000119 Python distinguishes between files opened in binary and text modes,
120 even when the underlying operating system doesn't. Files opened in
121 binary mode (appending 'b' to the mode argument) return contents as
122 bytes objects without any decoding. In text mode (the default, or when
123 't' is appended to the mode argument), the contents of the file are
124 returned as strings, the bytes having been first decoded using a
125 platform-dependent encoding or using the specified encoding if given.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000126
Antoine Pitroub9767262009-12-19 21:03:36 +0000127 buffering is an optional integer used to set the buffering policy.
128 Pass 0 to switch buffering off (only allowed in binary mode), 1 to select
129 line buffering (only usable in text mode), and an integer > 1 to indicate
130 the size of a fixed-size chunk buffer. When no buffering argument is
131 given, the default buffering policy works as follows:
132
133 * Binary files are buffered in fixed-size chunks; the size of the buffer
134 is chosen using a heuristic trying to determine the underlying device's
135 "block size" and falling back on `io.DEFAULT_BUFFER_SIZE`.
136 On many systems, the buffer will typically be 4096 or 8192 bytes long.
137
138 * "Interactive" text files (files for which isatty() returns True)
139 use line buffering. Other text files use the policy described above
140 for binary files.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000141
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000142 encoding is the name of the encoding used to decode or encode the
143 file. This should only be used in text mode. The default encoding is
144 platform dependent, but any encoding supported by Python can be
145 passed. See the codecs module for the list of supported encodings.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000146
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000147 errors is an optional string that specifies how encoding errors are to
148 be handled---this argument should not be used in binary mode. Pass
149 'strict' to raise a ValueError exception if there is an encoding error
150 (the default of None has the same effect), or pass 'ignore' to ignore
151 errors. (Note that ignoring encoding errors can lead to data loss.)
152 See the documentation for codecs.register for a list of the permitted
153 encoding error strings.
154
155 newline controls how universal newlines works (it only applies to text
156 mode). It can be None, '', '\n', '\r', and '\r\n'. It works as
157 follows:
158
159 * On input, if newline is None, universal newlines mode is
160 enabled. Lines in the input can end in '\n', '\r', or '\r\n', and
161 these are translated into '\n' before being returned to the
162 caller. If it is '', universal newline mode is enabled, but line
163 endings are returned to the caller untranslated. If it has any of
164 the other legal values, input lines are only terminated by the given
165 string, and the line ending is returned to the caller untranslated.
166
167 * On output, if newline is None, any '\n' characters written are
168 translated to the system default line separator, os.linesep. If
169 newline is '', no translation takes place. If newline is any of the
170 other legal values, any '\n' characters written are translated to
171 the given string.
172
173 If closefd is False, the underlying file descriptor will be kept open
174 when the file is closed. This does not work when a file name is given
175 and must be True in that case.
176
177 open() returns a file object whose type depends on the mode, and
178 through which the standard file operations such as reading and writing
179 are performed. When open() is used to open a file in a text mode ('w',
180 'r', 'wt', 'rt', etc.), it returns a TextIOWrapper. When used to open
181 a file in a binary mode, the returned class varies: in read binary
182 mode, it returns a BufferedReader; in write binary and append binary
183 modes, it returns a BufferedWriter, and in read/write mode, it returns
184 a BufferedRandom.
185
186 It is also possible to use a string or bytearray as a file for both
187 reading and writing. For strings StringIO can be used like a file
188 opened in a text mode, and for bytes a BytesIO can be used like a file
189 opened in a binary mode.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000190 """
Christian Heimes3784c6b2008-03-26 23:13:59 +0000191 if not isinstance(file, (basestring, int)):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000192 raise TypeError("invalid file: %r" % file)
Christian Heimes3784c6b2008-03-26 23:13:59 +0000193 if not isinstance(mode, basestring):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000194 raise TypeError("invalid mode: %r" % mode)
195 if buffering is not None and not isinstance(buffering, int):
196 raise TypeError("invalid buffering: %r" % buffering)
Christian Heimes3784c6b2008-03-26 23:13:59 +0000197 if encoding is not None and not isinstance(encoding, basestring):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000198 raise TypeError("invalid encoding: %r" % encoding)
Christian Heimes3784c6b2008-03-26 23:13:59 +0000199 if errors is not None and not isinstance(errors, basestring):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000200 raise TypeError("invalid errors: %r" % errors)
201 modes = set(mode)
202 if modes - set("arwb+tU") or len(mode) > len(modes):
203 raise ValueError("invalid mode: %r" % mode)
204 reading = "r" in modes
205 writing = "w" in modes
206 appending = "a" in modes
207 updating = "+" in modes
208 text = "t" in modes
209 binary = "b" in modes
210 if "U" in modes:
211 if writing or appending:
212 raise ValueError("can't use U and writing mode at once")
213 reading = True
214 if text and binary:
215 raise ValueError("can't have text and binary mode at once")
216 if reading + writing + appending > 1:
217 raise ValueError("can't have read/write/append mode at once")
218 if not (reading or writing or appending):
219 raise ValueError("must have exactly one of read/write/append mode")
220 if binary and encoding is not None:
221 raise ValueError("binary mode doesn't take an encoding argument")
222 if binary and errors is not None:
223 raise ValueError("binary mode doesn't take an errors argument")
224 if binary and newline is not None:
225 raise ValueError("binary mode doesn't take a newline argument")
226 raw = FileIO(file,
227 (reading and "r" or "") +
228 (writing and "w" or "") +
229 (appending and "a" or "") +
230 (updating and "+" or ""),
231 closefd)
232 if buffering is None:
233 buffering = -1
234 line_buffering = False
235 if buffering == 1 or buffering < 0 and raw.isatty():
236 buffering = -1
237 line_buffering = True
238 if buffering < 0:
239 buffering = DEFAULT_BUFFER_SIZE
240 try:
241 bs = os.fstat(raw.fileno()).st_blksize
242 except (os.error, AttributeError):
243 pass
244 else:
245 if bs > 1:
246 buffering = bs
247 if buffering < 0:
248 raise ValueError("invalid buffering size")
249 if buffering == 0:
250 if binary:
Christian Heimes1a6387e2008-03-26 12:49:49 +0000251 return raw
252 raise ValueError("can't have unbuffered text I/O")
253 if updating:
254 buffer = BufferedRandom(raw, buffering)
255 elif writing or appending:
256 buffer = BufferedWriter(raw, buffering)
257 elif reading:
258 buffer = BufferedReader(raw, buffering)
259 else:
260 raise ValueError("unknown mode: %r" % mode)
261 if binary:
Christian Heimes1a6387e2008-03-26 12:49:49 +0000262 return buffer
263 text = TextIOWrapper(buffer, encoding, errors, newline, line_buffering)
Christian Heimes1a6387e2008-03-26 12:49:49 +0000264 text.mode = mode
265 return text
266
267class _DocDescriptor:
268 """Helper for builtins.open.__doc__
269 """
270 def __get__(self, obj, typ):
271 return (
272 "open(file, mode='r', buffering=None, encoding=None, "
273 "errors=None, newline=None, closefd=True)\n\n" +
274 open.__doc__)
275
276class OpenWrapper:
277 """Wrapper for builtins.open
278
279 Trick so that open won't become a bound method when stored
280 as a class variable (as dumbdbm does).
281
282 See initstdio() in Python/pythonrun.c.
283 """
284 __doc__ = _DocDescriptor()
285
286 def __new__(cls, *args, **kwargs):
287 return open(*args, **kwargs)
288
289
290class UnsupportedOperation(ValueError, IOError):
291 pass
292
293
294class IOBase(object):
295
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000296 """The abstract base class for all I/O classes, acting on streams of
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000297 bytes. There is no public constructor.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000298
299 This class provides dummy implementations for many methods that
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000300 derived classes can override selectively; the default implementations
301 represent a file that cannot be read, written or seeked.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000302
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000303 Even though IOBase does not declare read, readinto, or write because
304 their signatures will vary, implementations and clients should
305 consider those methods part of the interface. Also, implementations
306 may raise a IOError when operations they do not support are called.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000307
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000308 The basic type used for binary data read from or written to a file is
309 bytes. bytearrays are accepted too, and in some cases (such as
310 readinto) needed. Text I/O classes work with str data.
311
312 Note that calling any method (even inquiries) on a closed stream is
313 undefined. Implementations may raise IOError in this case.
314
315 IOBase (and its subclasses) support the iterator protocol, meaning
316 that an IOBase object can be iterated over yielding the lines in a
317 stream.
318
319 IOBase also supports the :keyword:`with` statement. In this example,
320 fp is closed after the suite of the with statment is complete:
321
322 with open('spam.txt', 'r') as fp:
323 fp.write('Spam and eggs!')
Christian Heimes1a6387e2008-03-26 12:49:49 +0000324 """
325
326 __metaclass__ = abc.ABCMeta
327
328 ### Internal ###
329
330 def _unsupported(self, name):
331 """Internal: raise an exception for unsupported operations."""
332 raise UnsupportedOperation("%s.%s() not supported" %
333 (self.__class__.__name__, name))
334
335 ### Positioning ###
336
337 def seek(self, pos, whence = 0):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000338 """Change stream position.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000339
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000340 Change the stream position to byte offset offset. offset is
341 interpreted relative to the position indicated by whence. Values
342 for whence are:
343
344 * 0 -- start of stream (the default); offset should be zero or positive
345 * 1 -- current stream position; offset may be negative
346 * 2 -- end of stream; offset is usually negative
347
348 Return the new absolute position.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000349 """
350 self._unsupported("seek")
351
352 def tell(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000353 """Return current stream position."""
Christian Heimes1a6387e2008-03-26 12:49:49 +0000354 return self.seek(0, 1)
355
356 def truncate(self, pos = None):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000357 """Truncate file to size bytes.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000358
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000359 Size defaults to the current IO position as reported by tell(). Return
360 the new size.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000361 """
362 self._unsupported("truncate")
363
364 ### Flush and close ###
365
366 def flush(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000367 """Flush write buffers, if applicable.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000368
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000369 This is not implemented for read-only and non-blocking streams.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000370 """
Antoine Pitrou01a255a2010-05-03 16:48:13 +0000371 if self.__closed:
372 raise ValueError("flush of closed file")
373 #self._checkClosed()
Christian Heimes1a6387e2008-03-26 12:49:49 +0000374 # XXX Should this return the number of bytes written???
375
376 __closed = False
377
378 def close(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000379 """Flush and close the IO object.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000380
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000381 This method has no effect if the file is already closed.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000382 """
383 if not self.__closed:
Antoine Pitrou01a255a2010-05-03 16:48:13 +0000384 self.flush()
Christian Heimes1a6387e2008-03-26 12:49:49 +0000385 self.__closed = True
386
387 def __del__(self):
388 """Destructor. Calls close()."""
389 # The try/except block is in case this is called at program
390 # exit time, when it's possible that globals have already been
391 # deleted, and then the close() call might fail. Since
392 # there's nothing we can do about such failures and they annoy
393 # the end users, we suppress the traceback.
394 try:
395 self.close()
396 except:
397 pass
398
399 ### Inquiries ###
400
401 def seekable(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000402 """Return whether object supports random access.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000403
404 If False, seek(), tell() and truncate() will raise IOError.
405 This method may need to do a test seek().
406 """
407 return False
408
409 def _checkSeekable(self, msg=None):
410 """Internal: raise an IOError if file is not seekable
411 """
412 if not self.seekable():
413 raise IOError("File or stream is not seekable."
414 if msg is None else msg)
415
416
417 def readable(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000418 """Return whether object was opened for reading.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000419
420 If False, read() will raise IOError.
421 """
422 return False
423
424 def _checkReadable(self, msg=None):
425 """Internal: raise an IOError if file is not readable
426 """
427 if not self.readable():
428 raise IOError("File or stream is not readable."
429 if msg is None else msg)
430
431 def writable(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000432 """Return whether object was opened for writing.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000433
434 If False, write() and truncate() will raise IOError.
435 """
436 return False
437
438 def _checkWritable(self, msg=None):
439 """Internal: raise an IOError if file is not writable
440 """
441 if not self.writable():
442 raise IOError("File or stream is not writable."
443 if msg is None else msg)
444
445 @property
446 def closed(self):
447 """closed: bool. True iff the file has been closed.
448
449 For backwards compatibility, this is a property, not a predicate.
450 """
451 return self.__closed
452
453 def _checkClosed(self, msg=None):
454 """Internal: raise an ValueError if file is closed
455 """
456 if self.closed:
457 raise ValueError("I/O operation on closed file."
458 if msg is None else msg)
459
460 ### Context manager ###
461
462 def __enter__(self):
463 """Context management protocol. Returns self."""
464 self._checkClosed()
465 return self
466
467 def __exit__(self, *args):
468 """Context management protocol. Calls close()"""
469 self.close()
470
471 ### Lower-level APIs ###
472
473 # XXX Should these be present even if unimplemented?
474
475 def fileno(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000476 """Returns underlying file descriptor if one exists.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000477
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000478 An IOError is raised if the IO object does not use a file descriptor.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000479 """
480 self._unsupported("fileno")
481
482 def isatty(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000483 """Return whether this is an 'interactive' stream.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000484
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000485 Return False if it can't be determined.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000486 """
487 self._checkClosed()
488 return False
489
490 ### Readline[s] and writelines ###
491
492 def readline(self, limit = -1):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000493 r"""Read and return a line from the stream.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000494
495 If limit is specified, at most limit bytes will be read.
496
497 The line terminator is always b'\n' for binary files; for text
498 files, the newlines argument to open can be used to select the line
499 terminator(s) recognized.
500 """
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000501 self._checkClosed()
Christian Heimes1a6387e2008-03-26 12:49:49 +0000502 if hasattr(self, "peek"):
503 def nreadahead():
504 readahead = self.peek(1)
505 if not readahead:
506 return 1
507 n = (readahead.find(b"\n") + 1) or len(readahead)
508 if limit >= 0:
509 n = min(n, limit)
510 return n
511 else:
512 def nreadahead():
513 return 1
514 if limit is None:
515 limit = -1
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000516 if not isinstance(limit, (int, long)):
517 raise TypeError("limit must be an integer")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000518 res = bytearray()
519 while limit < 0 or len(res) < limit:
520 b = self.read(nreadahead())
521 if not b:
522 break
523 res += b
524 if res.endswith(b"\n"):
525 break
526 return bytes(res)
527
528 def __iter__(self):
529 self._checkClosed()
530 return self
531
532 def next(self):
533 line = self.readline()
534 if not line:
535 raise StopIteration
536 return line
537
538 def readlines(self, hint=None):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000539 """Return a list of lines from the stream.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000540
541 hint can be specified to control the number of lines read: no more
542 lines will be read if the total size (in bytes/characters) of all
543 lines so far exceeds hint.
544 """
Christian Heimes1a6387e2008-03-26 12:49:49 +0000545 if hint is None:
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000546 hint = -1
547 if not isinstance(hint, (int, long)):
548 raise TypeError("hint must be an integer")
549 if hint <= 0:
Christian Heimes1a6387e2008-03-26 12:49:49 +0000550 return list(self)
551 n = 0
552 lines = []
553 for line in self:
554 lines.append(line)
555 n += len(line)
556 if n >= hint:
557 break
558 return lines
559
560 def writelines(self, lines):
561 self._checkClosed()
562 for line in lines:
563 self.write(line)
564
565
566class RawIOBase(IOBase):
567
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000568 """Base class for raw binary I/O."""
Christian Heimes1a6387e2008-03-26 12:49:49 +0000569
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000570 # The read() method is implemented by calling readinto(); derived
571 # classes that want to support read() only need to implement
572 # readinto() as a primitive operation. In general, readinto() can be
573 # more efficient than read().
Christian Heimes1a6387e2008-03-26 12:49:49 +0000574
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000575 # (It would be tempting to also provide an implementation of
576 # readinto() in terms of read(), in case the latter is a more suitable
577 # primitive operation, but that would lead to nasty recursion in case
578 # a subclass doesn't implement either.)
Christian Heimes1a6387e2008-03-26 12:49:49 +0000579
580 def read(self, n = -1):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000581 """Read and return up to n bytes.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000582
583 Returns an empty bytes array on EOF, or None if the object is
584 set not to block and has no data to read.
585 """
586 if n is None:
587 n = -1
588 if n < 0:
589 return self.readall()
590 b = bytearray(n.__index__())
591 n = self.readinto(b)
592 del b[n:]
593 return bytes(b)
594
595 def readall(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000596 """Read until EOF, using multiple read() call."""
Christian Heimes1a6387e2008-03-26 12:49:49 +0000597 res = bytearray()
598 while True:
599 data = self.read(DEFAULT_BUFFER_SIZE)
600 if not data:
601 break
602 res += data
603 return bytes(res)
604
605 def readinto(self, b):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000606 """Read up to len(b) bytes into b.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000607
608 Returns number of bytes read (0 for EOF), or None if the object
609 is set not to block as has no data to read.
610 """
611 self._unsupported("readinto")
612
613 def write(self, b):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000614 """Write the given buffer to the IO stream.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000615
616 Returns the number of bytes written, which may be less than len(b).
617 """
618 self._unsupported("write")
619
620
621class FileIO(_fileio._FileIO, RawIOBase):
622
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000623 """Raw I/O implementation for OS files."""
Christian Heimes1a6387e2008-03-26 12:49:49 +0000624
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000625 # This multiply inherits from _FileIO and RawIOBase to make
626 # isinstance(io.FileIO(), io.RawIOBase) return True without requiring
627 # that _fileio._FileIO inherits from io.RawIOBase (which would be hard
628 # to do since _fileio.c is written in C).
Christian Heimes1a6387e2008-03-26 12:49:49 +0000629
Georg Brandld2094602008-12-05 08:51:30 +0000630 def __init__(self, name, mode="r", closefd=True):
631 _fileio._FileIO.__init__(self, name, mode, closefd)
632 self._name = name
633
Christian Heimes1a6387e2008-03-26 12:49:49 +0000634 def close(self):
635 _fileio._FileIO.close(self)
636 RawIOBase.close(self)
637
638 @property
639 def name(self):
640 return self._name
641
Christian Heimes1a6387e2008-03-26 12:49:49 +0000642
643class BufferedIOBase(IOBase):
644
645 """Base class for buffered IO objects.
646
647 The main difference with RawIOBase is that the read() method
648 supports omitting the size argument, and does not have a default
649 implementation that defers to readinto().
650
651 In addition, read(), readinto() and write() may raise
652 BlockingIOError if the underlying raw stream is in non-blocking
653 mode and not ready; unlike their raw counterparts, they will never
654 return None.
655
656 A typical implementation should not inherit from a RawIOBase
657 implementation, but wrap one.
658 """
659
660 def read(self, n = None):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000661 """Read and return up to n bytes.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000662
663 If the argument is omitted, None, or negative, reads and
664 returns all data until EOF.
665
666 If the argument is positive, and the underlying raw stream is
667 not 'interactive', multiple raw reads may be issued to satisfy
668 the byte count (unless EOF is reached first). But for
669 interactive raw streams (XXX and for pipes?), at most one raw
670 read will be issued, and a short result does not imply that
671 EOF is imminent.
672
673 Returns an empty bytes array on EOF.
674
675 Raises BlockingIOError if the underlying raw stream has no
676 data at the moment.
677 """
678 self._unsupported("read")
679
680 def readinto(self, b):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000681 """Read up to len(b) bytes into b.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000682
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000683 Like read(), this may issue multiple reads to the underlying raw
684 stream, unless the latter is 'interactive'.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000685
686 Returns the number of bytes read (0 for EOF).
687
688 Raises BlockingIOError if the underlying raw stream has no
689 data at the moment.
690 """
691 # XXX This ought to work with anything that supports the buffer API
692 data = self.read(len(b))
693 n = len(data)
694 try:
695 b[:n] = data
696 except TypeError as err:
697 import array
698 if not isinstance(b, array.array):
699 raise err
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000700 b[:n] = array.array(b'b', data)
Christian Heimes1a6387e2008-03-26 12:49:49 +0000701 return n
702
703 def write(self, b):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000704 """Write the given buffer to the IO stream.
Christian Heimes1a6387e2008-03-26 12:49:49 +0000705
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000706 Return the number of bytes written, which is never less than
Christian Heimes1a6387e2008-03-26 12:49:49 +0000707 len(b).
708
709 Raises BlockingIOError if the buffer is full and the
710 underlying raw stream cannot accept more data at the moment.
711 """
712 self._unsupported("write")
713
714
715class _BufferedIOMixin(BufferedIOBase):
716
717 """A mixin implementation of BufferedIOBase with an underlying raw stream.
718
719 This passes most requests on to the underlying raw stream. It
720 does *not* provide implementations of read(), readinto() or
721 write().
722 """
723
724 def __init__(self, raw):
725 self.raw = raw
726
727 ### Positioning ###
728
729 def seek(self, pos, whence=0):
730 return self.raw.seek(pos, whence)
731
732 def tell(self):
733 return self.raw.tell()
734
735 def truncate(self, pos=None):
736 # Flush the stream. We're mixing buffered I/O with lower-level I/O,
737 # and a flush may be necessary to synch both views of the current
738 # file state.
739 self.flush()
740
741 if pos is None:
742 pos = self.tell()
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000743 # XXX: Should seek() be used, instead of passing the position
744 # XXX directly to truncate?
Christian Heimes1a6387e2008-03-26 12:49:49 +0000745 return self.raw.truncate(pos)
746
747 ### Flush and close ###
748
749 def flush(self):
750 self.raw.flush()
751
752 def close(self):
753 if not self.closed:
Antoine Pitrou01a255a2010-05-03 16:48:13 +0000754 self.flush()
Christian Heimes1a6387e2008-03-26 12:49:49 +0000755 self.raw.close()
756
757 ### Inquiries ###
758
759 def seekable(self):
760 return self.raw.seekable()
761
762 def readable(self):
763 return self.raw.readable()
764
765 def writable(self):
766 return self.raw.writable()
767
768 @property
769 def closed(self):
770 return self.raw.closed
771
Georg Brandld2094602008-12-05 08:51:30 +0000772 @property
773 def name(self):
774 return self.raw.name
775
776 @property
777 def mode(self):
778 return self.raw.mode
779
Christian Heimes1a6387e2008-03-26 12:49:49 +0000780 ### Lower-level APIs ###
781
782 def fileno(self):
783 return self.raw.fileno()
784
785 def isatty(self):
786 return self.raw.isatty()
787
788
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000789class _BytesIO(BufferedIOBase):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000790
791 """Buffered I/O implementation using an in-memory bytes buffer."""
792
793 # XXX More docs
794
795 def __init__(self, initial_bytes=None):
796 buf = bytearray()
797 if initial_bytes is not None:
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000798 buf += bytearray(initial_bytes)
Christian Heimes1a6387e2008-03-26 12:49:49 +0000799 self._buffer = buf
800 self._pos = 0
801
802 def getvalue(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000803 """Return the bytes value (contents) of the buffer
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000804 """
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000805 if self.closed:
806 raise ValueError("getvalue on closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000807 return bytes(self._buffer)
808
809 def read(self, n=None):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000810 if self.closed:
811 raise ValueError("read from closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000812 if n is None:
813 n = -1
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000814 if not isinstance(n, (int, long)):
815 raise TypeError("argument must be an integer")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000816 if n < 0:
817 n = len(self._buffer)
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000818 if len(self._buffer) <= self._pos:
819 return b""
Christian Heimes1a6387e2008-03-26 12:49:49 +0000820 newpos = min(len(self._buffer), self._pos + n)
821 b = self._buffer[self._pos : newpos]
822 self._pos = newpos
823 return bytes(b)
824
825 def read1(self, n):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +0000826 """this is the same as read.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000827 """
Christian Heimes1a6387e2008-03-26 12:49:49 +0000828 return self.read(n)
829
830 def write(self, b):
831 if self.closed:
832 raise ValueError("write to closed file")
833 if isinstance(b, unicode):
834 raise TypeError("can't write unicode to binary stream")
835 n = len(b)
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000836 if n == 0:
837 return 0
Alexandre Vassalotti844f7572008-05-10 19:59:16 +0000838 pos = self._pos
839 if pos > len(self._buffer):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000840 # Inserts null bytes between the current end of the file
841 # and the new write position.
Alexandre Vassalotti844f7572008-05-10 19:59:16 +0000842 padding = b'\x00' * (pos - len(self._buffer))
843 self._buffer += padding
844 self._buffer[pos:pos + n] = b
845 self._pos += n
Christian Heimes1a6387e2008-03-26 12:49:49 +0000846 return n
847
848 def seek(self, pos, whence=0):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000849 if self.closed:
850 raise ValueError("seek on closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000851 try:
852 pos = pos.__index__()
853 except AttributeError as err:
854 raise TypeError("an integer is required") # from err
855 if whence == 0:
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000856 if pos < 0:
857 raise ValueError("negative seek position %r" % (pos,))
858 self._pos = pos
Christian Heimes1a6387e2008-03-26 12:49:49 +0000859 elif whence == 1:
860 self._pos = max(0, self._pos + pos)
861 elif whence == 2:
862 self._pos = max(0, len(self._buffer) + pos)
863 else:
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000864 raise ValueError("invalid whence value")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000865 return self._pos
866
867 def tell(self):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000868 if self.closed:
869 raise ValueError("tell on closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000870 return self._pos
871
872 def truncate(self, pos=None):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000873 if self.closed:
874 raise ValueError("truncate on closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +0000875 if pos is None:
876 pos = self._pos
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000877 elif pos < 0:
878 raise ValueError("negative truncate position %r" % (pos,))
Christian Heimes1a6387e2008-03-26 12:49:49 +0000879 del self._buffer[pos:]
Antoine Pitrouca5a06a2010-01-27 21:48:46 +0000880 return pos
Christian Heimes1a6387e2008-03-26 12:49:49 +0000881
882 def readable(self):
883 return True
884
885 def writable(self):
886 return True
887
888 def seekable(self):
889 return True
890
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +0000891# Use the faster implementation of BytesIO if available
892try:
893 import _bytesio
894
895 class BytesIO(_bytesio._BytesIO, BufferedIOBase):
896 __doc__ = _bytesio._BytesIO.__doc__
897
898except ImportError:
899 BytesIO = _BytesIO
900
Christian Heimes1a6387e2008-03-26 12:49:49 +0000901
902class BufferedReader(_BufferedIOMixin):
903
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +0000904 """BufferedReader(raw[, buffer_size])
905
906 A buffer for a readable, sequential BaseRawIO object.
907
908 The constructor creates a BufferedReader for the given readable raw
909 stream and buffer_size. If buffer_size is omitted, DEFAULT_BUFFER_SIZE
910 is used.
911 """
Christian Heimes1a6387e2008-03-26 12:49:49 +0000912
913 def __init__(self, raw, buffer_size=DEFAULT_BUFFER_SIZE):
914 """Create a new buffered reader using the given readable raw IO object.
915 """
916 raw._checkReadable()
917 _BufferedIOMixin.__init__(self, raw)
Christian Heimes1a6387e2008-03-26 12:49:49 +0000918 self.buffer_size = buffer_size
Benjamin Peterson01a24322008-07-28 23:35:27 +0000919 self._reset_read_buf()
Antoine Pitrou11ec65d2008-08-14 21:04:30 +0000920 self._read_lock = threading.Lock()
Benjamin Peterson01a24322008-07-28 23:35:27 +0000921
922 def _reset_read_buf(self):
923 self._read_buf = b""
924 self._read_pos = 0
Christian Heimes1a6387e2008-03-26 12:49:49 +0000925
926 def read(self, n=None):
927 """Read n bytes.
928
929 Returns exactly n bytes of data unless the underlying raw IO
930 stream reaches EOF or if the call would block in non-blocking
931 mode. If n is negative, read until EOF or until read() would
932 block.
933 """
Antoine Pitrou11ec65d2008-08-14 21:04:30 +0000934 with self._read_lock:
935 return self._read_unlocked(n)
936
937 def _read_unlocked(self, n=None):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000938 nodata_val = b""
Benjamin Peterson01a24322008-07-28 23:35:27 +0000939 empty_values = (b"", None)
940 buf = self._read_buf
941 pos = self._read_pos
942
943 # Special case for when the number of bytes to read is unspecified.
944 if n is None or n == -1:
945 self._reset_read_buf()
946 chunks = [buf[pos:]] # Strip the consumed bytes.
947 current_size = 0
948 while True:
949 # Read until EOF or until read() would block.
950 chunk = self.raw.read()
951 if chunk in empty_values:
952 nodata_val = chunk
953 break
954 current_size += len(chunk)
955 chunks.append(chunk)
956 return b"".join(chunks) or nodata_val
957
958 # The number of bytes to read is specified, return at most n bytes.
959 avail = len(buf) - pos # Length of the available buffered data.
960 if n <= avail:
961 # Fast path: the data to read is fully buffered.
962 self._read_pos += n
963 return buf[pos:pos+n]
964 # Slow path: read from the stream until enough bytes are read,
965 # or until an EOF occurs or until read() would block.
966 chunks = [buf[pos:]]
967 wanted = max(self.buffer_size, n)
968 while avail < n:
969 chunk = self.raw.read(wanted)
970 if chunk in empty_values:
971 nodata_val = chunk
Christian Heimes1a6387e2008-03-26 12:49:49 +0000972 break
Benjamin Peterson01a24322008-07-28 23:35:27 +0000973 avail += len(chunk)
974 chunks.append(chunk)
975 # n is more then avail only when an EOF occurred or when
976 # read() would have blocked.
977 n = min(n, avail)
978 out = b"".join(chunks)
979 self._read_buf = out[n:] # Save the extra data in the buffer.
980 self._read_pos = 0
981 return out[:n] if out else nodata_val
Christian Heimes1a6387e2008-03-26 12:49:49 +0000982
983 def peek(self, n=0):
984 """Returns buffered bytes without advancing the position.
985
986 The argument indicates a desired minimal number of bytes; we
987 do at most one raw read to satisfy it. We never return more
988 than self.buffer_size.
989 """
Antoine Pitrou11ec65d2008-08-14 21:04:30 +0000990 with self._read_lock:
991 return self._peek_unlocked(n)
992
993 def _peek_unlocked(self, n=0):
Christian Heimes1a6387e2008-03-26 12:49:49 +0000994 want = min(n, self.buffer_size)
Benjamin Peterson01a24322008-07-28 23:35:27 +0000995 have = len(self._read_buf) - self._read_pos
Christian Heimes1a6387e2008-03-26 12:49:49 +0000996 if have < want:
997 to_read = self.buffer_size - have
998 current = self.raw.read(to_read)
999 if current:
Benjamin Peterson01a24322008-07-28 23:35:27 +00001000 self._read_buf = self._read_buf[self._read_pos:] + current
1001 self._read_pos = 0
1002 return self._read_buf[self._read_pos:]
Christian Heimes1a6387e2008-03-26 12:49:49 +00001003
1004 def read1(self, n):
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001005 """Reads up to n bytes, with at most one read() system call."""
1006 # Returns up to n bytes. If at least one byte is buffered, we
1007 # only return buffered bytes. Otherwise, we do one raw read.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001008 if n <= 0:
1009 return b""
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001010 with self._read_lock:
1011 self._peek_unlocked(1)
1012 return self._read_unlocked(
1013 min(n, len(self._read_buf) - self._read_pos))
Christian Heimes1a6387e2008-03-26 12:49:49 +00001014
1015 def tell(self):
Benjamin Peterson01a24322008-07-28 23:35:27 +00001016 return self.raw.tell() - len(self._read_buf) + self._read_pos
Christian Heimes1a6387e2008-03-26 12:49:49 +00001017
1018 def seek(self, pos, whence=0):
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001019 with self._read_lock:
1020 if whence == 1:
1021 pos -= len(self._read_buf) - self._read_pos
1022 pos = self.raw.seek(pos, whence)
1023 self._reset_read_buf()
1024 return pos
Christian Heimes1a6387e2008-03-26 12:49:49 +00001025
1026
1027class BufferedWriter(_BufferedIOMixin):
1028
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001029 """A buffer for a writeable sequential RawIO object.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001030
1031 The constructor creates a BufferedWriter for the given writeable raw
1032 stream. If the buffer_size is not given, it defaults to
1033 DEAFULT_BUFFER_SIZE. If max_buffer_size is omitted, it defaults to
1034 twice the buffer size.
1035 """
Christian Heimes1a6387e2008-03-26 12:49:49 +00001036
1037 def __init__(self, raw,
1038 buffer_size=DEFAULT_BUFFER_SIZE, max_buffer_size=None):
1039 raw._checkWritable()
1040 _BufferedIOMixin.__init__(self, raw)
1041 self.buffer_size = buffer_size
1042 self.max_buffer_size = (2*buffer_size
1043 if max_buffer_size is None
1044 else max_buffer_size)
1045 self._write_buf = bytearray()
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001046 self._write_lock = threading.Lock()
Christian Heimes1a6387e2008-03-26 12:49:49 +00001047
1048 def write(self, b):
1049 if self.closed:
1050 raise ValueError("write to closed file")
1051 if isinstance(b, unicode):
1052 raise TypeError("can't write unicode to binary stream")
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001053 with self._write_lock:
1054 # XXX we can implement some more tricks to try and avoid
1055 # partial writes
1056 if len(self._write_buf) > self.buffer_size:
1057 # We're full, so let's pre-flush the buffer
1058 try:
1059 self._flush_unlocked()
1060 except BlockingIOError as e:
1061 # We can't accept anything else.
1062 # XXX Why not just let the exception pass through?
1063 raise BlockingIOError(e.errno, e.strerror, 0)
1064 before = len(self._write_buf)
1065 self._write_buf.extend(b)
1066 written = len(self._write_buf) - before
1067 if len(self._write_buf) > self.buffer_size:
1068 try:
1069 self._flush_unlocked()
1070 except BlockingIOError as e:
1071 if len(self._write_buf) > self.max_buffer_size:
1072 # We've hit max_buffer_size. We have to accept a
1073 # partial write and cut back our buffer.
1074 overage = len(self._write_buf) - self.max_buffer_size
1075 self._write_buf = self._write_buf[:self.max_buffer_size]
1076 raise BlockingIOError(e.errno, e.strerror, overage)
1077 return written
Christian Heimes1a6387e2008-03-26 12:49:49 +00001078
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001079 def truncate(self, pos=None):
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001080 with self._write_lock:
1081 self._flush_unlocked()
1082 if pos is None:
1083 pos = self.raw.tell()
1084 return self.raw.truncate(pos)
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001085
Christian Heimes1a6387e2008-03-26 12:49:49 +00001086 def flush(self):
Antoine Pitrou01a255a2010-05-03 16:48:13 +00001087 if self.closed:
1088 raise ValueError("flush of closed file")
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001089 with self._write_lock:
1090 self._flush_unlocked()
1091
1092 def _flush_unlocked(self):
Christian Heimes1a6387e2008-03-26 12:49:49 +00001093 if self.closed:
1094 raise ValueError("flush of closed file")
1095 written = 0
1096 try:
1097 while self._write_buf:
1098 n = self.raw.write(self._write_buf)
1099 del self._write_buf[:n]
1100 written += n
1101 except BlockingIOError as e:
1102 n = e.characters_written
1103 del self._write_buf[:n]
1104 written += n
1105 raise BlockingIOError(e.errno, e.strerror, written)
1106
1107 def tell(self):
1108 return self.raw.tell() + len(self._write_buf)
1109
1110 def seek(self, pos, whence=0):
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001111 with self._write_lock:
1112 self._flush_unlocked()
1113 return self.raw.seek(pos, whence)
Christian Heimes1a6387e2008-03-26 12:49:49 +00001114
1115
1116class BufferedRWPair(BufferedIOBase):
1117
1118 """A buffered reader and writer object together.
1119
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001120 A buffered reader object and buffered writer object put together to
1121 form a sequential IO object that can read and write. This is typically
1122 used with a socket or two-way pipe.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001123
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001124 reader and writer are RawIOBase objects that are readable and
1125 writeable respectively. If the buffer_size is omitted it defaults to
1126 DEFAULT_BUFFER_SIZE. The max_buffer_size (for the buffered writer)
1127 defaults to twice the buffer size.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001128 """
1129
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001130 # XXX The usefulness of this (compared to having two separate IO
1131 # objects) is questionable.
1132
Christian Heimes1a6387e2008-03-26 12:49:49 +00001133 def __init__(self, reader, writer,
1134 buffer_size=DEFAULT_BUFFER_SIZE, max_buffer_size=None):
1135 """Constructor.
1136
1137 The arguments are two RawIO instances.
1138 """
1139 reader._checkReadable()
1140 writer._checkWritable()
1141 self.reader = BufferedReader(reader, buffer_size)
1142 self.writer = BufferedWriter(writer, buffer_size, max_buffer_size)
1143
1144 def read(self, n=None):
1145 if n is None:
1146 n = -1
1147 return self.reader.read(n)
1148
1149 def readinto(self, b):
1150 return self.reader.readinto(b)
1151
1152 def write(self, b):
1153 return self.writer.write(b)
1154
1155 def peek(self, n=0):
1156 return self.reader.peek(n)
1157
1158 def read1(self, n):
1159 return self.reader.read1(n)
1160
1161 def readable(self):
1162 return self.reader.readable()
1163
1164 def writable(self):
1165 return self.writer.writable()
1166
1167 def flush(self):
1168 return self.writer.flush()
1169
1170 def close(self):
1171 self.writer.close()
1172 self.reader.close()
1173
1174 def isatty(self):
1175 return self.reader.isatty() or self.writer.isatty()
1176
1177 @property
1178 def closed(self):
Benjamin Peterson828a7062008-12-27 17:05:29 +00001179 return self.writer.closed
Christian Heimes1a6387e2008-03-26 12:49:49 +00001180
1181
1182class BufferedRandom(BufferedWriter, BufferedReader):
1183
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001184 """A buffered interface to random access streams.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001185
1186 The constructor creates a reader and writer for a seekable stream,
1187 raw, given in the first argument. If the buffer_size is omitted it
1188 defaults to DEFAULT_BUFFER_SIZE. The max_buffer_size (for the buffered
1189 writer) defaults to twice the buffer size.
1190 """
Christian Heimes1a6387e2008-03-26 12:49:49 +00001191
1192 def __init__(self, raw,
1193 buffer_size=DEFAULT_BUFFER_SIZE, max_buffer_size=None):
1194 raw._checkSeekable()
1195 BufferedReader.__init__(self, raw, buffer_size)
1196 BufferedWriter.__init__(self, raw, buffer_size, max_buffer_size)
1197
1198 def seek(self, pos, whence=0):
1199 self.flush()
1200 # First do the raw seek, then empty the read buffer, so that
1201 # if the raw seek fails, we don't lose buffered data forever.
1202 pos = self.raw.seek(pos, whence)
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001203 with self._read_lock:
1204 self._reset_read_buf()
Christian Heimes1a6387e2008-03-26 12:49:49 +00001205 return pos
1206
1207 def tell(self):
Benjamin Peterson01a24322008-07-28 23:35:27 +00001208 if self._write_buf:
Christian Heimes1a6387e2008-03-26 12:49:49 +00001209 return self.raw.tell() + len(self._write_buf)
1210 else:
Benjamin Peterson01a24322008-07-28 23:35:27 +00001211 return BufferedReader.tell(self)
Christian Heimes1a6387e2008-03-26 12:49:49 +00001212
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001213 def truncate(self, pos=None):
1214 if pos is None:
1215 pos = self.tell()
1216 # Use seek to flush the read buffer.
Antoine Pitrouca5a06a2010-01-27 21:48:46 +00001217 return BufferedWriter.truncate(self, pos)
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001218
Christian Heimes1a6387e2008-03-26 12:49:49 +00001219 def read(self, n=None):
1220 if n is None:
1221 n = -1
1222 self.flush()
1223 return BufferedReader.read(self, n)
1224
1225 def readinto(self, b):
1226 self.flush()
1227 return BufferedReader.readinto(self, b)
1228
1229 def peek(self, n=0):
1230 self.flush()
1231 return BufferedReader.peek(self, n)
1232
1233 def read1(self, n):
1234 self.flush()
1235 return BufferedReader.read1(self, n)
1236
1237 def write(self, b):
1238 if self._read_buf:
Benjamin Peterson01a24322008-07-28 23:35:27 +00001239 # Undo readahead
Antoine Pitrou11ec65d2008-08-14 21:04:30 +00001240 with self._read_lock:
1241 self.raw.seek(self._read_pos - len(self._read_buf), 1)
1242 self._reset_read_buf()
Christian Heimes1a6387e2008-03-26 12:49:49 +00001243 return BufferedWriter.write(self, b)
1244
1245
1246class TextIOBase(IOBase):
1247
1248 """Base class for text I/O.
1249
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001250 This class provides a character and line based interface to stream
1251 I/O. There is no readinto method because Python's character strings
1252 are immutable. There is no public constructor.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001253 """
1254
1255 def read(self, n = -1):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001256 """Read at most n characters from stream.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001257
1258 Read from underlying buffer until we have n characters or we hit EOF.
1259 If n is negative or omitted, read until EOF.
1260 """
1261 self._unsupported("read")
1262
1263 def write(self, s):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001264 """Write string s to stream."""
Christian Heimes1a6387e2008-03-26 12:49:49 +00001265 self._unsupported("write")
1266
1267 def truncate(self, pos = None):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001268 """Truncate size to pos."""
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001269 self._unsupported("truncate")
Christian Heimes1a6387e2008-03-26 12:49:49 +00001270
1271 def readline(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001272 """Read until newline or EOF.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001273
1274 Returns an empty string if EOF is hit immediately.
1275 """
1276 self._unsupported("readline")
1277
1278 @property
1279 def encoding(self):
1280 """Subclasses should override."""
1281 return None
1282
1283 @property
1284 def newlines(self):
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001285 """Line endings translated so far.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001286
1287 Only line endings translated during reading are considered.
1288
1289 Subclasses should override.
1290 """
1291 return None
1292
1293
1294class IncrementalNewlineDecoder(codecs.IncrementalDecoder):
1295 """Codec used when reading a file in universal newlines mode.
1296 It wraps another incremental decoder, translating \\r\\n and \\r into \\n.
1297 It also records the types of newlines encountered.
1298 When used with translate=False, it ensures that the newline sequence is
1299 returned in one piece.
1300 """
1301 def __init__(self, decoder, translate, errors='strict'):
1302 codecs.IncrementalDecoder.__init__(self, errors=errors)
Christian Heimes1a6387e2008-03-26 12:49:49 +00001303 self.translate = translate
1304 self.decoder = decoder
1305 self.seennl = 0
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001306 self.pendingcr = False
Christian Heimes1a6387e2008-03-26 12:49:49 +00001307
1308 def decode(self, input, final=False):
1309 # decode input (with the eventual \r from a previous pass)
Christian Heimes1a6387e2008-03-26 12:49:49 +00001310 output = self.decoder.decode(input, final=final)
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001311 if self.pendingcr and (output or final):
1312 output = "\r" + output
1313 self.pendingcr = False
Christian Heimes1a6387e2008-03-26 12:49:49 +00001314
1315 # retain last \r even when not translating data:
1316 # then readline() is sure to get \r\n in one pass
1317 if output.endswith("\r") and not final:
1318 output = output[:-1]
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001319 self.pendingcr = True
Christian Heimes1a6387e2008-03-26 12:49:49 +00001320
1321 # Record which newlines are read
1322 crlf = output.count('\r\n')
1323 cr = output.count('\r') - crlf
1324 lf = output.count('\n') - crlf
1325 self.seennl |= (lf and self._LF) | (cr and self._CR) \
1326 | (crlf and self._CRLF)
1327
1328 if self.translate:
1329 if crlf:
1330 output = output.replace("\r\n", "\n")
1331 if cr:
1332 output = output.replace("\r", "\n")
1333
1334 return output
1335
1336 def getstate(self):
1337 buf, flag = self.decoder.getstate()
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001338 flag <<= 1
1339 if self.pendingcr:
1340 flag |= 1
1341 return buf, flag
Christian Heimes1a6387e2008-03-26 12:49:49 +00001342
1343 def setstate(self, state):
1344 buf, flag = state
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001345 self.pendingcr = bool(flag & 1)
1346 self.decoder.setstate((buf, flag >> 1))
Christian Heimes1a6387e2008-03-26 12:49:49 +00001347
1348 def reset(self):
1349 self.seennl = 0
Antoine Pitrouf8638a82008-12-14 18:08:37 +00001350 self.pendingcr = False
Christian Heimes1a6387e2008-03-26 12:49:49 +00001351 self.decoder.reset()
1352
1353 _LF = 1
1354 _CR = 2
1355 _CRLF = 4
1356
1357 @property
1358 def newlines(self):
1359 return (None,
1360 "\n",
1361 "\r",
1362 ("\r", "\n"),
1363 "\r\n",
1364 ("\n", "\r\n"),
1365 ("\r", "\r\n"),
1366 ("\r", "\n", "\r\n")
1367 )[self.seennl]
1368
1369
1370class TextIOWrapper(TextIOBase):
1371
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001372 r"""Character and line based layer over a BufferedIOBase object, buffer.
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001373
1374 encoding gives the name of the encoding that the stream will be
1375 decoded or encoded with. It defaults to locale.getpreferredencoding.
1376
1377 errors determines the strictness of encoding and decoding (see the
1378 codecs.register) and defaults to "strict".
1379
1380 newline can be None, '', '\n', '\r', or '\r\n'. It controls the
1381 handling of line endings. If it is None, universal newlines is
1382 enabled. With this enabled, on input, the lines endings '\n', '\r',
1383 or '\r\n' are translated to '\n' before being returned to the
1384 caller. Conversely, on output, '\n' is translated to the system
Jesus Cea585ad8a2009-07-02 15:37:21 +00001385 default line separator, os.linesep. If newline is any other of its
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001386 legal values, that newline becomes the newline when the file is read
1387 and it is returned untranslated. On output, '\n' is converted to the
1388 newline.
1389
1390 If line_buffering is True, a call to flush is implied when a call to
1391 write contains a newline character.
Christian Heimes1a6387e2008-03-26 12:49:49 +00001392 """
1393
1394 _CHUNK_SIZE = 128
1395
1396 def __init__(self, buffer, encoding=None, errors=None, newline=None,
1397 line_buffering=False):
1398 if newline not in (None, "", "\n", "\r", "\r\n"):
1399 raise ValueError("illegal newline value: %r" % (newline,))
1400 if encoding is None:
1401 try:
1402 encoding = os.device_encoding(buffer.fileno())
1403 except (AttributeError, UnsupportedOperation):
1404 pass
1405 if encoding is None:
1406 try:
1407 import locale
1408 except ImportError:
1409 # Importing locale may fail if Python is being built
1410 encoding = "ascii"
1411 else:
1412 encoding = locale.getpreferredencoding()
1413
Christian Heimes3784c6b2008-03-26 23:13:59 +00001414 if not isinstance(encoding, basestring):
Christian Heimes1a6387e2008-03-26 12:49:49 +00001415 raise ValueError("invalid encoding: %r" % encoding)
1416
1417 if errors is None:
1418 errors = "strict"
1419 else:
Christian Heimes3784c6b2008-03-26 23:13:59 +00001420 if not isinstance(errors, basestring):
Christian Heimes1a6387e2008-03-26 12:49:49 +00001421 raise ValueError("invalid errors: %r" % errors)
1422
1423 self.buffer = buffer
1424 self._line_buffering = line_buffering
1425 self._encoding = encoding
1426 self._errors = errors
1427 self._readuniversal = not newline
1428 self._readtranslate = newline is None
1429 self._readnl = newline
1430 self._writetranslate = newline != ''
1431 self._writenl = newline or os.linesep
1432 self._encoder = None
1433 self._decoder = None
1434 self._decoded_chars = '' # buffer for text returned from decoder
1435 self._decoded_chars_used = 0 # offset into _decoded_chars for read()
1436 self._snapshot = None # info for reconstructing decoder state
1437 self._seekable = self._telling = self.buffer.seekable()
1438
1439 # self._snapshot is either None, or a tuple (dec_flags, next_input)
1440 # where dec_flags is the second (integer) item of the decoder state
1441 # and next_input is the chunk of input bytes that comes next after the
1442 # snapshot point. We use this to reconstruct decoder states in tell().
1443
1444 # Naming convention:
1445 # - "bytes_..." for integer variables that count input bytes
1446 # - "chars_..." for integer variables that count decoded characters
1447
Christian Heimes1a6387e2008-03-26 12:49:49 +00001448 @property
1449 def encoding(self):
1450 return self._encoding
1451
1452 @property
1453 def errors(self):
1454 return self._errors
1455
1456 @property
1457 def line_buffering(self):
1458 return self._line_buffering
1459
1460 def seekable(self):
1461 return self._seekable
1462
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001463 def readable(self):
1464 return self.buffer.readable()
1465
1466 def writable(self):
1467 return self.buffer.writable()
1468
Christian Heimes1a6387e2008-03-26 12:49:49 +00001469 def flush(self):
1470 self.buffer.flush()
1471 self._telling = self._seekable
1472
1473 def close(self):
Antoine Pitrou01a255a2010-05-03 16:48:13 +00001474 if not self.closed:
Christian Heimes1a6387e2008-03-26 12:49:49 +00001475 self.flush()
Antoine Pitrou01a255a2010-05-03 16:48:13 +00001476 self.buffer.close()
Christian Heimes1a6387e2008-03-26 12:49:49 +00001477
1478 @property
1479 def closed(self):
1480 return self.buffer.closed
1481
Georg Brandld2094602008-12-05 08:51:30 +00001482 @property
1483 def name(self):
1484 return self.buffer.name
1485
Christian Heimes1a6387e2008-03-26 12:49:49 +00001486 def fileno(self):
1487 return self.buffer.fileno()
1488
1489 def isatty(self):
1490 return self.buffer.isatty()
1491
1492 def write(self, s):
1493 if self.closed:
1494 raise ValueError("write to closed file")
1495 if not isinstance(s, unicode):
1496 raise TypeError("can't write %s to text stream" %
1497 s.__class__.__name__)
1498 length = len(s)
1499 haslf = (self._writetranslate or self._line_buffering) and "\n" in s
1500 if haslf and self._writetranslate and self._writenl != "\n":
1501 s = s.replace("\n", self._writenl)
1502 encoder = self._encoder or self._get_encoder()
1503 # XXX What if we were just reading?
1504 b = encoder.encode(s)
1505 self.buffer.write(b)
1506 if self._line_buffering and (haslf or "\r" in s):
1507 self.flush()
1508 self._snapshot = None
1509 if self._decoder:
1510 self._decoder.reset()
1511 return length
1512
1513 def _get_encoder(self):
1514 make_encoder = codecs.getincrementalencoder(self._encoding)
1515 self._encoder = make_encoder(self._errors)
1516 return self._encoder
1517
1518 def _get_decoder(self):
1519 make_decoder = codecs.getincrementaldecoder(self._encoding)
1520 decoder = make_decoder(self._errors)
1521 if self._readuniversal:
1522 decoder = IncrementalNewlineDecoder(decoder, self._readtranslate)
1523 self._decoder = decoder
1524 return decoder
1525
1526 # The following three methods implement an ADT for _decoded_chars.
1527 # Text returned from the decoder is buffered here until the client
1528 # requests it by calling our read() or readline() method.
1529 def _set_decoded_chars(self, chars):
1530 """Set the _decoded_chars buffer."""
1531 self._decoded_chars = chars
1532 self._decoded_chars_used = 0
1533
1534 def _get_decoded_chars(self, n=None):
1535 """Advance into the _decoded_chars buffer."""
1536 offset = self._decoded_chars_used
1537 if n is None:
1538 chars = self._decoded_chars[offset:]
1539 else:
1540 chars = self._decoded_chars[offset:offset + n]
1541 self._decoded_chars_used += len(chars)
1542 return chars
1543
1544 def _rewind_decoded_chars(self, n):
1545 """Rewind the _decoded_chars buffer."""
1546 if self._decoded_chars_used < n:
1547 raise AssertionError("rewind decoded_chars out of bounds")
1548 self._decoded_chars_used -= n
1549
1550 def _read_chunk(self):
1551 """
1552 Read and decode the next chunk of data from the BufferedReader.
1553
1554 The return value is True unless EOF was reached. The decoded string
1555 is placed in self._decoded_chars (replacing its previous value).
1556 The entire input chunk is sent to the decoder, though some of it
1557 may remain buffered in the decoder, yet to be converted.
1558 """
1559
1560 if self._decoder is None:
1561 raise ValueError("no decoder")
1562
1563 if self._telling:
1564 # To prepare for tell(), we need to snapshot a point in the
1565 # file where the decoder's input buffer is empty.
1566
1567 dec_buffer, dec_flags = self._decoder.getstate()
1568 # Given this, we know there was a valid snapshot point
1569 # len(dec_buffer) bytes ago with decoder state (b'', dec_flags).
1570
1571 # Read a chunk, decode it, and put the result in self._decoded_chars.
1572 input_chunk = self.buffer.read1(self._CHUNK_SIZE)
1573 eof = not input_chunk
1574 self._set_decoded_chars(self._decoder.decode(input_chunk, eof))
1575
1576 if self._telling:
1577 # At the snapshot point, len(dec_buffer) bytes before the read,
1578 # the next input to be decoded is dec_buffer + input_chunk.
1579 self._snapshot = (dec_flags, dec_buffer + input_chunk)
1580
1581 return not eof
1582
1583 def _pack_cookie(self, position, dec_flags=0,
1584 bytes_to_feed=0, need_eof=0, chars_to_skip=0):
1585 # The meaning of a tell() cookie is: seek to position, set the
1586 # decoder flags to dec_flags, read bytes_to_feed bytes, feed them
1587 # into the decoder with need_eof as the EOF flag, then skip
1588 # chars_to_skip characters of the decoded result. For most simple
1589 # decoders, tell() will often just give a byte offset in the file.
1590 return (position | (dec_flags<<64) | (bytes_to_feed<<128) |
1591 (chars_to_skip<<192) | bool(need_eof)<<256)
1592
1593 def _unpack_cookie(self, bigint):
1594 rest, position = divmod(bigint, 1<<64)
1595 rest, dec_flags = divmod(rest, 1<<64)
1596 rest, bytes_to_feed = divmod(rest, 1<<64)
1597 need_eof, chars_to_skip = divmod(rest, 1<<64)
1598 return position, dec_flags, bytes_to_feed, need_eof, chars_to_skip
1599
1600 def tell(self):
1601 if not self._seekable:
1602 raise IOError("underlying stream is not seekable")
1603 if not self._telling:
1604 raise IOError("telling position disabled by next() call")
1605 self.flush()
1606 position = self.buffer.tell()
1607 decoder = self._decoder
1608 if decoder is None or self._snapshot is None:
1609 if self._decoded_chars:
1610 # This should never happen.
1611 raise AssertionError("pending decoded text")
1612 return position
1613
1614 # Skip backward to the snapshot point (see _read_chunk).
1615 dec_flags, next_input = self._snapshot
1616 position -= len(next_input)
1617
1618 # How many decoded characters have been used up since the snapshot?
1619 chars_to_skip = self._decoded_chars_used
1620 if chars_to_skip == 0:
1621 # We haven't moved from the snapshot point.
1622 return self._pack_cookie(position, dec_flags)
1623
1624 # Starting from the snapshot position, we will walk the decoder
1625 # forward until it gives us enough decoded characters.
1626 saved_state = decoder.getstate()
1627 try:
1628 # Note our initial start point.
1629 decoder.setstate((b'', dec_flags))
1630 start_pos = position
1631 start_flags, bytes_fed, chars_decoded = dec_flags, 0, 0
1632 need_eof = 0
1633
1634 # Feed the decoder one byte at a time. As we go, note the
1635 # nearest "safe start point" before the current location
1636 # (a point where the decoder has nothing buffered, so seek()
1637 # can safely start from there and advance to this location).
Amaury Forgeot d'Arc7684f852008-05-03 12:21:13 +00001638 for next_byte in next_input:
Christian Heimes1a6387e2008-03-26 12:49:49 +00001639 bytes_fed += 1
1640 chars_decoded += len(decoder.decode(next_byte))
1641 dec_buffer, dec_flags = decoder.getstate()
1642 if not dec_buffer and chars_decoded <= chars_to_skip:
1643 # Decoder buffer is empty, so this is a safe start point.
1644 start_pos += bytes_fed
1645 chars_to_skip -= chars_decoded
1646 start_flags, bytes_fed, chars_decoded = dec_flags, 0, 0
1647 if chars_decoded >= chars_to_skip:
1648 break
1649 else:
1650 # We didn't get enough decoded data; signal EOF to get more.
1651 chars_decoded += len(decoder.decode(b'', final=True))
1652 need_eof = 1
1653 if chars_decoded < chars_to_skip:
1654 raise IOError("can't reconstruct logical file position")
1655
1656 # The returned cookie corresponds to the last safe start point.
1657 return self._pack_cookie(
1658 start_pos, start_flags, bytes_fed, need_eof, chars_to_skip)
1659 finally:
1660 decoder.setstate(saved_state)
1661
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001662 def truncate(self, pos=None):
1663 self.flush()
1664 if pos is None:
1665 pos = self.tell()
Antoine Pitrouca5a06a2010-01-27 21:48:46 +00001666 return self.buffer.truncate(pos)
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001667
Christian Heimes1a6387e2008-03-26 12:49:49 +00001668 def seek(self, cookie, whence=0):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001669 if self.closed:
1670 raise ValueError("tell on closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +00001671 if not self._seekable:
1672 raise IOError("underlying stream is not seekable")
1673 if whence == 1: # seek relative to current position
1674 if cookie != 0:
1675 raise IOError("can't do nonzero cur-relative seeks")
1676 # Seeking to the current position should attempt to
1677 # sync the underlying buffer with the current position.
1678 whence = 0
1679 cookie = self.tell()
1680 if whence == 2: # seek relative to end of file
1681 if cookie != 0:
1682 raise IOError("can't do nonzero end-relative seeks")
1683 self.flush()
1684 position = self.buffer.seek(0, 2)
1685 self._set_decoded_chars('')
1686 self._snapshot = None
1687 if self._decoder:
1688 self._decoder.reset()
1689 return position
1690 if whence != 0:
1691 raise ValueError("invalid whence (%r, should be 0, 1 or 2)" %
1692 (whence,))
1693 if cookie < 0:
1694 raise ValueError("negative seek position %r" % (cookie,))
1695 self.flush()
1696
1697 # The strategy of seek() is to go back to the safe start point
1698 # and replay the effect of read(chars_to_skip) from there.
1699 start_pos, dec_flags, bytes_to_feed, need_eof, chars_to_skip = \
1700 self._unpack_cookie(cookie)
1701
1702 # Seek back to the safe start point.
1703 self.buffer.seek(start_pos)
1704 self._set_decoded_chars('')
1705 self._snapshot = None
1706
1707 # Restore the decoder to its state from the safe start point.
1708 if self._decoder or dec_flags or chars_to_skip:
1709 self._decoder = self._decoder or self._get_decoder()
1710 self._decoder.setstate((b'', dec_flags))
1711 self._snapshot = (dec_flags, b'')
1712
1713 if chars_to_skip:
1714 # Just like _read_chunk, feed the decoder and save a snapshot.
1715 input_chunk = self.buffer.read(bytes_to_feed)
1716 self._set_decoded_chars(
1717 self._decoder.decode(input_chunk, need_eof))
1718 self._snapshot = (dec_flags, input_chunk)
1719
1720 # Skip chars_to_skip of the decoded characters.
1721 if len(self._decoded_chars) < chars_to_skip:
1722 raise IOError("can't restore logical file position")
1723 self._decoded_chars_used = chars_to_skip
1724
1725 return cookie
1726
1727 def read(self, n=None):
1728 if n is None:
1729 n = -1
1730 decoder = self._decoder or self._get_decoder()
1731 if n < 0:
1732 # Read everything.
1733 result = (self._get_decoded_chars() +
1734 decoder.decode(self.buffer.read(), final=True))
1735 self._set_decoded_chars('')
1736 self._snapshot = None
1737 return result
1738 else:
1739 # Keep reading chunks until we have n characters to return.
1740 eof = False
1741 result = self._get_decoded_chars(n)
1742 while len(result) < n and not eof:
1743 eof = not self._read_chunk()
1744 result += self._get_decoded_chars(n - len(result))
1745 return result
1746
1747 def next(self):
1748 self._telling = False
1749 line = self.readline()
1750 if not line:
1751 self._snapshot = None
1752 self._telling = self._seekable
1753 raise StopIteration
1754 return line
1755
1756 def readline(self, limit=None):
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001757 if self.closed:
1758 raise ValueError("read from closed file")
Christian Heimes1a6387e2008-03-26 12:49:49 +00001759 if limit is None:
1760 limit = -1
Alexandre Vassalotti1aed6242008-05-09 21:49:43 +00001761 if not isinstance(limit, (int, long)):
1762 raise TypeError("limit must be an integer")
Christian Heimes1a6387e2008-03-26 12:49:49 +00001763
1764 # Grab all the decoded text (we will rewind any extra bits later).
1765 line = self._get_decoded_chars()
1766
1767 start = 0
1768 decoder = self._decoder or self._get_decoder()
1769
1770 pos = endpos = None
1771 while True:
1772 if self._readtranslate:
1773 # Newlines are already translated, only search for \n
1774 pos = line.find('\n', start)
1775 if pos >= 0:
1776 endpos = pos + 1
1777 break
1778 else:
1779 start = len(line)
1780
1781 elif self._readuniversal:
1782 # Universal newline search. Find any of \r, \r\n, \n
1783 # The decoder ensures that \r\n are not split in two pieces
1784
1785 # In C we'd look for these in parallel of course.
1786 nlpos = line.find("\n", start)
1787 crpos = line.find("\r", start)
1788 if crpos == -1:
1789 if nlpos == -1:
1790 # Nothing found
1791 start = len(line)
1792 else:
1793 # Found \n
1794 endpos = nlpos + 1
1795 break
1796 elif nlpos == -1:
1797 # Found lone \r
1798 endpos = crpos + 1
1799 break
1800 elif nlpos < crpos:
1801 # Found \n
1802 endpos = nlpos + 1
1803 break
1804 elif nlpos == crpos + 1:
1805 # Found \r\n
1806 endpos = crpos + 2
1807 break
1808 else:
1809 # Found \r
1810 endpos = crpos + 1
1811 break
1812 else:
1813 # non-universal
1814 pos = line.find(self._readnl)
1815 if pos >= 0:
1816 endpos = pos + len(self._readnl)
1817 break
1818
1819 if limit >= 0 and len(line) >= limit:
1820 endpos = limit # reached length limit
1821 break
1822
1823 # No line ending seen yet - get more data
1824 more_line = ''
1825 while self._read_chunk():
1826 if self._decoded_chars:
1827 break
1828 if self._decoded_chars:
1829 line += self._get_decoded_chars()
1830 else:
1831 # end of file
1832 self._set_decoded_chars('')
1833 self._snapshot = None
1834 return line
1835
1836 if limit >= 0 and endpos > limit:
1837 endpos = limit # don't exceed limit
1838
1839 # Rewind _decoded_chars to just after the line ending we found.
1840 self._rewind_decoded_chars(len(line) - endpos)
1841 return line[:endpos]
1842
1843 @property
1844 def newlines(self):
1845 return self._decoder.newlines if self._decoder else None
1846
1847class StringIO(TextIOWrapper):
1848
Benjamin Peterson9ae080e2008-05-04 22:39:33 +00001849 """An in-memory stream for text. The initial_value argument sets the
Benjamin Peterson7bb4d2d2008-04-13 02:01:27 +00001850 value of object. The other arguments are like those of TextIOWrapper's
1851 constructor.
1852 """
Christian Heimes1a6387e2008-03-26 12:49:49 +00001853
1854 def __init__(self, initial_value="", encoding="utf-8",
1855 errors="strict", newline="\n"):
1856 super(StringIO, self).__init__(BytesIO(),
1857 encoding=encoding,
1858 errors=errors,
1859 newline=newline)
Alexandre Vassalottidd0b90a2009-06-12 21:20:23 +00001860 # Issue #5645: make universal newlines semantics the same as in the
1861 # C version, even under Windows.
1862 if newline is None:
1863 self._writetranslate = False
Christian Heimes1a6387e2008-03-26 12:49:49 +00001864 if initial_value:
1865 if not isinstance(initial_value, unicode):
1866 initial_value = unicode(initial_value)
1867 self.write(initial_value)
1868 self.seek(0)
1869
1870 def getvalue(self):
1871 self.flush()
1872 return self.buffer.getvalue().decode(self._encoding, self._errors)