@@ -141,7 +141,7 @@ def __init__(
141
141
else :
142
142
self ._default_zstd_compressor = None
143
143
144
- self ._zstd_decompressors [0 ] = zstd .ZstdDecompressor ()
144
+ self ._zstd_decompressors [0 ] = zstd .ZstdDecompressor (format = zstd . FORMAT_ZSTD1_MAGICLESS )
145
145
146
146
def _build_dict (self , dictionary : bytes ) -> Tuple [int , zstd .ZstdCompressionDict ]:
147
147
zstd_dict = zstd .ZstdCompressionDict (dictionary )
@@ -151,7 +151,7 @@ def _build_dict(self, dictionary: bytes) -> Tuple[int, zstd.ZstdCompressionDict]
151
151
def _add_dict_decompressor (
152
152
self , dict_id : int , zstd_dict : zstd .ZstdCompressionDict
153
153
) -> zstd .ZstdDecompressor :
154
- self ._zstd_decompressors [dict_id ] = zstd .ZstdDecompressor (dict_data = zstd_dict )
154
+ self ._zstd_decompressors [dict_id ] = zstd .ZstdDecompressor (dict_data = zstd_dict , format = zstd . FORMAT_ZSTD1_MAGICLESS )
155
155
return self ._zstd_decompressors [dict_id ]
156
156
157
157
def _add_dict_compressor (
@@ -174,8 +174,7 @@ def _compress(self, key: Key, data: bytes) -> Tuple[bytes, int]:
174
174
return zlib .compress (data ), self .ZLIB_COMPRESSED
175
175
176
176
def _decompress (self , data : bytes ) -> bytes :
177
- data = self .ZSTD_MAGIC + data
178
- dict_id = zstd .get_frame_parameters (data ).dict_id
177
+ dict_id = zstd .get_frame_parameters (self .ZSTD_MAGIC + data ).dict_id
179
178
if decompressor := self ._zstd_decompressors .get (dict_id ):
180
179
return decompressor .decompress (data )
181
180
raise ValueError (f"Unknown dictionary id: { dict_id } " )
0 commit comments