Przeglądaj źródła

Move extras.jinja2.ConfigTemplateLoader to utilities.jinja2.DataFileLoader

Jeremy Stretch 1 rok temu
rodzic
commit
c30d22335a
2 zmienionych plików z 4 dodań i 4 usunięć
  1. 2 2
      netbox/extras/models/configs.py
  2. 2 2
      netbox/utilities/jinja2.py

+ 2 - 2
netbox/extras/models/configs.py

@@ -7,12 +7,12 @@ from django.utils.translation import gettext_lazy as _
 from jinja2.loaders import BaseLoader
 from jinja2.loaders import BaseLoader
 from jinja2.sandbox import SandboxedEnvironment
 from jinja2.sandbox import SandboxedEnvironment
 
 
-from extras.jinja2 import ConfigTemplateLoader
 from extras.querysets import ConfigContextQuerySet
 from extras.querysets import ConfigContextQuerySet
 from netbox.config import get_config
 from netbox.config import get_config
 from netbox.registry import registry
 from netbox.registry import registry
 from netbox.models import ChangeLoggedModel
 from netbox.models import ChangeLoggedModel
 from netbox.models.features import CloningMixin, CustomLinksMixin, ExportTemplatesMixin, SyncedDataMixin, TagsMixin
 from netbox.models.features import CloningMixin, CustomLinksMixin, ExportTemplatesMixin, SyncedDataMixin, TagsMixin
+from utilities.jinja2 import DataFileLoader
 from utilities.utils import deepmerge
 from utilities.utils import deepmerge
 
 
 __all__ = (
 __all__ = (
@@ -290,7 +290,7 @@ class ConfigTemplate(SyncedDataMixin, CustomLinksMixin, ExportTemplatesMixin, Ta
         """
         """
         # Initialize the template loader & cache the base template code (if applicable)
         # Initialize the template loader & cache the base template code (if applicable)
         if self.data_file:
         if self.data_file:
-            loader = ConfigTemplateLoader(data_source=self.data_source)
+            loader = DataFileLoader(data_source=self.data_source)
             loader.cache_templates({
             loader.cache_templates({
                 self.data_file.path: self.template_code
                 self.data_file.path: self.template_code
             })
             })

+ 2 - 2
netbox/extras/jinja2.py → netbox/utilities/jinja2.py

@@ -3,11 +3,11 @@ from jinja2 import BaseLoader, TemplateNotFound
 from jinja2.meta import find_referenced_templates
 from jinja2.meta import find_referenced_templates
 
 
 __all__ = (
 __all__ = (
-    'ConfigTemplateLoader',
+    'DataFileLoader',
 )
 )
 
 
 
 
-class ConfigTemplateLoader(BaseLoader):
+class DataFileLoader(BaseLoader):
     """
     """
     Custom Jinja2 loader to facilitate populating template content from DataFiles.
     Custom Jinja2 loader to facilitate populating template content from DataFiles.
     """
     """