Add basic support for .heic files

pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
-all = ["boto", "brotli", "feedgenerator", "zopfli", "cryptography"]
+all = ["boto", "brotli", "feedgenerator", "zopfli", "cryptography", "pillow-heif"]
 tests = ["pytest", "pytest-cov"]
 docs = ["Sphinx>=4.1.0", "furo", "cryptography"]
 

src/sigal/gallery.py

@@ -264,7 +264,7 @@ def exif(self):
         datetime_format = self.settings["datetime_format"]
         return (
             get_exif_tags(self.raw_exif, datetime_format=datetime_format)
-            if self.raw_exif and self.src_ext in (".jpg", ".jpeg")
+            if self.raw_exif and self.src_ext in (".jpg", ".jpeg", ".heic")
             else None
         )
 
@@ -289,7 +289,7 @@ def _get_markdown_metadata(self):
     @cached_property
     def raw_exif(self):
         """If not `None`, contains the raw EXIF tags."""
-        if self.src_ext in (".jpg", ".jpeg"):
+        if self.src_ext in (".jpg", ".jpeg", ".heic"):
             return self.file_metadata["exif"]
 
     @cached_property

src/sigal/image.py

@@ -44,6 +44,11 @@
 from pilkit.processors import Transpose
 from pilkit.utils import save_image
 
+try:
+    from pillow_heif import HeifImagePlugin
+except:
+    pass
+
 from . import signals, utils
 
 # Force loading of truncated files
@@ -220,7 +225,11 @@ def get_exif_data(filename):
 
     try:
         with warnings.catch_warnings(record=True) as caught_warnings:
-            exif = img._getexif() or {}
+            exif = {}
+            if hasattr(img, "_getexif") and callable(img._getexif):
+                exif = img._getexif()
+            elif "exif" in img.info:
+                exif = img.getexif()._get_merged_dict()
     except ZeroDivisionError:
         logger.warning("Failed to read EXIF data.")
         return None
@@ -290,7 +299,7 @@ def get_image_metadata(filename):
         logger.error("Could not open image %s metadata: %s", filename, e)
     else:
         try:
-            if os.path.splitext(filename)[1].lower() in (".jpg", ".jpeg"):
+            if os.path.splitext(filename)[1].lower() in (".jpg", ".jpeg", ".heic"):
                 exif = get_exif_data(img)
         except Exception as e:
             logger.warning("Could not read EXIF data from %s: %s", filename, e)