Update template_content.py

PieterL75 · May 28, 2024 · 0cea7ed · 0cea7ed
1 parent cf34ee6
commit 0cea7ed
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/netbox_ipcalculator/template_content.py b/netbox_ipcalculator/template_content.py
@@ -1,4 +1,8 @@
-from netbox.plugins import PluginTemplateExtension
+from netbox.settings import VERSION
+if VERSION.startswith("3."):
+    from extras.plugins import PluginTemplateExtension
+else:
+    from netbox.plugins import PluginTemplateExtension
 import json
 
 class IPCalcAggregate(PluginTemplateExtension):
@@ -28,4 +32,4 @@ def right_page(self):
         })        
         return output
 
-template_extensions = [IPCalcAggregate,IPCalcPrefix,IPCalcIPAddress]
+template_extensions = [IPCalcAggregate,IPCalcPrefix,IPCalcIPAddress]