Added self-healing to test_every_image_has_primary_variant

images/management/commands/test_integrity.py

@@ -54,6 +54,18 @@ class Command(BaseCommand):
 
         for image in Image.objects.all():
             if image.imagevariant_set.filter(is_primary_variant=True).count() == 0:
+                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