From 790cfd7b5bd371622fe689129f5efcb877e104f1 Mon Sep 17 00:00:00 2001 From: Jeremy Stretch Date: Thu, 9 Jan 2020 16:19:47 -0500 Subject: [PATCH] Fix CableTable status coloring --- netbox/dcim/models.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/netbox/dcim/models.py b/netbox/dcim/models.py index a7d865167..c8ee9fa8a 100644 --- a/netbox/dcim/models.py +++ b/netbox/dcim/models.py @@ -3256,6 +3256,11 @@ class Cable(ChangeLoggedModel): 'color', 'length', 'length_unit', ] + STATUS_CLASS_MAP = { + CableStatusChoices.STATUS_CONNECTED: 'success', + CableStatusChoices.STATUS_PLANNED: 'info', + } + class Meta: ordering = ['pk'] unique_together = ( @@ -3393,7 +3398,7 @@ class Cable(ChangeLoggedModel): ) def get_status_class(self): - return 'success' if self.status else 'info' + return self.STATUS_CLASS_MAP.get(self.status) def get_compatible_types(self): """