From 93175888f088aa99c58b05e8deef806f0b640c24 Mon Sep 17 00:00:00 2001
From: Abhimanyu Saharan <desk.abhimanyu@gmail.com>
Date: Wed, 14 Jun 2023 13:36:30 +0530
Subject: [PATCH] add color to ChangeActionChoices #12828

---
 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 6fc14b965..63bdbf7db 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 b46d6a7bc..6d86e0dfe 100644
--- a/netbox/extras/models/staging.py
+++ b/netbox/extras/models/staging.py
@@ -112,3 +112,6 @@ class StagedChange(ChangeLoggedModel):
             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)