Explorar o código

add color to ChangeActionChoices #12828

Abhimanyu Saharan %!s(int64=2) %!d(string=hai) anos
pai
achega
93175888f0
Modificáronse 2 ficheiros con 6 adicións e 3 borrados
  1. 3 3
      netbox/extras/choices.py
  2. 3 0
      netbox/extras/models/staging.py

+ 3 - 3
netbox/extras/choices.py

@@ -210,7 +210,7 @@ class ChangeActionChoices(ChoiceSet):
     ACTION_DELETE = 'delete'
     ACTION_DELETE = 'delete'
 
 
     CHOICES = (
     CHOICES = (
-        (ACTION_CREATE, 'Create'),
-        (ACTION_UPDATE, 'Update'),
-        (ACTION_DELETE, 'Delete'),
+        (ACTION_CREATE, 'Create', 'green'),
+        (ACTION_UPDATE, 'Update', 'blue'),
+        (ACTION_DELETE, 'Delete', 'red'),
     )
     )

+ 3 - 0
netbox/extras/models/staging.py

@@ -112,3 +112,6 @@ class StagedChange(ChangeLoggedModel):
             instance = self.model.objects.get(pk=self.object_id)
             instance = self.model.objects.get(pk=self.object_id)
             logger.info(f'Deleting {self.model._meta.verbose_name} {instance}')
             logger.info(f'Deleting {self.model._meta.verbose_name} {instance}')
             instance.delete()
             instance.delete()
+
+    def get_action_color(self):
+        return ChangeActionChoices.colors.get(self.action)