Skip to content

Commit f081297

Browse files
authored
PYTHON-3159 Fix typo in zlib compression support (#894)
1 parent 671d1e6 commit f081297

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

pymongo/compression_support.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
# limitations under the License.
1414

1515
import warnings
16-
from typing import Callable
1716

1817
try:
1918
import snappy
@@ -105,12 +104,6 @@ def get_compression_context(self, compressors):
105104
return ZstdContext()
106105

107106

108-
def _zlib_no_compress(data, level=None):
109-
"""Compress data with zlib level 0."""
110-
cobj = zlib.compressobj(0)
111-
return b"".join([cobj.compress(data), cobj.flush()])
112-
113-
114107
class SnappyContext(object):
115108
compressor_id = 1
116109

@@ -123,16 +116,10 @@ class ZlibContext(object):
123116
compressor_id = 2
124117

125118
def __init__(self, level):
126-
self.compress: Callable[[bytes], bytes]
127-
128-
# Jython zlib.compress doesn't support -1
129-
if level == -1:
130-
self.compress = zlib.compress
131-
# Jython zlib.compress also doesn't support 0
132-
elif level == 0:
133-
self.compress = _zlib_no_compress
134-
else:
135-
self.compresss = lambda data, _: zlib.compress(data, level)
119+
self.level = level
120+
121+
def compress(self, data: bytes) -> bytes:
122+
return zlib.compress(data, self.level)
136123

137124

138125
class ZstdContext(object):

0 commit comments

Comments
 (0)