Update khandy/image/misc.py

khandy/image/misc.py

@@ -1,5 +1,7 @@
 import os
-import imghdr
+# import imghdr
+import io
+from PIL import Image
 import numbers
 import warnings
 from io import BytesIO
@@ -80,7 +82,12 @@ def imwrite_bytes(filename, image_bytes: bytes, update_extension: bool = True):
             whether update extension according to image_bytes or not.
             the cost of update extension is smaller than update image format.
     """
-    extension = imghdr.what('', image_bytes)
+    # extension = imghdr.what('', image_bytes)
+    try:
+    with Image.open(io.BytesIO(image_bytes)) as img:
+        extension = img.format.lower()  # .format renvoie 'JPEG', 'PNG', etc.
+except Exception:
+    extension = None
     file_extension = khandy.get_path_extension(filename)
     # imghdr.what fails to determine image format sometimes!
     # so when its return value is None, never update extension.