mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PYTHON-3159 Fix typo in zlib compression support (#894)

Browse Source
This commit is contained in:
Shane Harvey 2022-03-04 17:29:33 -08:00 committed by GitHub
parent 671d1e622c
commit f081297a86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pymongo/compression_support.py
View File

@ -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):