Closes netbox-community#12828: Add color mapping to ChangeActionChoices

ingvaldlorentzen · Jun 7, 2023 · a678db3 · a678db3
1 parent 4a88d5e
commit a678db3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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)