def read(pngFilename): yuml = '<<no yuml found>>' pngFile = file(pngFilename, 'rb') png.read_signature(pngFile) for chunk in png.all_chunks(pngFile): if chunk.chunk_type == 'iTXt': chunk = png.iTXtChunk(chunk) if chunk.keyword == 'yuml': yuml = chunk.text break pngFile.close() return yuml
def read(pngFilename): yuml = "<<no yuml found>>" pngFile = file(pngFilename, "rb") png.read_signature(pngFile) for chunk in png.all_chunks(pngFile): if chunk.chunk_type == "iTXt": chunk = png.iTXtChunk(chunk) if chunk.keyword == "yuml": yuml = chunk.text break pngFile.close() return yuml
def add_yuml_to_png(yuml, in_stream, out_stream): signature = png.read_signature(in_stream) out_stream.write(signature) for chunk in png.all_chunks(in_stream): if chunk.chunk_type == 'IEND': break chunk.write(out_stream) itxt_chunk = png.iTXtChunk.create('yuml', yuml) itxt_chunk.write(out_stream) # write the IEND chunk chunk.write(out_stream)