Add avif and heif image format support

iscc · Dec 3, 2023 · 1d7bf80 · 1d7bf80
1 parent 854ff52
commit 1d7bf80
Show file tree

Hide file tree

Showing 4 changed files with 184 additions and 70 deletions.
diff --git a/iscc_sdk/image.py b/iscc_sdk/image.py
@@ -1,4 +1,5 @@
 """*Image handling module*."""
+import pillow_avif
 import base64
 import io
 import shutil
@@ -12,6 +13,9 @@
 from loguru import logger as log
 from PIL import Image, ImageEnhance, ImageChops, ImageOps
 import iscc_sdk as idk
+from pillow_heif import register_heif_opener
+
+register_heif_opener()
 
 
 __all__ = [

diff --git a/iscc_sdk/mediatype.py b/iscc_sdk/mediatype.py
@@ -198,6 +198,8 @@ def mediatype_to_mode(mime_type):
     "application/vnd.openxmlformats-officedocument.wordprocessingml.document", ".docx"
 )
 mimetypes.add_type("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", ".xlsx")
+mimetypes.add_type("image/heif", ".heif")
+mimetypes.add_type("image/avif", ".avif")
 
 
 SUPPORTED_MEDIATYPES = {
@@ -244,6 +246,8 @@ def mediatype_to_mode(mime_type):
     "image/tiff": {"mode": "image", "ext": "tif"},
     "image/vnd.adobe.photoshop": {"mode": "image", "ext": "psd"},
     "application/postscript": {"mode": "image", "ext": "eps"},
+    "image/avif": {"mode": "image", "ext": "avif"},
+    "image/heif": {"mode": "image", "ext": "heif"},
     # Audio Formats
     "audio/mpeg": {"mode": "audio", "ext": "mp3"},
     "audio/wav": {"mode": "audio", "ext": "wav"},