Merge ca296b41c6 into 333567f481

sticker/lib/util.py

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