deflate.py

import sys
import zlib


class DeflateIO:

    def __init__(self, stream) -> None:
        self.stream = stream
        self.data = stream.read()

    def read(self):
        return zlib.decompress(self.data, wbits=-10)

    def write(self, input_data):
        compressor = zlib.compressobj(wbits=-10)
        compressed_data = compressor.compress(input_data)
        compressed_data += compressor.flush()
        self.stream.seek(0)  # Ensure we overwrite the stream from the beginning
        self.stream.write(compressed_data)
        self.stream.truncate()  # Remove any remaining part of the old data

    def __enter__(self):
        # Return the instance itself when entering the context
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # Handle cleanup here if necessary
        pass


class Deflate:
    DeflateIO = DeflateIO

if "deflate" not in sys.modules:
    sys.modules["deflate"] = Deflate