From 9f67da00d18f57f535042cabcde4718277fc8629 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Thu, 6 Apr 2017 14:12:30 -0400
Subject: [PATCH] Colored nodes in topology maps

---
 netbox/extras/models.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/netbox/extras/models.py b/netbox/extras/models.py
index 7d9c8d43d..5d90613b0 100644
--- a/netbox/extras/models.py
+++ b/netbox/extras/models.py
@@ -337,9 +337,10 @@ class TopologyMap(models.Model):
             # Add each device to the graph
             devices = []
             for query in device_set.split(';'):  # Split regexes on semicolons
-                devices += Device.objects.filter(name__regex=query)
+                devices += Device.objects.filter(name__regex=query).select_related('device_role')
             for d in devices:
-                subgraph.node(d.name)
+                fillcolor = '#{}'.format(d.device_role.color)
+                subgraph.node(d.name, style='filled', fillcolor=fillcolor)
 
             # Add an invisible connection to each successive device in a set to enforce horizontal order
             for j in range(0, len(devices) - 1):