python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-06-29 17:33:08 +03:00
Code Issues Packages Projects Releases Wiki Activity

Simplified use of struct

Browse Source
This commit is contained in:
Andrew Murray 2021-06-29 20:20:34 +10:00
parent f5558f42f4
commit d07a085757
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/PIL/IcnsImagePlugin.py
View File

@ -331,24 +331,22 @@ def _save(im, fp, filename):
entries.append( entries.append(
{"type": _to_int(size_str[index]), "size": len(stream), "stream": stream} {"type": _to_int(size_str[index]), "size": len(stream), "stream": stream}
) )
file_size = sum(entry["size"] for entry in entries)
# Header # Header
fp.write(struct.pack("<i", _to_int(MAGIC))[::-1]) fp.write(struct.pack(">i", _to_int(MAGIC)))
fp.write(struct.pack("<i", file_size)[::-1]) fp.write(struct.pack(">i", sum(entry["size"] for entry in entries)))
# TOC # TOC
toc_size = HEADERSIZE + (len(entries) * HEADERSIZE) fp.write(struct.pack(">i", _to_int(TOC)))
fp.write(struct.pack("<i", _to_int(TOC))[::-1]) fp.write(struct.pack(">i", HEADERSIZE + len(entries) * HEADERSIZE))
fp.write(struct.pack("<i", toc_size)[::-1])
for entry in entries: for entry in entries:
fp.write(struct.pack("<i", entry["type"])[::-1]) fp.write(struct.pack(">i", entry["type"]))
fp.write(struct.pack("<i", HEADERSIZE + entry["size"])[::-1]) fp.write(struct.pack(">i", HEADERSIZE + entry["size"]))
# Data # Data
for entry in entries: for entry in entries:
fp.write(struct.pack("<i", entry["type"])[::-1]) fp.write(struct.pack(">i", entry["type"]))
fp.write(struct.pack("<i", HEADERSIZE + entry["size"])[::-1]) fp.write(struct.pack(">i", HEADERSIZE + entry["size"]))
fp.write(entry["stream"]) fp.write(entry["stream"])
if hasattr(fp, "flush"): if hasattr(fp, "flush"):