From b69d55636226f069e3d5bdc7f3526d068352ed1a Mon Sep 17 00:00:00 2001
From: Remilia Da Costa Faro <remilia@remilia.ch>
Date: Sun, 4 May 2025 23:18:24 +0200
Subject: [PATCH] Added self-healing to test_every_image_has_primary_variant

---
 images/management/commands/test_integrity.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/images/management/commands/test_integrity.py b/images/management/commands/test_integrity.py
index c41d8af..77a5f8e 100644
--- a/images/management/commands/test_integrity.py
+++ b/images/management/commands/test_integrity.py
@@ -54,7 +54,19 @@ class Command(BaseCommand):
 
         for image in Image.objects.all():
             if image.imagevariant_set.filter(is_primary_variant=True).count() == 0:
-                images_with_problems.append(image)
+                image_variants = image.imagevariant_set.filter(
+                    is_full_size=True,
+                    file_type__in=["jpg", "png"],
+                    gaussian_blur=0,
+                    brightness=1,
+                )
+                if image_variants.count() == 1:
+                    image_variants.update(is_primary_variant=True)
+                    print(
+                        f"SELF-HEALED: Image {image.id} had no primary variant, we assigned one"
+                    )
+                else:
+                    images_with_problems.append(image)
 
         return images_with_problems