python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-26 18:06:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Lint fixes

Browse Source
This commit is contained in:
Andrew Murray 2020-04-04 15:02:02 +11:00
parent b685c18e95
commit 4500acb274
1 changed files with 16 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/PIL/IcnsImagePlugin.py
View File

@ -19,11 +19,8 @@
import io
import os
import shutil
import struct
import subprocess
import sys
import tempfile
from PIL import Image, ImageFile, PngImagePlugin
from PIL._binary import i8
@ -304,13 +301,13 @@ class IcnsImageFile(ImageFile.ImageFile):
def to_int(s):
b = s.encode('ascii')
b = s.encode("ascii")
return (b[0] << 24) | (b[1] << 16) | (b[2] << 8) | b[3]
MAGIC = to_int("icns")
HEADER_SIZE = 8
TOC = 'TOC '
TOC = "TOC "
def _save(im, fp, filename):
@ -325,37 +322,35 @@ def _save(im, fp, filename):
# size
sizes = [128, 256, 512, 32, 64, 256, 512, 1024]
size_str = ['ic07', 'ic08', 'ic09', 'ic11', 'ic12', 'ic13', 'ic14', 'ic10']
size_str = ["ic07", "ic08", "ic09", "ic11", "ic12", "ic13", "ic14", "ic10"]
file_size = 0
entries = []
for index, s in enumerate(sizes):
temp = io.BytesIO()
nb = im.resize((s, s))
nb.save(temp, 'png')
nb.save(temp, "png")
file_size += len(temp.getvalue())
entries.append({
'type': size_str[index],
'size': len(temp.getvalue()),
'stream': temp
})
entries.append(
{"type": size_str[index], "size": len(temp.getvalue()), "stream": temp}
)
# Header
fp.write(struct.pack('i', MAGIC)[::-1])
fp.write(struct.pack('i', file_size)[::-1])
fp.write(struct.pack("i", MAGIC)[::-1])
fp.write(struct.pack("i", file_size)[::-1])
# TOC
toc_size = HEADER_SIZE + (len(entries) * HEADER_SIZE)
fp.write(struct.pack('i', to_int(TOC))[::-1])
fp.write(struct.pack('i', toc_size)[::-1])
fp.write(struct.pack("i", to_int(TOC))[::-1])
fp.write(struct.pack("i", toc_size)[::-1])
for e in entries:
fp.write(struct.pack('i', to_int(e.get('type')))[::-1])
fp.write(struct.pack('i', HEADER_SIZE + e.get('size'))[::-1])
fp.write(struct.pack("i", to_int(e.get("type")))[::-1])
fp.write(struct.pack("i", HEADER_SIZE + e.get("size"))[::-1])
# Data
for index, e in enumerate(entries):
fp.write(struct.pack('i', to_int(e.get('type')))[::-1])
fp.write(struct.pack('i', HEADER_SIZE + e.get('size'))[::-1])
fp.write(e.get('stream').getvalue())
fp.write(struct.pack("i", to_int(e.get("type")))[::-1])
fp.write(struct.pack("i", HEADER_SIZE + e.get("size"))[::-1])
fp.write(e.get("stream").getvalue())
fp.flush()