Module refinery.units.compression.zstd

Expand source code Browse git 
from __future__ import annotations

from refinery.lib.shared import pyzstd
from refinery.units import RefineryPartialResult, Unit


class zstd(Unit):
    """
    ZStandard (ZSTD) compression and decompression.
    """
    def process(self, data):
        zd = pyzstd.ZstdDecompressor()
        out = zd.decompress(data)
        if zd.needs_input:
            raise RefineryPartialResult('Incomplete ZSTD stream.', out)
        return out

    def reverse(self, data):
        zc = pyzstd.ZstdCompressor()
        return zc.compress(data) + zc.flush()

    @classmethod
    def handles(cls, data) -> bool:
        return data[:4] == B'\x28\xB5\x2F\xFD'

Classes

class zstd

ZStandard (ZSTD) compression and decompression.

Expand source code Browse git 
class zstd(Unit):
    """
    ZStandard (ZSTD) compression and decompression.
    """
    def process(self, data):
        zd = pyzstd.ZstdDecompressor()
        out = zd.decompress(data)
        if zd.needs_input:
            raise RefineryPartialResult('Incomplete ZSTD stream.', out)
        return out

    def reverse(self, data):
        zc = pyzstd.ZstdCompressor()
        return zc.compress(data) + zc.flush()

    @classmethod
    def handles(cls, data) -> bool:
        return data[:4] == B'\x28\xB5\x2F\xFD'

Ancestors

Subclasses

Class variables

var required_dependencies
var optional_dependencies
var console

Methods

def reverse(self, data)
Expand source code Browse git 
def reverse(self, data):
    zc = pyzstd.ZstdCompressor()
    return zc.compress(data) + zc.flush()

Inherited members