o ˳i"@spddlmZddlmZddlmZzddlmZmZGdddeZ WdSe y7GdddeZ YdSw) ) annotations)BytesIO)Codec)ZstdCompressorZstdDecompressorc@(eZdZed ddZed ddZd S) ZStandardCodecdatabytesreturntuple[bytes, int]cCst|}|t|fS)N)rcompresslen)r compressed_datarY/home/ubuntu/.local/lib/python3.10/site-packages/pyiceberg/avro/codecs/zstandard_codec.pyr s  ZStandardCodec.compresscCsft}t}|t|} |d}|sn||qWdt|S1s*wYt|S)NTi@) bytearrayr stream_readerrreadextendr )r uncompresseddctxreaderchunkrrr decompress s   ZStandardCodec.decompressNr r r r r r r r __name__ __module__ __qualname__ staticmethodr rrrrrrs  rc@r) rr r r r cCtdNzXZstandard support not installed, please install using `pip install pyiceberg[zstandard]` ImportErrorr rrrr /rcCr$r%r&r(rrrr3r)rNrrrrrrrr.s  N) __future__riorpyiceberg.avro.codecs.codecr zstandardrrrr'rrrrs