Add GenericRelation to originating cable paths on PathEndpoint

netbox/dcim/models/device_components.py

@@ -252,15 +252,19 @@ class CableTermination(models.Model):
         return endpoints
 
 
-class PathEndpoint:
+class PathEndpoint(models.Model):
+    """
+    Any object which may serve as either endpoint of a CablePath.
+    """
+    paths = GenericRelation(
+        to='dcim.CablePath',
+        content_type_field='origin_type',
+        object_id_field='origin_id',
+        related_query_name='%(class)s'
+    )
 
-    def get_connections(self):
-        from dcim.models import CablePath
-        return CablePath.objects.filter(
-            origin_type=ContentType.objects.get_for_model(self),
-            origin_id=self.pk,
-            destination_id__isnull=False
-        )
+    class Meta:
+        abstract = True
 
 
 #

netbox/dcim/views.py

@@ -32,7 +32,7 @@ from . import filters, forms, tables
 from .choices import DeviceFaceChoices
 from .constants import NONCONNECTABLE_IFACE_TYPES
 from .models import (
-    Cable, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceBay,
+    Cable, CablePath, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceBay,
     DeviceBayTemplate, DeviceRole, DeviceType, FrontPort, FrontPortTemplate, Interface, InterfaceTemplate,
     InventoryItem, Manufacturer, Platform, PowerFeed, PowerOutlet, PowerOutletTemplate, PowerPanel, PowerPort,
     PowerPortTemplate, Rack, RackGroup, RackReservation, RackRole, RearPort, RearPortTemplate, Region, Site,
@@ -1018,32 +1018,36 @@ class DeviceView(ObjectView):
 
         # Console ports
         consoleports = ConsolePort.objects.restrict(request.user, 'view').filter(device=device).prefetch_related(
-            'connected_endpoint__device', 'cable',
+            Prefetch('paths', queryset=CablePath.objects.filter(destination_id__isnull=False)),
+            'cable',
         )
 
         # Console server ports
         consoleserverports = ConsoleServerPort.objects.restrict(request.user, 'view').filter(
             device=device
         ).prefetch_related(
-            'connected_endpoint__device', 'cable',
+            Prefetch('paths', queryset=CablePath.objects.filter(destination_id__isnull=False)),
+            'cable',
         )
 
         # Power ports
         powerports = PowerPort.objects.restrict(request.user, 'view').filter(device=device).prefetch_related(
-            '_connected_poweroutlet__device', 'cable',
+            Prefetch('paths', queryset=CablePath.objects.filter(destination_id__isnull=False)),
+            'cable',
         )
 
         # Power outlets
         poweroutlets = PowerOutlet.objects.restrict(request.user, 'view').filter(device=device).prefetch_related(
-            'connected_endpoint__device', 'cable', 'power_port',
+            Prefetch('paths', queryset=CablePath.objects.filter(destination_id__isnull=False)),
+            'cable', 'power_port',
         )
 
         # Interfaces
         interfaces = device.vc_interfaces.restrict(request.user, 'view').prefetch_related(
+            Prefetch('paths', queryset=CablePath.objects.filter(destination_id__isnull=False)),
             Prefetch('ip_addresses', queryset=IPAddress.objects.restrict(request.user)),
             Prefetch('member_interfaces', queryset=Interface.objects.restrict(request.user)),
-            'lag', '_connected_interface__device', '_connected_circuittermination__circuit', 'cable',
-            'cable__termination_a', 'cable__termination_b', 'tags'
+            'lag', 'cable', 'tags',
         )
 
         # Front ports

netbox/templates/dcim/inc/consoleport.html

@@ -36,18 +36,7 @@
     </td>
 
     {# Connection #}
-    {% if cp.connected_endpoint %}
-        <td>
-            <a href="{% url 'dcim:device' pk=cp.connected_endpoint.device.pk %}">{{ cp.connected_endpoint.device }}</a>
-        </td>
-        <td>
-            {{ cp.connected_endpoint }}
-        </td>
-    {% else %}
-        <td colspan="2">
-            <span class="text-muted">Not connected</span>
-        </td>
-    {% endif %}
+    {% include 'dcim/inc/endpoint_connection.html' with paths=cp.paths.all %}
 
     {# Actions #}
     <td class="text-right noprint">

netbox/templates/dcim/inc/consoleserverport.html

@@ -38,18 +38,7 @@
     </td>
 
     {# Connection #}
-    {% if csp.connected_endpoint %}
-        <td>
-            <a href="{% url 'dcim:device' pk=csp.connected_endpoint.device.pk %}">{{ csp.connected_endpoint.device }}</a>
-        </td>
-        <td>
-            {{ csp.connected_endpoint }}
-        </td>
-    {% else %}
-        <td colspan="2">
-            <span class="text-muted">Not connected</span>
-        </td>
-    {% endif %}
+    {% include 'dcim/inc/endpoint_connection.html' with paths=csp.paths.all %}
 
     {# Actions #}
     <td class="text-right noprint">

netbox/templates/dcim/inc/endpoint_connection.html

@@ -0,0 +1,10 @@
+{% if paths|length > 1 %}
+    <td colspan="2">Multiple connections</td>
+{% elif paths %}
+    {% with endpoint=paths.0.destination %}
+        <td><a href="{{ endpoint.parent.get_absolute_url }}">{{ endpoint.parent }}</a></td>
+        <td><a href="{{ endpoint.get_absolute_url }}">{{ endpoint }}</a></td>
+    {% endwith %}
+{% else %}
+    <td colspan="2" class="text-muted">Not connected</td>
+{% endif %}

netbox/templates/dcim/inc/interface.html

@@ -76,18 +76,7 @@
     {% elif iface.is_wireless %}
         <td colspan="2" class="text-muted">Wireless interface</td>
     {% else %}
-        {% with path_count=iface.get_connections.count %}
-            {% if path_count > 1 %}
-                <td colspan="2">Multiple connections</td>
-            {% elif path_count %}
-                {% with endpoint=iface.get_connections.first.destination %}
-                    <td><a href="{{ endpoint.parent.get_absolute_url }}">{{ endpoint.parent }}</a></td>
-                    <td><a href="{{ endpoint.get_absolute_url }}">{{ endpoint }}</a></td>
-                {% endwith %}
-            {% else %}
-                <td colspan="2" class="text-muted">Not connected</td>
-            {% endif %}
-        {% endwith %}
+        {% include 'dcim/inc/endpoint_connection.html' with paths=iface.paths.all %}
     {% endif %}
 
     {# Buttons #}

netbox/templates/dcim/inc/poweroutlet.html

@@ -49,27 +49,20 @@
     </td>
 
     {# Connection #}
-    {% if po.connected_endpoint %}
-        {% with pp=po.connected_endpoint %}
-            <td>
-                <a href="{% url 'dcim:device' pk=pp.device.pk %}">{{ pp.device }}</a>
-            </td>
-            <td>
-                {{ pp }}
-            </td>
-            <td>
-                {% if pp.allocated_draw %}
-                    {{ pp.allocated_draw }}W{% if pp.maximum_draw %} ({{ pp.maximum_draw }}W max){% endif %}
-                {% elif pp.maximum_draw %}
-                    {{ pp.maximum_draw }}W
-                {% endif %}
-            </td>
-        {% endwith %}
-    {% else %}
-        <td colspan="3">
-            <span class="text-muted">Not connected</span>
+    {% with paths=po.paths.all %}
+        {% include 'dcim/inc/endpoint_connection.html' %}
+        <td>
+            {% if paths|length == 1 %}
+                {% with pp=paths.0.destination %}
+                    {% if pp.allocated_draw %}
+                        {{ pp.allocated_draw }}W{% if pp.maximum_draw %} ({{ pp.maximum_draw }}W max){% endif %}
+                    {% elif pp.maximum_draw %}
+                        {{ pp.maximum_draw }}W
+                    {% endif %}
+                {% endwith %}
+            {% endif %}
         </td>
-    {% endif %}
+    {% endwith %}
 
     {# Actions #}
     <td class="text-right noprint">

netbox/templates/dcim/inc/powerport.html

@@ -45,22 +45,7 @@
     </td>
 
     {# Connection #}
-    {% if pp.connected_endpoint.device %}
-        <td>
-            <a href="{% url 'dcim:device' pk=pp.connected_endpoint.device.pk %}">{{ pp.connected_endpoint.device }}</a>
-        </td>
-        <td>
-            {{ pp.connected_endpoint }}
-        </td>
-    {% elif pp.connected_endpoint %}
-        <td colspan="2">
-            <a href="{{ pp.connected_endpoint.get_absolute_url }}">{{ pp.connected_endpoint }}</a>
-        </td>
-    {% else %}
-        <td colspan="2">
-            <span class="text-muted">Not connected</span>
-        </td>
-    {% endif %}
+    {% include 'dcim/inc/endpoint_connection.html' with paths=pp.paths.all %}
 
     {# Actions #}
     <td class="text-right noprint">