PNG_SIGNATURE = b'\x89PNG\r\n\x1a\n' CHUNK_TYPE_MPNG = b'mPNg' # Custom MPNG chunk CHUNK_TYPE_IDAT = b'IDAT' CHUNK_TYPE_IEND = b'IEND'

def read_chunk(f): """Read PNG chunk: length, type, data, crc""" len_data = struct.unpack('>I', f.read(4))[0] chunk_type = f.read(4) data = f.read(len_data) crc = struct.unpack('>I', f.read(4))[0] return chunk_type, data, crc

replace_map = None if replace_data: # Format: "idx:file.zlib" or "idx:file.raw" idx_str, path_str = replace_data.split(':') idx = int(idx_str) data = Path(path_str).read_bytes() # If raw, compress it if path_str.endswith('.raw'): data = zlib.compress(data, level) replace_map = {idx: data}

def rebuild_png_with_mpng(input_path, output_path, streams, recompress_level=6, replace_map=None): """Rebuild PNG with modified MPNG streams""" with open(input_path, 'rb') as fin, open(output_path, 'wb') as fout: fout.write(fin.read(8)) # PNG signature

Here’s a solid, production-ready feature implementation for that adds support for MPNG (Multi-Piece PNG) + reflate (recompressing/zlib stream repair).