From f081297a8634abb77f85a8f06c552c76a51f1120 Mon Sep 17 00:00:00 2001 From: Shane Harvey Date: Fri, 4 Mar 2022 17:29:33 -0800 Subject: [PATCH] PYTHON-3159 Fix typo in zlib compression support (#894) --- pymongo/compression_support.py | 19 +++---------------- 1 file changed, 3 insertions(+), 16 deletions(-) diff --git a/pymongo/compression_support.py b/pymongo/compression_support.py index ed7021494..c9632a43d 100644 --- a/pymongo/compression_support.py +++ b/pymongo/compression_support.py @@ -13,7 +13,6 @@ # limitations under the License. import warnings -from typing import Callable try: import snappy @@ -105,12 +104,6 @@ class CompressionSettings(object): return ZstdContext() -def _zlib_no_compress(data, level=None): - """Compress data with zlib level 0.""" - cobj = zlib.compressobj(0) - return b"".join([cobj.compress(data), cobj.flush()]) - - class SnappyContext(object): compressor_id = 1 @@ -123,16 +116,10 @@ class ZlibContext(object): compressor_id = 2 def __init__(self, level): - self.compress: Callable[[bytes], bytes] + self.level = level - # Jython zlib.compress doesn't support -1 - if level == -1: - self.compress = zlib.compress - # Jython zlib.compress also doesn't support 0 - elif level == 0: - self.compress = _zlib_no_compress - else: - self.compresss = lambda data, _: zlib.compress(data, level) + def compress(self, data: bytes) -> bytes: + return zlib.compress(data, self.level) class ZstdContext(object):