Patch for Netbox 2.10.1 and above (#35)

Workaround changed cable trace behavior.

Fix #34

Author: iDebugAll

nextbox_ui_plugin/__init__.py

@@ -4,7 +4,7 @@ class NextBoxUIConfig(PluginConfig):
     name = 'nextbox_ui_plugin'
     verbose_name = 'NextBox UI'
     description = 'A topology visualization plugin for Netbox powered by NextUI Toolkit.'
-    version = '0.6.7'
+    version = '0.6.8'
     author = 'Igor Korotchenkov'
     author_email = 'iDebugAll@gmail.com'
     base_url = 'nextbox-ui'

nextbox_ui_plugin/views.py

@@ -5,9 +5,11 @@
 from dcim.models import Cable, Device, Interface
 from django.contrib.auth.mixins import PermissionRequiredMixin
 from django.conf import settings
+from packaging import version
 import json
 import re
 
+NETBOX_CURRENT_VERSION = version.parse(settings.VERSION)
 
 # Default NeXt UI icons
 SUPPORTED_ICONS = {
@@ -301,10 +303,18 @@ def get_site_topology(site_id):
             "srcIfName": if_shortname(link.termination_a.name),
             "tgtIfName": if_shortname(link.termination_b.name)
         })
+        if NETBOX_CURRENT_VERSION >= version.parse("2.10.1"):
+            # Version 2.10.1 introduces some changes in cable trace behavior.
+            if not isinstance(link.termination_a, Interface):
+                continue
         trace_result = link.termination_a.trace()
         if not trace_result:
             continue
-        cable_path, *ignore = trace_result
+        if NETBOX_CURRENT_VERSION >= version.parse("2.10.1"):
+            # Version 2.10.1 introduces some changes in cable trace behavior.
+            cable_path = trace_result
+        else:
+            cable_path, *ignore = trace_result
         # identify segmented cable paths between end-devices
         if len(cable_path) < 2:
             continue

setup.py

@@ -7,9 +7,9 @@
 
 setup(
     name='nextbox_ui_plugin',
-    version='0.6.7',
+    version='0.6.8',
     url='https://github.com/iDebugAll/nextbox-ui-plugin',
-    download_url='https://github.com/iDebugAll/nextbox-ui-plugin/archive/v0.6.7.tar.gz',
+    download_url='https://github.com/iDebugAll/nextbox-ui-plugin/archive/v0.6.8.tar.gz',
     description='A topology visualization plugin for Netbox powered by NextUI Toolkit.',
     long_description=long_description,
     long_description_content_type='text/markdown',