NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-24 08:25:17 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Include only registered models; permit passed context data to overwrite apps

Browse Source
This commit is contained in:
Jeremy Stretch 2023-08-09 14:39:02 -04:00
parent 93f6574902
commit 2d3d8850c2
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
netbox/extras/models/configs.py
View File

@ -257,14 +257,19 @@ class ConfigTemplate(SyncedDataMixin, ExportTemplatesMixin, TagsMixin, ChangeLog
"""
Render the contents of the template.
"""
context = context or {}
_context = dict()
app_ns = registry['model_features']['custom_fields'].keys()
for app in app_ns:
context.setdefault(app, {})
models = apps.get_app_config(app).get_models()
for model in models:
context[app][model.__name__] = model
# Populate the default template context with NetBox model classes, namespaced by app
# TODO: Devise a canonical mechanism for identifying the models to include (see #13427)
for app, model_names in registry['model_features']['custom_fields'].items():
_context.setdefault(app, {})
for model_name in model_names:
model = apps.get_registered_model(app, model_name)
_context[app][model.__name__] = model
# Add the provided context data, if any
if context is not None:
_context.update(context)
# Initialize the Jinja2 environment and instantiate the Template
environment = self._get_environment()
@ -272,7 +277,7 @@ class ConfigTemplate(SyncedDataMixin, ExportTemplatesMixin, TagsMixin, ChangeLog
template = environment.get_template(self.data_file.path)
else:
template = environment.from_string(self.template_code)
output = template.render(**context)
output = template.render(**_context)
# Replace CRLF-style line terminators
return output.replace('\r\n', '\n')