Update handle-image-data command to report truly broken manifests

Author: gerrod3

pulp_container/app/management/commands/container-handle-image-data.py

@@ -1,4 +1,3 @@
-from contextlib import suppress
 from gettext import gettext as _
 from json.decoder import JSONDecodeError
 
@@ -34,6 +33,7 @@ class Command(BaseCommand):
 
     def handle(self, *args, **options):
         manifests_updated_count = 0
+        self._broken_manifests = []
 
         manifests_v1 = Manifest.objects.filter(
             Q(media_type=MEDIA_TYPE.MANIFEST_V1),
@@ -68,6 +68,13 @@ def handle(self, *args, **options):
             self.style.SUCCESS("Successfully updated %d manifests." % manifests_updated_count)
         )
 
+        if self._broken_manifests:
+            self.stdout.write(
+                self.style.WARNING("Found %d broken manifests. PKs:" % len(self._broken_manifests))
+            )
+            for manifest in self._broken_manifests:
+                self.stdout.write(self.style.WARNING("  %s" % manifest.pk))
+
         if settings.CACHE_ENABLED and manifests_updated_count != 0:
             base_paths = ContainerDistribution.objects.values_list("base_path", flat=True)
             if base_paths:
@@ -91,11 +98,13 @@ def update_manifests(self, manifests_qs):
         ]
 
         for manifest in manifests_qs.iterator():
-            # suppress non-existing/already migrated artifacts and corrupted JSON files
-            with suppress(ObjectDoesNotExist, JSONDecodeError):
+            try:
                 needs_update = self.init_manifest(manifest)
-                if needs_update:
-                    manifests_to_update.append(manifest)
+            except (ObjectDoesNotExist, JSONDecodeError):
+                self._broken_manifests.append(manifest)
+                continue
+            if needs_update:
+                manifests_to_update.append(manifest)
 
             if len(manifests_to_update) > 1000:
                 manifests_qs.model.objects.bulk_update(