Add support for modern image formats (WebP and AVIF) and implement image conversion commands

twitch/management/commands/download_box_art.py

@@ -10,6 +10,7 @@ from django.conf import settings
 from django.core.files.base import ContentFile
 from django.core.management.base import BaseCommand
 from django.core.management.base import CommandParser
+from PIL import Image
 
 from twitch.models import Game
 from twitch.utils import is_twitch_box_art_url
@@ -89,6 +90,10 @@ class Command(BaseCommand):
                     continue
 
                 game.box_art_file.save(file_name, ContentFile(response.content), save=True)
+
+                # Auto-convert to WebP and AVIF
+                self._convert_to_modern_formats(game.box_art_file.path)
+
                 downloaded += 1
 
         self.stdout.write(
@@ -99,3 +104,40 @@ class Command(BaseCommand):
         )
         box_art_dir: Path = Path(settings.MEDIA_ROOT) / "games" / "box_art"
         self.stdout.write(self.style.SUCCESS(f"Saved box art to: {box_art_dir}"))
+
+    def _convert_to_modern_formats(self, image_path: str) -> None:
+        """Convert downloaded image to WebP and AVIF formats.
+
+        Args:
+            image_path: Absolute path to the downloaded image file
+        """
+        try:
+            source_path = Path(image_path)
+            if not source_path.exists() or source_path.suffix.lower() not in {".jpg", ".jpeg", ".png"}:
+                return
+
+            base_path = source_path.with_suffix("")
+            webp_path = base_path.with_suffix(".webp")
+            avif_path = base_path.with_suffix(".avif")
+
+            with Image.open(source_path) as img:
+                # Convert to RGB if needed
+                if img.mode in {"RGBA", "LA"} or (img.mode == "P" and "transparency" in img.info):
+                    background = Image.new("RGB", img.size, (255, 255, 255))
+                    rgba_img = img.convert("RGBA") if img.mode == "P" else img
+                    background.paste(rgba_img, mask=rgba_img.split()[-1] if rgba_img.mode in {"RGBA", "LA"} else None)
+                    rgb_img = background
+                elif img.mode != "RGB":
+                    rgb_img = img.convert("RGB")
+                else:
+                    rgb_img = img
+
+                # Save WebP
+                rgb_img.save(webp_path, "WEBP", quality=85, method=6)
+
+                # Save AVIF
+                rgb_img.save(avif_path, "AVIF", quality=85, speed=4)
+
+        except (OSError, ValueError) as e:
+            # Don't fail the download if conversion fails
+            self.stdout.write(self.style.WARNING(f"Failed to convert {image_path}: {e}"))