208 lines
11 KiB
Python
208 lines
11 KiB
Python
from django.shortcuts import get_object_or_404, render
|
|
from django.db.models import Q
|
|
from django.views.generic import View
|
|
from django.contrib.auth.mixins import PermissionRequiredMixin
|
|
from django.conf import settings
|
|
from django.http import QueryDict
|
|
from django.http import HttpResponseRedirect
|
|
|
|
from .forms import DeviceFilterForm
|
|
from .filters import DeviceFilterSet
|
|
|
|
import json
|
|
|
|
from dcim.models import Device, Cable, DeviceRole, DeviceType
|
|
from extras.models import Tag
|
|
|
|
def get_topology_data(queryset, hide_unconnected):
|
|
nodes = []
|
|
nodes_ids = []
|
|
edges = []
|
|
edge_ids = 0
|
|
cable_ids = []
|
|
circuit_ids = []
|
|
if not queryset:
|
|
return None
|
|
|
|
ignore_cable_type = settings.PLUGINS_CONFIG["netbox_topology_views"]["ignore_cable_type"]
|
|
|
|
device_ids = [d.id for d in queryset]
|
|
|
|
for qs_device in queryset:
|
|
device_has_connections = False
|
|
|
|
links_device = Cable.objects.filter(Q(_termination_a_device_id=qs_device.id) | Q(_termination_b_device_id=qs_device.id) )
|
|
for link_from in links_device:
|
|
if link_from.termination_a_type.name != "circuit termination" and link_from.termination_b_type.name != "circuit termination":
|
|
if link_from.termination_a_type.name not in ignore_cable_type and link_from.termination_b_type.name not in ignore_cable_type:
|
|
if link_from.id not in cable_ids:
|
|
if link_from.termination_a.device.id in device_ids and link_from.termination_b.device.id in device_ids:
|
|
device_has_connections = True
|
|
cable_ids.append(link_from.id)
|
|
edge_ids += 1
|
|
cable_a_dev_name = link_from.termination_a.device.name
|
|
if cable_a_dev_name is None:
|
|
cable_a_dev_name = "device A name unknown"
|
|
cable_a_name = link_from.termination_a.name
|
|
if cable_a_name is None:
|
|
cable_a_name = "cable A name unknown"
|
|
cable_b_dev_name = link_from.termination_b.device.name
|
|
if cable_b_dev_name is None:
|
|
cable_b_dev_name = "device B name unknown"
|
|
cable_b_name = link_from.termination_b.name
|
|
if cable_b_name is None:
|
|
cable_b_name = "cable B name unknown"
|
|
|
|
edge = {}
|
|
edge["id"] = edge_ids
|
|
edge["from"] = link_from.termination_a.device.id
|
|
edge["to"] = link_from.termination_b.device.id
|
|
edge["title"] = "Cable between <br> " + cable_a_dev_name + " [" + cable_a_name + "]<br>" + cable_b_dev_name + " [" + cable_b_name + "]"
|
|
if link_from.color != "":
|
|
edge["color"] = "#" + link_from.color
|
|
edges.append(edge)
|
|
else:
|
|
if link_from.termination_a.device.id in device_ids and link_from.termination_b.device.id in device_ids:
|
|
device_has_connections = True
|
|
else:
|
|
if settings.PLUGINS_CONFIG["netbox_topology_views"]["enable_circuit_terminations"]:
|
|
if link_from.termination_a.circuit.id not in circuit_ids:
|
|
circuit_ids.append(link_from.termination_a.circuit.id)
|
|
edge_ids += 1
|
|
|
|
cable_b_dev_name = link_from.termination_b.device.name
|
|
if cable_b_dev_name is None:
|
|
cable_b_dev_name = "device B name unknown"
|
|
cable_b_name = link_from.termination_b.name
|
|
if cable_b_name is None:
|
|
cable_b_name = "cable B name unknown"
|
|
|
|
edge = {}
|
|
edge["id"] = edge_ids
|
|
edge["to"] = link_from.termination_b.device.id
|
|
edge["dashes"] = True
|
|
title = ""
|
|
|
|
title += "Circuit provider: " + link_from.termination_a.circuit.provider.name + "<br>"
|
|
title += "Termination between <br>"
|
|
title += cable_b_dev_name + " [" + cable_b_name + "]<br>"
|
|
|
|
if link_from.termination_a.circuit.termination_a is not None and link_from.termination_a.circuit.termination_a.cable is not None and link_from.termination_a.circuit.termination_a.cable.id != link_from.id and link_from.termination_a.circuit.termination_a.cable.termination_b is not None and link_from.termination_a.circuit.termination_a.cable.termination_b.device is not None:
|
|
edge["from"] = link_from.termination_a.circuit.termination_a.cable.termination_b.device.id
|
|
|
|
cable_a_dev_name = link_from.termination_a.circuit.termination_a.cable.termination_b.device.name
|
|
if cable_a_dev_name is None:
|
|
cable_a_dev_name = "device B name unknown"
|
|
cable_b_name = link_from.termination_a.circuit.termination_a.cable.termination_b.name
|
|
if cable_a_name is None:
|
|
cable_a_name = "cable B name unknown"
|
|
title += cable_a_dev_name + " [" + cable_a_name + "]<br>"
|
|
edge["title"] = title
|
|
edges.append(edge)
|
|
|
|
if link_from.termination_a.circuit.termination_z is not None and link_from.termination_a.circuit.termination_z.cable is not None and link_from.termination_a.circuit.termination_z.cable.id != link_from.id and link_from.termination_a.circuit.termination_z.cable.termination_b is not None and link_from.termination_a.circuit.termination_z.cable.termination_b.device is not None:
|
|
edge["from"] = link_from.termination_a.circuit.termination_z.cable.termination_b.device.id
|
|
|
|
cable_a_dev_name = link_from.termination_a.circuit.termination_z.cable.termination_b.device.name
|
|
if cable_a_dev_name is None:
|
|
cable_a_dev_name = "device B name unknown"
|
|
cable_a_name = link_from.termination_a.circuit.termination_z.cable.termination_b.name
|
|
if cable_a_name is None:
|
|
cable_a_name = "cable B name unknown"
|
|
title += cable_a_dev_name + " [" + cable_a_name + "]<br>"
|
|
edge["title"] = title
|
|
edges.append(edge)
|
|
|
|
if qs_device.id not in nodes_ids:
|
|
if hide_unconnected == None or (hide_unconnected is True and device_has_connections is True):
|
|
nodes_ids.append(qs_device.id)
|
|
|
|
dev_name = qs_device.name
|
|
if dev_name is None:
|
|
dev_name = "device name unknown"
|
|
|
|
node_content = ""
|
|
|
|
if qs_device.device_type is not None:
|
|
node_content += "<tr><th>Type: </th><td>" + qs_device.device_type.model + "</td></tr>"
|
|
if qs_device.device_role.name is not None:
|
|
node_content += "<tr><th>Role: </th><td>" + qs_device.device_role.name + "</td></tr>"
|
|
if qs_device.serial != "":
|
|
node_content += "<tr><th>Serial: </th><td>" + qs_device.serial + "</td></tr>"
|
|
if qs_device.primary_ip is not None:
|
|
node_content += "<tr><th>IP Address: </th><td>" + str(qs_device.primary_ip.address) + "</td></tr>"
|
|
|
|
dev_title = "<table> %s </table>" % (node_content)
|
|
|
|
node = {}
|
|
node["id"] = qs_device.id
|
|
node["name"] = dev_name
|
|
node["label"] = dev_name
|
|
node["title"] = dev_title
|
|
node["shape"] = 'image'
|
|
if qs_device.device_role.slug in settings.PLUGINS_CONFIG["netbox_topology_views"]["device_img"]:
|
|
node["image"] = '../../static/netbox_topology_views/img/' + qs_device.device_role.slug + ".png"
|
|
else:
|
|
node["image"] = "../../static/netbox_topology_views/img/role-unknown.png"
|
|
|
|
if qs_device.device_role.color != "":
|
|
node["color.border"] = "#" + qs_device.device_role.color
|
|
|
|
if "coordinates" in qs_device.custom_field_data:
|
|
if qs_device.custom_field_data["coordinates"] is not None:
|
|
if ';' in qs_device.custom_field_data["coordinates"]:
|
|
cords = qs_device.custom_field_data["coordinates"].split(";")
|
|
node["x"] = int(cords[0])
|
|
node["y"] = int(cords[1])
|
|
node["physics"] = False
|
|
nodes.append(node)
|
|
|
|
results = {}
|
|
results["nodes"] = nodes
|
|
results["edges"] = edges
|
|
return results
|
|
|
|
|
|
class TopologyHomeView(PermissionRequiredMixin, View):
|
|
permission_required = ('dcim.view_site', 'dcim.view_device')
|
|
|
|
"""
|
|
Show the home page
|
|
"""
|
|
def get(self, request):
|
|
self.filterset = DeviceFilterSet
|
|
self.queryset = Device.objects.all()
|
|
self.queryset = self.filterset(request.GET, self.queryset).qs
|
|
topo_data = None
|
|
|
|
if request.GET:
|
|
hide_unconnected = None
|
|
if 'hide_unconnected' in request.GET:
|
|
if request.GET["hide_unconnected"] == "on" :
|
|
hide_unconnected = True
|
|
|
|
if 'draw_init' in request.GET:
|
|
if request.GET["draw_init"].lower() == 'true':
|
|
topo_data = get_topology_data(self.queryset, hide_unconnected)
|
|
else:
|
|
topo_data = get_topology_data(self.queryset, hide_unconnected)
|
|
else:
|
|
preselected_device_roles = settings.PLUGINS_CONFIG["netbox_topology_views"]["preselected_device_roles"]
|
|
preselected_tags = settings.PLUGINS_CONFIG["netbox_topology_views"]["preselected_tags"]
|
|
|
|
q_device_role_id = DeviceRole.objects.filter(name__in=preselected_device_roles).values_list('id', flat=True)
|
|
q_tags = Tag.objects.filter(name__in=preselected_tags).values_list('name', flat=True)
|
|
|
|
q = QueryDict(mutable=True)
|
|
q.setlist('device_role_id', list(q_device_role_id))
|
|
q.setlist('tag', list(q_tags))
|
|
q['draw_init'] = settings.PLUGINS_CONFIG["netbox_topology_views"]["draw_default_layout"]
|
|
query_string = q.urlencode()
|
|
return HttpResponseRedirect(request.path + "?" + query_string)
|
|
|
|
return render(request, 'netbox_topology_views/index.html' , {
|
|
'filter_form': DeviceFilterForm(request.GET, label_suffix=''),
|
|
'topology_data': json.dumps(topo_data)
|
|
}
|
|
)
|