NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 09:16:10 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Pre-fetch referenced templates

Browse Source
This commit is contained in:
jeremystretch 2023-02-10 16:01:41 -05:00
parent 8681e6d823
commit adaeedd6a2
2 changed files with 28 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -230,16 +230,20 @@ class ConfigTemplate(SyncedDataMixin, ExportTemplatesMixin, ChangeLoggedModel):
Render the contents of the template. Render the contents of the template.
""" """
context = context or {} context = context or {}
template_code = self.template_code template_path = self.data_file.path
# output = render_jinja2(template_code, context)
environment = SandboxedEnvironment( # Initialize the template loader & cache the base template code
loader=ConfigTemplateLoader(data_source=self.data_source) loader = ConfigTemplateLoader(data_source=self.data_source)
) loader.cache_templates({
template_path: self.template_code
})
# Initialize the Jinja2 environment
environment = SandboxedEnvironment(loader=loader)
environment.filters.update(get_config().JINJA2_FILTERS) environment.filters.update(get_config().JINJA2_FILTERS)
output = environment.from_string(source=template_code).render(**context)
# Render the template
output = environment.get_template(template_path).render(**context)
# Replace CRLF-style line terminators # Replace CRLF-style line terminators
output = output.replace('\r\n', '\n') return output.replace('\r\n', '\n')
return output

20
netbox/utilities/jinja2.py
View File

@ -1,7 +1,6 @@
import os
from django.apps import apps from django.apps import apps
from jinja2 import BaseLoader, TemplateNotFound from jinja2 import BaseLoader, TemplateNotFound
from jinja2.meta import find_referenced_templates
__all__ = ( __all__ = (
'ConfigTemplateLoader', 'ConfigTemplateLoader',
@ -14,14 +13,25 @@ class ConfigTemplateLoader(BaseLoader):
""" """
def __init__(self, data_source): def __init__(self, data_source):
self.data_source = data_source self.data_source = data_source
self._template_cache = {}
def get_source(self, environment, template): def get_source(self, environment, template):
DataFile = apps.get_model('core', 'DataFile') DataFile = apps.get_model('core', 'DataFile')
# Retrieve template content from cache
try: try:
datafile = DataFile.objects.get(source=self.data_source, path=template) template_source = self._template_cache[template]
template_source = datafile.data_as_string except KeyError:
except DataFile.DoesNotExist:
raise TemplateNotFound(template) raise TemplateNotFound(template)
# Find and pre-fetch referenced templates
if referenced_templates := find_referenced_templates(environment.parse(template_source)):
self.cache_templates({
df.path: df.data_as_string for df in
DataFile.objects.filter(source=self.data_source, path__in=referenced_templates)
})
return template_source, template, True return template_source, template, True
def cache_templates(self, templates):
self._template_cache.update(templates)