From d1a3ae4e148656645222282371c91030661f1458 Mon Sep 17 00:00:00 2001
From: Yaguang Tang <yaguang.tang@vexxhost.com>
Date: Fri, 5 Jul 2024 00:22:05 +0800
Subject: [PATCH] Fix  KeyError: 'status' for node group updates

---
 magnum_cluster_api/driver.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/magnum_cluster_api/driver.py b/magnum_cluster_api/driver.py
index 86518cef..5422deec 100644
--- a/magnum_cluster_api/driver.py
+++ b/magnum_cluster_api/driver.py
@@ -137,11 +137,11 @@ def update_cluster_control_plane_status(
         if generation > 1:
             action = "UPDATE"
 
-        ready = kcp.obj["status"].get("ready", False)
-        failure_message = kcp.obj["status"].get("failureMessage")
+        ready = kcp.obj.get("status", {}).get("ready", False)
+        failure_message = kcp.obj.get("status", {}).get("failureMessage")
 
-        updated_replicas = kcp.obj["status"].get("updatedReplicas")
-        replicas = kcp.obj["status"].get("replicas")
+        updated_replicas = kcp.obj.get("status", {}).get("updatedReplicas")
+        replicas = kcp.obj.get("status", {}).get("replicas")
 
         if updated_replicas != replicas:
             nodegroup.status = f"{action}_IN_PROGRESS"