added avif support to sorl-thumbnail

Author: amirreza8002

sorl_thumbnail_avif/__init__.py

@@ -0,0 +1 @@
+from sorl_thumbnail_avif.thumbnail.base import AvifThumbnail

sorl_thumbnail_avif/thumbnail/__init__.py

@@ -0,0 +1,40 @@
+from sorl.thumbnail.base import ThumbnailBackend
+from sorl.thumbnail.conf import settings
+from sorl.thumbnail.conf import defaults as default_settings
+from sorl.thumbnail.helpers import serialize, tokey
+
+
+EXTENSIONS = {
+    "JPEG": "jpg",
+    "PNG": "png",
+    "GIF": "gif",
+    "WEBP": "webp",
+    "AVIF": "avif",
+}
+
+
+class AvifThumbnail(ThumbnailBackend):
+    def _get_format(self, source):
+        file_extension = self.file_extension(source)
+
+        if file_extension == ".avif":
+            return "AVIF"
+        elif file_extension == ".jpg" or file_extension == ".jpeg":
+            return "JPEG"
+        elif file_extension == ".png":
+            return "PNG"
+        elif file_extension == ".gif":
+            return "GIF"
+        elif file_extension == ".webp":
+            return "WEBP"
+        else:
+            from django.conf import settings
+
+            return getattr(
+                settings, "THUMBNAIL_FORMAT", default_settings.THUMBNAIL_FORMAT
+            )
+
+    def _get_thumbnail_filename(self, source, geometry_string, options):
+        key = tokey(source.key, geometry_string, serialize(options))
+        path = f"{key[:2]}{key[2:4]}{key}"
+        return f"{settings.THUMBNAIL_PREFIX}{path}.{EXTENSIONS[options["format"]]}"

sorl_thumbnail_avif/thumbnail/base.py

@@ -0,0 +1 @@
+from sorl_thumbnail_avif.thumbnail.engines.pil_engine import AvifEngine

sorl_thumbnail_avif/thumbnail/engines/__init__.py

@@ -0,0 +1,65 @@
+from io import BytesIO
+from sorl.thumbnail.engines.pil_engine import Engine
+
+from PIL import Image, ImageFile
+import pillow_avif  # noqa: F401
+
+
+class AvifEngine(Engine):
+    def get_image(self, source):
+        buffer = BytesIO(source.read())
+        return Image.open(buffer)
+
+    def is_valid_image(self, raw_data):
+        buffer = BytesIO(raw_data)
+        try:
+            trial_image = Image.open(buffer)
+            trial_image.verify()
+        except Exception:
+            return False
+        return True
+
+    def _padding(self, image, geometry, options):
+        x_image, y_image = self.get_image_size(image)
+        left = int((geometry[0] - x_image) / 2)
+        top = int((geometry[1] - y_image) / 2)
+        color = options.get("padding_color")
+        im = Image.new(image.mode, geometry, color)
+        im.paste(image, (left, top))
+        return im
+
+    def _get_raw_data(
+        self, image, format_, quality, image_info=None, progressive=False
+    ):
+        # Increase (but never decrease) PIL buffer size
+        ImageFile.MAXBLOCK = max(ImageFile.MAXBLOCK, image.size[0] * image.size[1])
+        bf = BytesIO()
+
+        params = {
+            "format": format_,
+            "quality": quality,
+            "optimize": 1,
+        }
+
+        # keeps icc_profile
+        if "icc_profile" in image_info:
+            params["icc_profile"] = image_info["icc_profile"]
+
+        raw_data = None
+
+        if format_ == "JPEG" and progressive:
+            params["progressive"] = True
+        try:
+            # Do not save unnecessary exif data for smaller thumbnail size
+            params.pop("exif", {})
+            image.save(bf, **params)
+        except OSError:
+            # Try without optimization.
+            params.pop("optimize")
+            image.save(bf, **params)
+        else:
+            raw_data = bf.getvalue()
+        finally:
+            bf.close()
+
+        return raw_data