engine.py 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. import functools
  2. from django.core.exceptions import ImproperlyConfigured
  3. from django.utils.functional import cached_property
  4. from django.utils.module_loading import import_string
  5. from .base import Template
  6. from .context import Context, _builtin_context_processors
  7. from .exceptions import TemplateDoesNotExist
  8. from .library import import_library
  9. class Engine:
  10. default_builtins = [
  11. "django.template.defaulttags",
  12. "django.template.defaultfilters",
  13. "django.template.loader_tags",
  14. ]
  15. def __init__(
  16. self,
  17. dirs=None,
  18. app_dirs=False,
  19. context_processors=None,
  20. debug=False,
  21. loaders=None,
  22. string_if_invalid="",
  23. file_charset="utf-8",
  24. libraries=None,
  25. builtins=None,
  26. autoescape=True,
  27. ):
  28. if dirs is None:
  29. dirs = []
  30. if context_processors is None:
  31. context_processors = []
  32. if loaders is None:
  33. loaders = ["django.template.loaders.filesystem.Loader"]
  34. if app_dirs:
  35. loaders += ["django.template.loaders.app_directories.Loader"]
  36. loaders = [("django.template.loaders.cached.Loader", loaders)]
  37. else:
  38. if app_dirs:
  39. raise ImproperlyConfigured(
  40. "app_dirs must not be set when loaders is defined."
  41. )
  42. if libraries is None:
  43. libraries = {}
  44. if builtins is None:
  45. builtins = []
  46. self.dirs = dirs
  47. self.app_dirs = app_dirs
  48. self.autoescape = autoescape
  49. self.context_processors = context_processors
  50. self.debug = debug
  51. self.loaders = loaders
  52. self.string_if_invalid = string_if_invalid
  53. self.file_charset = file_charset
  54. self.libraries = libraries
  55. self.template_libraries = self.get_template_libraries(libraries)
  56. self.builtins = self.default_builtins + builtins
  57. self.template_builtins = self.get_template_builtins(self.builtins)
  58. def __repr__(self):
  59. return (
  60. "<%s:%s app_dirs=%s%s debug=%s loaders=%s string_if_invalid=%s "
  61. "file_charset=%s%s%s autoescape=%s>"
  62. ) % (
  63. self.__class__.__qualname__,
  64. "" if not self.dirs else " dirs=%s" % repr(self.dirs),
  65. self.app_dirs,
  66. (
  67. ""
  68. if not self.context_processors
  69. else " context_processors=%s" % repr(self.context_processors)
  70. ),
  71. self.debug,
  72. repr(self.loaders),
  73. repr(self.string_if_invalid),
  74. repr(self.file_charset),
  75. "" if not self.libraries else " libraries=%s" % repr(self.libraries),
  76. "" if not self.builtins else " builtins=%s" % repr(self.builtins),
  77. repr(self.autoescape),
  78. )
  79. @staticmethod
  80. @functools.lru_cache
  81. def get_default():
  82. """
  83. Return the first DjangoTemplates backend that's configured, or raise
  84. ImproperlyConfigured if none are configured.
  85. This is required for preserving historical APIs that rely on a
  86. globally available, implicitly configured engine such as:
  87. >>> from django.template import Context, Template
  88. >>> template = Template("Hello {{ name }}!")
  89. >>> context = Context({'name': "world"})
  90. >>> template.render(context)
  91. 'Hello world!'
  92. """
  93. # Since Engine is imported in django.template and since
  94. # DjangoTemplates is a wrapper around this Engine class,
  95. # local imports are required to avoid import loops.
  96. from django.template import engines
  97. from django.template.backends.django import DjangoTemplates
  98. for engine in engines.all():
  99. if isinstance(engine, DjangoTemplates):
  100. return engine.engine
  101. raise ImproperlyConfigured("No DjangoTemplates backend is configured.")
  102. @cached_property
  103. def template_context_processors(self):
  104. context_processors = _builtin_context_processors
  105. context_processors += tuple(self.context_processors)
  106. return tuple(import_string(path) for path in context_processors)
  107. def get_template_builtins(self, builtins):
  108. return [import_library(x) for x in builtins]
  109. def get_template_libraries(self, libraries):
  110. loaded = {}
  111. for name, path in libraries.items():
  112. loaded[name] = import_library(path)
  113. return loaded
  114. @cached_property
  115. def template_loaders(self):
  116. return self.get_template_loaders(self.loaders)
  117. def get_template_loaders(self, template_loaders):
  118. loaders = []
  119. for template_loader in template_loaders:
  120. loader = self.find_template_loader(template_loader)
  121. if loader is not None:
  122. loaders.append(loader)
  123. return loaders
  124. def find_template_loader(self, loader):
  125. if isinstance(loader, (tuple, list)):
  126. loader, *args = loader
  127. else:
  128. args = []
  129. if isinstance(loader, str):
  130. loader_class = import_string(loader)
  131. return loader_class(self, *args)
  132. else:
  133. raise ImproperlyConfigured(
  134. "Invalid value in template loaders configuration: %r" % loader
  135. )
  136. def find_template(self, name, dirs=None, skip=None):
  137. tried = []
  138. for loader in self.template_loaders:
  139. try:
  140. template = loader.get_template(name, skip=skip)
  141. return template, template.origin
  142. except TemplateDoesNotExist as e:
  143. tried.extend(e.tried)
  144. raise TemplateDoesNotExist(name, tried=tried)
  145. def from_string(self, template_code):
  146. """
  147. Return a compiled Template object for the given template code,
  148. handling template inheritance recursively.
  149. """
  150. return Template(template_code, engine=self)
  151. def get_template(self, template_name):
  152. """
  153. Return a compiled Template object for the given template name,
  154. handling template inheritance recursively.
  155. """
  156. template, origin = self.find_template(template_name)
  157. if not hasattr(template, "render"):
  158. # template needs to be compiled
  159. template = Template(template, origin, template_name, engine=self)
  160. return template
  161. def render_to_string(self, template_name, context=None):
  162. """
  163. Render the template specified by template_name with the given context.
  164. For use in Django's test suite.
  165. """
  166. if isinstance(template_name, (list, tuple)):
  167. t = self.select_template(template_name)
  168. else:
  169. t = self.get_template(template_name)
  170. # Django < 1.8 accepted a Context in `context` even though that's
  171. # unintended. Preserve this ability but don't rewrap `context`.
  172. if isinstance(context, Context):
  173. return t.render(context)
  174. else:
  175. return t.render(Context(context, autoescape=self.autoescape))
  176. def select_template(self, template_name_list):
  177. """
  178. Given a list of template names, return the first that can be loaded.
  179. """
  180. if not template_name_list:
  181. raise TemplateDoesNotExist("No template names provided")
  182. not_found = []
  183. for template_name in template_name_list:
  184. try:
  185. return self.get_template(template_name)
  186. except TemplateDoesNotExist as exc:
  187. if exc.args[0] not in not_found:
  188. not_found.append(exc.args[0])
  189. continue
  190. # If we get here, none of the templates could be loaded
  191. raise TemplateDoesNotExist(", ".join(not_found))