update for NetBox 4.0

netbox-community · Apr 4, 2024 · 49f4b7b · 49f4b7b
1 parent 013b70c
commit 49f4b7b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/netbox_napalm_plugin/__init__.py b/netbox_napalm_plugin/__init__.py
@@ -5,7 +5,7 @@
 __version__ = "0.1.6"
 
 
-from extras.plugins import PluginConfig
+from netbox.plugins import PluginConfig
 
 
 class NapalmPlatformConfig(PluginConfig):
@@ -21,8 +21,8 @@ class NapalmPlatformConfig(PluginConfig):
         'NAPALM_TIMEOUT': 30,
         'NAPALM_ARGS': {},
     }
-    min_version = '3.5.0-dev'
-    max_version = '3.7.99'
+    min_version = '4.0-beta1'
+    max_version = '4.0.99'
 
 
 config = NapalmPlatformConfig
diff --git a/netbox_napalm_plugin/api/views.py b/netbox_napalm_plugin/api/views.py
@@ -1,6 +1,6 @@
 from dcim.models import Device
 from django.shortcuts import get_object_or_404, redirect, render
-from extras.plugins.utils import get_plugin_config
+from netbox.plugins.utils import get_plugin_config
 from netbox.api.exceptions import ServiceUnavailable
 from netbox.api.pagination import StripCountAnnotationsPaginator
 from netbox.api.viewsets import NetBoxModelViewSet

diff --git a/netbox_napalm_plugin/navigation.py b/netbox_napalm_plugin/navigation.py
@@ -1,12 +1,10 @@
-from extras.plugins import PluginMenuButton, PluginMenuItem
-from utilities.choices import ButtonColorChoices
+from netbox.plugins import PluginMenuButton, PluginMenuItem
 
 plugin_buttons = [
     PluginMenuButton(
         link="plugins:netbox_napalm_plugin:napalmplatformconfig_add",
         title="Add",
         icon_class="mdi mdi-plus-thick",
-        color=ButtonColorChoices.GREEN,
     )
 ]