Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 1 | |
| 2 | :mod:`bz2` --- Compression compatible with :program:`bzip2` |
| 3 | =========================================================== |
| 4 | |
| 5 | .. module:: bz2 |
| 6 | :synopsis: Interface to compression and decompression routines compatible with bzip2. |
| 7 | .. moduleauthor:: Gustavo Niemeyer <niemeyer@conectiva.com> |
| 8 | .. sectionauthor:: Gustavo Niemeyer <niemeyer@conectiva.com> |
| 9 | |
| 10 | |
| 11 | .. versionadded:: 2.3 |
| 12 | |
| 13 | This module provides a comprehensive interface for the bz2 compression library. |
| 14 | It implements a complete file interface, one-shot (de)compression functions, and |
| 15 | types for sequential (de)compression. |
| 16 | |
Mark Summerfield | aea6e59 | 2007-11-05 09:22:48 +0000 | [diff] [blame] | 17 | Here is a summary of the features offered by the bz2 module: |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 18 | |
| 19 | * :class:`BZ2File` class implements a complete file interface, including |
Ezio Melotti | c4785a7 | 2010-03-13 00:15:36 +0000 | [diff] [blame] | 20 | :meth:`~BZ2File.readline`, :meth:`~BZ2File.readlines`, |
| 21 | :meth:`~BZ2File.writelines`, :meth:`~BZ2File.seek`, etc; |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 22 | |
Ezio Melotti | c4785a7 | 2010-03-13 00:15:36 +0000 | [diff] [blame] | 23 | * :class:`BZ2File` class implements emulated :meth:`~BZ2File.seek` support; |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 24 | |
| 25 | * :class:`BZ2File` class implements universal newline support; |
| 26 | |
| 27 | * :class:`BZ2File` class offers an optimized line iteration using the readahead |
| 28 | algorithm borrowed from file objects; |
| 29 | |
| 30 | * Sequential (de)compression supported by :class:`BZ2Compressor` and |
| 31 | :class:`BZ2Decompressor` classes; |
| 32 | |
| 33 | * One-shot (de)compression supported by :func:`compress` and :func:`decompress` |
| 34 | functions; |
| 35 | |
Mark Summerfield | aea6e59 | 2007-11-05 09:22:48 +0000 | [diff] [blame] | 36 | * Thread safety uses individual locking mechanism. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 37 | |
| 38 | |
| 39 | (De)compression of files |
| 40 | ------------------------ |
| 41 | |
| 42 | Handling of compressed files is offered by the :class:`BZ2File` class. |
| 43 | |
| 44 | |
R David Murray | 5618aaa | 2012-08-15 11:15:39 -0400 | [diff] [blame] | 45 | .. index:: |
| 46 | single: universal newlines; bz2.BZ2File class |
| 47 | |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 48 | .. class:: BZ2File(filename[, mode[, buffering[, compresslevel]]]) |
| 49 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 50 | Open a bz2 file. Mode can be either ``'r'`` or ``'w'``, for reading (default) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 51 | or writing. When opened for writing, the file will be created if it doesn't |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 52 | exist, and truncated otherwise. If *buffering* is given, ``0`` means |
| 53 | unbuffered, and larger numbers specify the buffer size; the default is |
| 54 | ``0``. If *compresslevel* is given, it must be a number between ``1`` and |
| 55 | ``9``; the default is ``9``. Add a ``'U'`` to mode to open the file for input |
R David Murray | 5618aaa | 2012-08-15 11:15:39 -0400 | [diff] [blame] | 56 | in :term:`universal newlines` mode. Any line ending in the input file will be |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 57 | seen as a ``'\n'`` in Python. Also, a file so opened gains the attribute |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 58 | :attr:`newlines`; the value for this attribute is one of ``None`` (no newline |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 59 | read yet), ``'\r'``, ``'\n'``, ``'\r\n'`` or a tuple containing all the |
| 60 | newline types seen. Universal newlines are available only when |
| 61 | reading. Instances support iteration in the same way as normal :class:`file` |
| 62 | instances. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 63 | |
Benjamin Peterson | 6d83429 | 2009-03-09 20:38:56 +0000 | [diff] [blame] | 64 | :class:`BZ2File` supports the :keyword:`with` statement. |
| 65 | |
| 66 | .. versionchanged:: 2.7 |
| 67 | Support for the :keyword:`with` statement was added. |
| 68 | |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 69 | |
Nadeem Vawda | acd51f9 | 2012-02-04 23:43:25 +0200 | [diff] [blame] | 70 | .. note:: |
| 71 | |
| 72 | This class does not support input files containing multiple streams (such |
| 73 | as those produced by the :program:`pbzip2` tool). When reading such an |
| 74 | input file, only the first stream will be accessible. If you require |
Nadeem Vawda | d15c5a8 | 2012-02-05 14:27:01 +0200 | [diff] [blame] | 75 | support for multi-stream files, consider using the third-party |
| 76 | :mod:`bz2file` module (available from |
Georg Brandl | 06f3b3b | 2014-10-29 08:36:35 +0100 | [diff] [blame] | 77 | `PyPI <https://pypi.python.org/pypi/bz2file>`_). This module provides a |
Nadeem Vawda | d15c5a8 | 2012-02-05 14:27:01 +0200 | [diff] [blame] | 78 | backport of Python 3.3's :class:`BZ2File` class, which does support |
| 79 | multi-stream files. |
Nadeem Vawda | acd51f9 | 2012-02-04 23:43:25 +0200 | [diff] [blame] | 80 | |
| 81 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 82 | .. method:: close() |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 83 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 84 | Close the file. Sets data attribute :attr:`closed` to true. A closed file |
| 85 | cannot be used for further I/O operations. :meth:`close` may be called |
| 86 | more than once without error. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 87 | |
| 88 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 89 | .. method:: read([size]) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 90 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 91 | Read at most *size* uncompressed bytes, returned as a string. If the |
| 92 | *size* argument is negative or omitted, read until EOF is reached. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 93 | |
| 94 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 95 | .. method:: readline([size]) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 96 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 97 | Return the next line from the file, as a string, retaining newline. A |
| 98 | non-negative *size* argument limits the maximum number of bytes to return |
| 99 | (an incomplete line may be returned then). Return an empty string at EOF. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 100 | |
| 101 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 102 | .. method:: readlines([size]) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 103 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 104 | Return a list of lines read. The optional *size* argument, if given, is an |
| 105 | approximate bound on the total number of bytes in the lines returned. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 106 | |
| 107 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 108 | .. method:: xreadlines() |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 109 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 110 | For backward compatibility. :class:`BZ2File` objects now include the |
| 111 | performance optimizations previously implemented in the :mod:`xreadlines` |
| 112 | module. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 113 | |
Georg Brandl | c62ef8b | 2009-01-03 20:55:06 +0000 | [diff] [blame] | 114 | .. deprecated:: 2.3 |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 115 | This exists only for compatibility with the method by this name on |
| 116 | :class:`file` objects, which is deprecated. Use ``for line in file`` |
| 117 | instead. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 118 | |
| 119 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 120 | .. method:: seek(offset[, whence]) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 121 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 122 | Move to new file position. Argument *offset* is a byte count. Optional |
| 123 | argument *whence* defaults to ``os.SEEK_SET`` or ``0`` (offset from start |
| 124 | of file; offset should be ``>= 0``); other values are ``os.SEEK_CUR`` or |
| 125 | ``1`` (move relative to current position; offset can be positive or |
| 126 | negative), and ``os.SEEK_END`` or ``2`` (move relative to end of file; |
| 127 | offset is usually negative, although many platforms allow seeking beyond |
| 128 | the end of a file). |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 129 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 130 | Note that seeking of bz2 files is emulated, and depending on the |
| 131 | parameters the operation may be extremely slow. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 132 | |
| 133 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 134 | .. method:: tell() |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 135 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 136 | Return the current file position, an integer (may be a long integer). |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 137 | |
| 138 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 139 | .. method:: write(data) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 140 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 141 | Write string *data* to file. Note that due to buffering, :meth:`close` may |
| 142 | be needed before the file on disk reflects the data written. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 143 | |
| 144 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 145 | .. method:: writelines(sequence_of_strings) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 146 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 147 | Write the sequence of strings to the file. Note that newlines are not |
| 148 | added. The sequence can be any iterable object producing strings. This is |
| 149 | equivalent to calling write() for each string. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 150 | |
| 151 | |
| 152 | Sequential (de)compression |
| 153 | -------------------------- |
| 154 | |
| 155 | Sequential compression and decompression is done using the classes |
| 156 | :class:`BZ2Compressor` and :class:`BZ2Decompressor`. |
| 157 | |
| 158 | |
| 159 | .. class:: BZ2Compressor([compresslevel]) |
| 160 | |
| 161 | Create a new compressor object. This object may be used to compress data |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 162 | sequentially. If you want to compress data in one shot, use the |
| 163 | :func:`compress` function instead. The *compresslevel* parameter, if given, |
| 164 | must be a number between ``1`` and ``9``; the default is ``9``. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 165 | |
| 166 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 167 | .. method:: compress(data) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 168 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 169 | Provide more data to the compressor object. It will return chunks of |
| 170 | compressed data whenever possible. When you've finished providing data to |
| 171 | compress, call the :meth:`flush` method to finish the compression process, |
| 172 | and return what is left in internal buffers. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 173 | |
| 174 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 175 | .. method:: flush() |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 176 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 177 | Finish the compression process and return what is left in internal |
| 178 | buffers. You must not use the compressor object after calling this method. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 179 | |
| 180 | |
| 181 | .. class:: BZ2Decompressor() |
| 182 | |
| 183 | Create a new decompressor object. This object may be used to decompress data |
| 184 | sequentially. If you want to decompress data in one shot, use the |
| 185 | :func:`decompress` function instead. |
| 186 | |
| 187 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 188 | .. method:: decompress(data) |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 189 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 190 | Provide more data to the decompressor object. It will return chunks of |
| 191 | decompressed data whenever possible. If you try to decompress data after |
| 192 | the end of stream is found, :exc:`EOFError` will be raised. If any data |
| 193 | was found after the end of stream, it'll be ignored and saved in |
| 194 | :attr:`unused_data` attribute. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 195 | |
| 196 | |
| 197 | One-shot (de)compression |
| 198 | ------------------------ |
| 199 | |
| 200 | One-shot compression and decompression is provided through the :func:`compress` |
| 201 | and :func:`decompress` functions. |
| 202 | |
| 203 | |
| 204 | .. function:: compress(data[, compresslevel]) |
| 205 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 206 | Compress *data* in one shot. If you want to compress data sequentially, use |
| 207 | an instance of :class:`BZ2Compressor` instead. The *compresslevel* parameter, |
| 208 | if given, must be a number between ``1`` and ``9``; the default is ``9``. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 209 | |
| 210 | |
| 211 | .. function:: decompress(data) |
| 212 | |
Benjamin Peterson | c7b0592 | 2008-04-25 01:29:10 +0000 | [diff] [blame] | 213 | Decompress *data* in one shot. If you want to decompress data sequentially, |
| 214 | use an instance of :class:`BZ2Decompressor` instead. |
Georg Brandl | 8ec7f65 | 2007-08-15 14:28:01 +0000 | [diff] [blame] | 215 | |