Add image dimensions to models, and add backfill command

twitch/management/commands/backfill_image_dimensions.py

@@ -0,0 +1,73 @@
+"""Management command to backfill image dimensions for existing cached images."""
+
+from __future__ import annotations
+
+from django.core.management.base import BaseCommand
+
+from twitch.models import DropBenefit
+from twitch.models import DropCampaign
+from twitch.models import Game
+from twitch.models import RewardCampaign
+
+
+class Command(BaseCommand):
+    """Backfill image width and height fields for existing cached images."""
+
+    help = "Backfill image dimensions for existing cached images"
+
+    def handle(self, *args, **options) -> None:  # noqa: ARG002
+        """Execute the command."""
+        total_updated = 0
+
+        # Update Game box art
+        self.stdout.write("Processing Game box_art_file...")
+        for game in Game.objects.exclude(box_art_file=""):
+            if game.box_art_file and not game.box_art_width:
+                try:
+                    # Opening the file and saving triggers dimension calculation
+                    game.box_art_file.open()
+                    game.save()
+                    total_updated += 1
+                    self.stdout.write(self.style.SUCCESS(f"  Updated {game}"))
+                except (OSError, ValueError, AttributeError) as exc:
+                    self.stdout.write(self.style.ERROR(f"  Failed {game}: {exc}"))
+
+        # Update DropCampaign images
+        self.stdout.write("Processing DropCampaign image_file...")
+        for campaign in DropCampaign.objects.exclude(image_file=""):
+            if campaign.image_file and not campaign.image_width:
+                try:
+                    campaign.image_file.open()
+                    campaign.save()
+                    total_updated += 1
+                    self.stdout.write(self.style.SUCCESS(f"  Updated {campaign}"))
+                except (OSError, ValueError, AttributeError) as exc:
+                    self.stdout.write(self.style.ERROR(f"  Failed {campaign}: {exc}"))
+
+        # Update DropBenefit images
+        self.stdout.write("Processing DropBenefit image_file...")
+        for benefit in DropBenefit.objects.exclude(image_file=""):
+            if benefit.image_file and not benefit.image_width:
+                try:
+                    benefit.image_file.open()
+                    benefit.save()
+                    total_updated += 1
+                    self.stdout.write(self.style.SUCCESS(f"  Updated {benefit}"))
+                except (OSError, ValueError, AttributeError) as exc:
+                    self.stdout.write(self.style.ERROR(f"  Failed {benefit}: {exc}"))
+
+        # Update RewardCampaign images
+        self.stdout.write("Processing RewardCampaign image_file...")
+        for reward in RewardCampaign.objects.exclude(image_file=""):
+            if reward.image_file and not reward.image_width:
+                try:
+                    reward.image_file.open()
+                    reward.save()
+                    total_updated += 1
+                    self.stdout.write(self.style.SUCCESS(f"  Updated {reward}"))
+                except (OSError, ValueError, AttributeError) as exc:
+                    self.stdout.write(self.style.ERROR(f"  Failed {reward}: {exc}"))
+
+        self.stdout.write(
+            self.style.SUCCESS(f"\nBackfill complete! Updated {total_updated} images."),
+        )