jan/fatamorgana
2
1
Fork 1
Code Issues Pull requests Releases Wiki Activity

Add default for CBlock decompression_args

Browse source
This commit is contained in:
Jan Petykiewicz 2018-07-21 13:46:48 -07:00
commit ffb211b009
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
fatamorgana/records.py
View file

@ -524,7 +524,7 @@ class CBlock(Record):
return CBlock(compression_type, count, compressed_bytes) return CBlock(compression_type, count, compressed_bytes)
def decompress(self, decompression_args) -> bytes: def decompress(self, decompression_args: Dict = None) -> bytes:
""" """
Decompress the contents of this CBlock. Decompress the contents of this CBlock.
@ -533,6 +533,8 @@ class CBlock(Record):
:raises: InvalidDataError if data is malformed or compression type is :raises: InvalidDataError if data is malformed or compression type is
unknonwn. unknonwn.
""" """
if decompression_args is None:
decompression_args = {}
if self.compression_type == 0: if self.compression_type == 0:
decompressor = zlib.decompressobj(wbits=-zlib.MAX_WBITS, **decompression_args) decompressor = zlib.decompressobj(wbits=-zlib.MAX_WBITS, **decompression_args)
decompressed_bytes = decompressor.decompress(self.compressed_bytes) + \ decompressed_bytes = decompressor.decompress(self.compressed_bytes) + \