From a678db3cc43467c110f98e340a7cae78758be658 Mon Sep 17 00:00:00 2001
From: Ingvald Lorentzen <ingvald.lorentzen@intility.no>
Date: Wed, 7 Jun 2023 15:21:56 +0200
Subject: [PATCH] Closes #12828: Add color mapping to ChangeActionChoices

---
 netbox/extras/choices.py        | 6 +++---
 netbox/extras/models/staging.py | 3 +++
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/netbox/extras/choices.py b/netbox/extras/choices.py
index 6fc14b9658..63bdbf7dbe 100644
--- a/netbox/extras/choices.py
+++ b/netbox/extras/choices.py
@@ -210,7 +210,7 @@ class ChangeActionChoices(ChoiceSet):
     ACTION_DELETE = 'delete'
 
     CHOICES = (
-        (ACTION_CREATE, 'Create'),
-        (ACTION_UPDATE, 'Update'),
-        (ACTION_DELETE, 'Delete'),
+        (ACTION_CREATE, 'Create', 'green'),
+        (ACTION_UPDATE, 'Update', 'blue'),
+        (ACTION_DELETE, 'Delete', 'red'),
     )
diff --git a/netbox/extras/models/staging.py b/netbox/extras/models/staging.py
index b46d6a7bcc..6d86e0dfea 100644
--- a/netbox/extras/models/staging.py
+++ b/netbox/extras/models/staging.py
@@ -112,3 +112,6 @@ def apply(self):
             instance = self.model.objects.get(pk=self.object_id)
             logger.info(f'Deleting {self.model._meta.verbose_name} {instance}')
             instance.delete()
+
+    def get_action_color(self):
+        return ChangeActionChoices.colors.get(self.action)