import collections import importlib import inspect from django.core.exceptions import ImproperlyConfigured from django.template.loader import get_template from extras.utils import registry from .signals import register_detail_page_content_classes, register_nav_menu_link_classes # # Template content injection # class PluginTemplateContent: """ This class is used to register plugin content to be injected into core NetBox templates. It contains methods that are overriden by plugin authors to return template content. The `model` attribute on the class defines the which model detail page this class renders content for. It should be set as a string in the form '.'. """ model = None def __init__(self, obj, context): self.obj = obj self.context = context def render(self, template, extra_context=None): """ Convenience menthod for rendering the provided template name. The detail page object is automatically passed into the template context as `obj` and the origional detail page's context is available as `obj_context`. An additional context dictionary may be passed as `extra_context`. """ context = { 'obj': self.obj, 'obj_context': self.context } if isinstance(extra_context, dict): context.update(extra_context) return get_template(template).render(context) def left_page(self): """ Content that will be rendered on the left of the detail page view. Content should be returned as an HTML string. Note that content does not need to be marked as safe because this is automatically handled. """ raise NotImplementedError def right_page(self): """ Content that will be rendered on the right of the detail page view. Content should be returned as an HTML string. Note that content does not need to be marked as safe because this is automatically handled. """ raise NotImplementedError def full_width_page(self): """ Content that will be rendered within the full width of the detail page view. Content should be returned as an HTML string. Note that content does not need to be marked as safe because this is automatically handled. """ raise NotImplementedError def buttons(self): """ Buttons that will be rendered and added to the existing list of buttons on the detail page view. Content should be returned as an HTML string. Note that content does not need to be marked as safe because this is automatically handled. """ raise NotImplementedError def register_content_classes(): """ Helper method that populates the registry with all template content classes that have been registered by plugins """ registry.plugin_template_content_classes = collections.defaultdict(list) responses = register_detail_page_content_classes.send('registration_event') for receiver, response in responses: if not isinstance(response, list): response = [response] for template_class in response: if not inspect.isclass(template_class): raise TypeError('Plugin content class {} was passes as an instance!'.format(template_class)) if not issubclass(template_class, PluginTemplateContent): raise TypeError('{} is not a subclass of extras.plugins.PluginTemplateContent!'.format(template_class)) if template_class.model is None: raise TypeError('Plugin content class {} does not define a valid model!'.format(template_class)) registry.plugin_template_content_classes[template_class.model].append(template_class) def get_content_classes(model): """ Given a model string, return the list of all registered template content classes. Populate the registry if it is empty. """ if not hasattr(registry, 'plugin_template_content_classes'): register_content_classes() return registry.plugin_template_content_classes.get(model, []) # # Nav menu links # class PluginNavMenuLink: """ This class represents a nav menu item. This constitutes primary link and its text, but also allows for specifying additional link buttons that appear to the right of the item in the van menu. Links are specified as Django reverse URL strings. Buttons are each specified as a list of PluginNavMenuButton instances. """ link = None link_text = None link_permission = None buttons = [] 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): self.link = link self.title = title self.icon_class = icon_class self.color = color self.permission = permission def register_nav_menu_links(): """ Helper method that populates the registry with all nav menu link classes that have been registered by plugins """ registry.plugin_nav_menu_link_classes = {} responses = register_nav_menu_link_classes.send('registration_event') for receiver, response in responses: # Import the app config for the plugin to get the name to be used as the nav menu section text module = importlib.import_module(receiver.__module__.split('.')[0]) default_app_config = getattr(module, 'default_app_config') module, app_config = default_app_config.rsplit('.', 1) app_config = getattr(importlib.import_module(module), app_config) section_name = app_config.NetBoxPluginMeta.name if not isinstance(response, list): response = [response] for link_class in response: if not inspect.isclass(link_class): raise TypeError('Plugin nav menu link class {} was passes as an instance!'.format(link_class)) if not issubclass(link_class, PluginNavMenuLink): raise TypeError('{} is not a subclass of extras.plugins.PluginNavMenuLink!'.format(link_class)) if link_class.link is None or link_class.link_text is None: raise TypeError('Plugin nav menu link {} must specify at least link and link_text'.format(link_class)) for button in link_class.buttons: if not isinstance(button, PluginNavMenuButton): raise TypeError('{} must be an instance of PluginNavMenuButton!'.format(button)) registry.plugin_nav_menu_link_classes[section_name] = response def get_nav_menu_link_classes(): """ Return the list of all registered nav menu link classes. Populate the registry if it is empty. """ if not hasattr(registry, 'plugin_nav_menu_link_classes'): register_nav_menu_links() return registry.plugin_nav_menu_link_classes