From 30c913c97af2cabbf19343df6bb2506400ec2059 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@netboxlabs.com>
Date: Wed, 19 Feb 2025 09:27:11 -0500
Subject: [PATCH] Update virtualization enums

---
 netbox/virtualization/graphql/enums.py | 36 +++++---------------------
 1 file changed, 7 insertions(+), 29 deletions(-)

diff --git a/netbox/virtualization/graphql/enums.py b/netbox/virtualization/graphql/enums.py
index 081669f1f..5b1c54e0c 100644
--- a/netbox/virtualization/graphql/enums.py
+++ b/netbox/virtualization/graphql/enums.py
@@ -1,33 +1,11 @@
-from enum import Enum
 import strawberry
 
-__all__ = ['ClusterStatusEnum', 'VirtualMachineStatusEnum']
+from virtualization.choices import *
 
+__all__ = (
+    'ClusterStatusEnum',
+    'VirtualMachineStatusEnum',
+)
 
-#
-# Clusters
-#
-
-@strawberry.enum
-class ClusterStatusEnum(Enum):
-
-    STATUS_PLANNED = 'planned'
-    STATUS_STAGING = 'staging'
-    STATUS_ACTIVE = 'active'
-    STATUS_DECOMMISSIONING = 'decommissioning'
-    STATUS_OFFLINE = 'offline'
-
-
-#
-# VirtualMachines
-#
-
-@strawberry.enum
-class VirtualMachineStatusEnum(Enum):
-
-    STATUS_OFFLINE = 'offline'
-    STATUS_ACTIVE = 'active'
-    STATUS_PLANNED = 'planned'
-    STATUS_STAGED = 'staged'
-    STATUS_FAILED = 'failed'
-    STATUS_DECOMMISSIONING = 'decommissioning'
+ClusterStatusEnum = strawberry.enum(ClusterStatusChoices.as_enum())
+VirtualMachineStatusEnum = strawberry.enum(VirtualMachineStatusChoices.as_enum())