From 6dd8530ad8fe209eff74e0e2d6a7a553908ac235 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@netboxlabs.com>
Date: Tue, 18 Feb 2025 16:43:14 -0500
Subject: [PATCH] Update circuits enums

---
 netbox/circuits/graphql/enums.py | 86 ++++----------------------------
 1 file changed, 10 insertions(+), 76 deletions(-)

diff --git a/netbox/circuits/graphql/enums.py b/netbox/circuits/graphql/enums.py
index 9fd3b8b09..609e4435b 100644
--- a/netbox/circuits/graphql/enums.py
+++ b/netbox/circuits/graphql/enums.py
@@ -1,86 +1,20 @@
-from enum import Enum
 import strawberry
 
-__all__ = [
+from circuits.choices import *
+
+__all__ = (
     'CircuitStatusEnum',
     'CircuitCommitRateEnum',
     'CircuitTerminationSideEnum',
     'CircuitTerminationPortSpeedEnum',
     'CircuitPriorityEnum',
     'VirtualCircuitTerminationRoleEnum',
-]
-
-#
-# Circuits
-#
+)
 
 
-@strawberry.enum
-class CircuitStatusEnum(Enum):
-    STATUS_DEPROVISIONING = 'deprovisioning'
-    STATUS_ACTIVE = 'active'
-    STATUS_PLANNED = 'planned'
-    STATUS_PROVISIONING = 'provisioning'
-    STATUS_OFFLINE = 'offline'
-    STATUS_DECOMMISSIONED = 'decommissioned'
-
-
-@strawberry.enum
-class CircuitCommitRateEnum(Enum):
-    TEN_MBPS = 10000
-    HUNDRED_MBPS = 100000
-    ONE_GBPS = 1000000
-    TEN_GBPS = 10000000
-    TWENTY_FIVE_GBPS = 25000000
-    FORTY_GBPS = 40000000
-    HUNDRED_GBPS = 100000000
-    TWO_HUNDRED_GBPS = 200000000
-    FOUR_HUNDRED_GBPS = 400000000
-    T1 = 1544
-    E1 = 2048
-
-
-#
-# CircuitTerminations
-#
-
-
-@strawberry.enum
-class CircuitTerminationSideEnum(Enum):
-    SIDE_A = 'A'
-    SIDE_Z = 'Z'
-
-
-@strawberry.enum
-class CircuitTerminationPortSpeedEnum(Enum):
-    TEN_MBPS = 10000
-    HUNDRED_MBPS = 100000
-    ONE_GBPS = 1000000
-    TEN_GBPS = 10000000
-    TWENTY_FIVE_GBPS = 25000000
-    FORTY_GBPS = 40000000
-    HUNDRED_GBPS = 100000000
-    TWO_HUNDRED_GBPS = 200000000
-    FOUR_HUNDRED_GBPS = 400000000
-    T1 = 1544
-    E1 = 2048
-
-
-@strawberry.enum
-class CircuitPriorityEnum(Enum):
-    PRIORITY_PRIMARY = 'primary'
-    PRIORITY_SECONDARY = 'secondary'
-    PRIORITY_TERTIARY = 'tertiary'
-    PRIORITY_INACTIVE = 'inactive'
-
-
-#
-# Virtual circuits
-#
-
-
-@strawberry.enum
-class VirtualCircuitTerminationRoleEnum(Enum):
-    ROLE_PEER = 'peer'
-    ROLE_HUB = 'hub'
-    ROLE_SPOKE = 'spoke'
+CircuitCommitRateEnum = strawberry.enum(CircuitCommitRateChoices.as_enum())
+CircuitPriorityEnum = strawberry.enum(CircuitPriorityChoices.as_enum())
+CircuitStatusEnum = strawberry.enum(CircuitStatusChoices.as_enum())
+CircuitTerminationSideEnum = strawberry.enum(CircuitTerminationSideChoices.as_enum())
+CircuitTerminationPortSpeedEnum = strawberry.enum(CircuitTerminationPortSpeedChoices.as_enum())
+VirtualCircuitTerminationRoleEnum = strawberry.enum(VirtualCircuitTerminationRoleChoices.as_enum())