NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-22 20:12:00 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

inject origional context as obj_context

Browse Source
This commit is contained in:
John Anderson 2020-03-17 00:03:58 -04:00
parent 2522b88fc6
commit 457354c244
2 changed files with 24 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/extras/plugins/__init__.py
View File

@ -18,15 +18,20 @@ class PluginTemplateContent:
"""
model = None
def __init__(self, obj):
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` but an additional context dictionary may be passed as `extra_context`.
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}
context = {
'obj': self.obj,
'obj_context': self.context
}
if isinstance(extra_context, dict):
context.update(extra_context)

32
netbox/extras/templatetags/plugins.py
View File

@ -8,16 +8,16 @@ from extras.plugins import get_content_classes
register = template_.Library()
def _get_registered_content(obj, method):
def _get_registered_content(obj, method, context):
"""
Given an object and a PluginTemplateContent method name, return all the registered content for the
object's model.
Given an object and a PluginTemplateContent method name and the template context, return all the
registered content for the object's model.
"""
html = ''
plugin_template_classes = get_content_classes(obj._meta.label_lower)
for plugin_template_class in plugin_template_classes:
plugin_template_renderer = plugin_template_class(obj)
plugin_template_renderer = plugin_template_class(obj, context)
try:
content = getattr(plugin_template_renderer, method)()
except NotImplementedError:
@ -28,33 +28,33 @@ def _get_registered_content(obj, method):
return mark_safe(html)
@register.simple_tag()
def plugin_buttons(obj):
@register.simple_tag(takes_context=True)
def plugin_buttons(context, obj):
"""
Fire signal to collect all buttons registered by plugins
"""
return _get_registered_content(obj, 'buttons')
return _get_registered_content(obj, 'buttons', context)
@register.simple_tag()
def plugin_left_page(obj):
@register.simple_tag(takes_context=True)
def plugin_left_page(context, obj):
"""
Fire signal to collect all left page content registered by plugins
"""
return _get_registered_content(obj, 'left_page')
return _get_registered_content(obj, 'left_page', context)
@register.simple_tag()
def plugin_right_page(obj):
@register.simple_tag(takes_context=True)
def plugin_right_page(context, obj):
"""
Fire signal to collect all right page content registered by plugins
"""
return _get_registered_content(obj, 'right_page')
return _get_registered_content(obj, 'right_page', context)
@register.simple_tag()
def plugin_full_width_page(obj):
@register.simple_tag(takes_context=True)
def plugin_full_width_page(context, obj):
"""
Fire signal to collect all full width page content registered by plugins
"""
return _get_registered_content(obj, 'full_width_page')
return _get_registered_content(obj, 'full_width_page', context)