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."""
def compress(self, stream):
return BytesIO(bz2.compress(b''.join(stream)))
def decompress(self, stream):
return BytesIO(bz2.decompress(stream.read()))