Source code for pysparkling.fileio.codec.bz2

from __future__ import absolute_import

import bz2
from io import BytesIO
import logging

from .codec import Codec

log = logging.getLogger(__name__)


[docs]class Bz2(Codec):
    """Implementation of :class:`.Codec` for bz2 compression."""

[docs]    def compress(self, stream):
        return BytesIO(bz2.compress(b''.join(stream)))

[docs]    def decompress(self, stream):
        return BytesIO(bz2.decompress(stream.read()))