From 3fb52a13c250acf2882bd234c684ecd0df8a75dd Mon Sep 17 00:00:00 2001
From: "Joshua M. Boniface" <joshua@boniface.me>
Date: Tue, 20 Aug 2024 11:51:57 -0400
Subject: [PATCH] Add missing VM states from snapshots

---
 daemon-common/cluster.py | 2 ++
 daemon-common/common.py  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/daemon-common/cluster.py b/daemon-common/cluster.py
index 2b805b34..9a090e8b 100644
--- a/daemon-common/cluster.py
+++ b/daemon-common/cluster.py
@@ -1051,6 +1051,8 @@ def get_resource_metrics(zkhandler):
             "restart": 6,
             "stop": 7,
             "fail": 8,
+            "import": 9,
+            "restore": 10,
         }
         state = vm["state"]
         output_lines.append(
diff --git a/daemon-common/common.py b/daemon-common/common.py
index 5f478f4e..1fd565e2 100644
--- a/daemon-common/common.py
+++ b/daemon-common/common.py
@@ -82,6 +82,8 @@ vm_state_combinations = [
     "migrate",
     "unmigrate",
     "provision",
+    "import",
+    "restore",
 ]
 ceph_osd_state_combinations = [
     "up,in",