cmd-coreos-prune: Ensure action completion of each build

Refactor policy-cleanup updates to ensure action completion across all
arches. Modified  to track completed actions
per arch. This bug came after we refactored the code to to iterate by
build, arch, then action.

Author: gursewak1997

src/cmd-coreos-prune

@@ -125,6 +125,8 @@ def main():
     for build in reversed(builds):
         build_id = build["id"]
         build_date = parse_fcos_version_to_timestamp(build_id)
+        # Initialize action status for all arches
+        action_completion_status = {action: [] for action in policy[stream]}
 
         # For each build, iterate over arches first to minimize downloads of meta.json per arch
         for arch in build["arches"]:
@@ -188,20 +190,14 @@ def main():
                                         prune_container(tag, args.dry_run, container_repo, args.registry_auth_file)
                                 else:
                                     print(f"No container tags to prune for build {build_id}.")
+                    action_completion_status[action].append(arch)  # Track success for this action and arch
         # Update policy-cleanup after pruning actions for the architecture
         policy_cleanup = build.setdefault("policy-cleanup", {})
-        for action in policy[stream].keys():  # Only update actions specified in policy[stream]
-            match action:
-                case "cloud-uploads":
-                    if "cloud-uploads" not in policy_cleanup:
-                        policy_cleanup["cloud-uploads"] = True
-                case "images":
-                    if "images" not in policy_cleanup:
-                        policy_cleanup["images"] = True
+        for action, completed_arches in action_completion_status.items():
+            if set(completed_arches) == set(build["arches"]):  # Ensure all arches are completed
+                policy_cleanup[action] = True
+                if action == "images":
                     policy_cleanup["images-kept"] = images_to_keep
-                case "containers":
-                    if "containers" not in policy_cleanup:
-                        policy_cleanup["containers"] = True
 
     if pruned_build_ids:
         if "tombstone-builds" not in builds_json_data: