TheLovinator/ttvdrops
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enhance backfill_image_dimensions command to update image metadata; add tests for edge cases
All checks were successful
Deploy to Server / deploy (push) Successful in 9s
Details

Browse source
This commit is contained in:
Joakim Hellsén 2026-03-09 19:57:13 +01:00
commit 53765c3b27
Signed by: Joakim Hellsén
SSH key fingerprint: SHA256:/9h/CsExpFp+PRhsfA0xznFx2CGfTT5R/kpuFfUgEQk
2 changed files with 45 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

23
twitch/management/commands/backfill_image_dimensions.py
View file

@ -14,17 +14,22 @@ class Command(BaseCommand):
help = "Backfill image dimensions for existing cached images"
def handle(self, *args, **options) -> None: # noqa: ARG002
def handle(self, *args, **options) -> None: # noqa: ARG002, PLR0915
"""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:
needs_update = (
not game.box_art_size_bytes
or not game.box_art_mime_type
or not game.box_art_width
)
if game.box_art_file and needs_update:
try:
# Opening the file and saving triggers dimension calculation
game.box_art_file.open()
if not game.box_art_width:
game.box_art_file.open()
# populate size and mime if available
with contextlib.suppress(Exception):
@ -42,9 +47,15 @@ class Command(BaseCommand):
# 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:
needs_update: bool = (
not campaign.image_size_bytes
or not campaign.image_mime_type
or not campaign.image_width
)
if campaign.image_file and needs_update:
try:
campaign.image_file.open()
if not campaign.image_width:
campaign.image_file.open()
with contextlib.suppress(Exception):
campaign.image_size_bytes = campaign.image_file.size