From 00cf5642a0cb2a69968179fa7faf19c9e0c888a3 Mon Sep 17 00:00:00 2001 From: Derick Vigne Date: Sun, 13 Nov 2022 18:28:01 -0500 Subject: [PATCH] Added Colors to SVG for Front and Reaer Ports Fix for feature request 10904 thanks to @TheZackCodec --- netbox/dcim/models/device_components.py | 6 ++++++ netbox/dcim/svg/cables.py | 2 ++ 2 files changed, 8 insertions(+) diff --git a/netbox/dcim/models/device_components.py b/netbox/dcim/models/device_components.py index 8f1285901..2ca9af797 100644 --- a/netbox/dcim/models/device_components.py +++ b/netbox/dcim/models/device_components.py @@ -902,6 +902,9 @@ class FrontPort(ModularComponentModel, CabledObjectModel): ('rear_port', 'rear_port_position'), ) + def get_color(self): + return self.color + def get_absolute_url(self): return reverse('dcim:frontport', kwargs={'pk': self.pk}) @@ -948,6 +951,9 @@ class RearPort(ModularComponentModel, CabledObjectModel): ordering = ('device', '_name') unique_together = ('device', 'name') + def get_color(self): + return self.color + def get_absolute_url(self): return reverse('dcim:rearport', kwargs={'pk': self.pk}) diff --git a/netbox/dcim/svg/cables.py b/netbox/dcim/svg/cables.py index 9a847acc9..cc1306565 100644 --- a/netbox/dcim/svg/cables.py +++ b/netbox/dcim/svg/cables.py @@ -166,6 +166,8 @@ class CableTraceSVG: """ if hasattr(instance, 'parent_object'): # Termination + if(hasattr(instance, 'get_color') and instance.get_color()): + return instance.get_color() return 'f0f0f0' if hasattr(instance, 'device_role'): # Device