Add a default button color

netbox/extras/plugins/__init__.py

@@ -6,6 +6,7 @@ from django.template.loader import get_template
 from django.utils.module_loading import import_string
 
 from extras.registry import registry
+from utilities.choices import ButtonColorChoices
 
 
 # Initialize plugin registry stores
@@ -174,13 +175,17 @@ class PluginNavMenuButton:
     This class represents a button which is a part of the nav menu link item.
     Note that button colors should come from ButtonColorChoices
     """
-    def __init__(self, link, title, icon_class, color, permission=None):
+    color = ButtonColorChoices.DEFAULT
+
+    def __init__(self, link, title, icon_class, color=None, permission=None):
         self.link = link
         self.title = title
         self.icon_class = icon_class
-        self.color = color
         self.permission = permission
 
+        if color is not None:
+            self.color = color
+
 
 def register_menu_items(section_name, class_list):
     """