rei-mirror
/
stickerpicker
mirror of https://github.com/maunium/stickerpicker
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

perf: improved performance of webp_to_others

This commit is contained in:
xz-dev 2024-09-16 20:08:05 +08:00
parent ca296b41c6
commit aa8616a40a
No known key found for this signature in database
GPG Key ID: A20912F811313E36
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
sticker/lib/util.py
View File

@ -120,28 +120,26 @@ def process_frame(frame):
def webp_to_others(data: bytes, mimetype: str) -> bytes:
with tempfile.NamedTemporaryFile(suffix=".webp") as webp:
webp.write(data)
webp.flush()
ext = mimetypes.guess_extension(mimetype)
with tempfile.NamedTemporaryFile(suffix=ext) as img:
format = mimetypes.guess_extension(mimetype)[1:]
print(format)
with Image.open(BytesIO(data)) as webp:
with BytesIO() as img:
print(".", end="", flush=True)
im = Image.open(webp.name)
im.info.pop('background', None)
webp.info.pop('background', None)
if mimetype == "image/gif":
frames = []
duration = []
duration = [100, ]
for frame in ImageSequence.Iterator(im):
for frame in ImageSequence.Iterator(webp):
frame = process_frame(frame)
frames.append(frame)
duration.append(frame.info.get('duration', 100))
duration.append(frame.info.get('duration', duration[-1]))
frames[0].save(img.name, save_all=True, lossless=True, quality=100, method=6,
append_images=frames[1:], loop=0, duration=duration, disposal=2)
frames[0].save(img, format=format, save_all=True, lossless=True, quality=100, method=6,
append_images=frames[1:], loop=0, duration=duration[1:], disposal=2)
else:
im.save(img.name, save_all=True, lossless=True, quality=100, method=6)
webp.save(img, format=format, lossless=True, quality=100, method=6)
img.seek(0)
return img.read()