import pyzstd from ..exceptions import CompressorError from .base import BaseCompressor class ZStdCompressor(BaseCompressor): min_length = 15 def compress(self, value: bytes) -> bytes: if len(value) > self.min_length: return pyzstd.compress(value) return value def decompress(self, value: bytes) -> bytes: try: return pyzstd.decompress(value) except pyzstd.ZstdError as e: raise CompressorError(e)