1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
# Copyright 2016 Free Software Foundation, Inc.
# This file is part of GNU Radio
#
# SPDX-License-Identifier: GPL-2.0-or-later
#
"""
This dict class holds a (shared) cache of compiled mako templates.
These
"""
from __future__ import absolute_import, print_function
from mako.template import Template
from mako.exceptions import SyntaxException
from ..errors import TemplateError
class MakoTemplates(dict):
_template_cache = {}
def __init__(self, _bind_to=None, *args, **kwargs):
self.instance = _bind_to
dict.__init__(self, *args, **kwargs)
def __get__(self, instance, owner):
if instance is None or self.instance is not None:
return self
copy = self.__class__(_bind_to=instance, **self)
if getattr(instance.__class__, 'templates', None) is self:
setattr(instance, 'templates', copy)
return copy
@classmethod
def compile(cls, text):
text = str(text)
try:
template = Template(text, strict_undefined=True)
except SyntaxException as error:
raise TemplateError(text, *error.args)
cls._template_cache[text] = template
return template
def _get_template(self, text):
try:
return self._template_cache[str(text)]
except KeyError:
return self.compile(text)
def render(self, item):
text = self.get(item)
if not text:
return ''
namespace = self.instance.namespace_templates
try:
if isinstance(text, list):
templates = (self._get_template(t) for t in text)
return [template.render(**namespace) for template in templates]
else:
template = self._get_template(text)
return template.render(**namespace)
except Exception as error:
raise TemplateError(error, text)
|
