| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- # ext/turbogears.py
- # Copyright (C) 2006-2016 the Mako authors and contributors <see AUTHORS file>
- #
- # This module is part of Mako and is released under
- # the MIT License: http://www.opensource.org/licenses/mit-license.php
- from mako import compat
- from mako.lookup import TemplateLookup
- from mako.template import Template
- class TGPlugin(object):
- """TurboGears compatible Template Plugin."""
- def __init__(self, extra_vars_func=None, options=None, extension='mak'):
- self.extra_vars_func = extra_vars_func
- self.extension = extension
- if not options:
- options = {}
- # Pull the options out and initialize the lookup
- lookup_options = {}
- for k, v in options.items():
- if k.startswith('mako.'):
- lookup_options[k[5:]] = v
- elif k in ['directories', 'filesystem_checks', 'module_directory']:
- lookup_options[k] = v
- self.lookup = TemplateLookup(**lookup_options)
- self.tmpl_options = {}
- # transfer lookup args to template args, based on those available
- # in getargspec
- for kw in compat.inspect_getargspec(Template.__init__)[0]:
- if kw in lookup_options:
- self.tmpl_options[kw] = lookup_options[kw]
- def load_template(self, templatename, template_string=None):
- """Loads a template from a file or a string"""
- if template_string is not None:
- return Template(template_string, **self.tmpl_options)
- # Translate TG dot notation to normal / template path
- if '/' not in templatename:
- templatename = '/' + templatename.replace('.', '/') + '.' +\
- self.extension
- # Lookup template
- return self.lookup.get_template(templatename)
- def render(self, info, format="html", fragment=False, template=None):
- if isinstance(template, compat.string_types):
- template = self.load_template(template)
- # Load extra vars func if provided
- if self.extra_vars_func:
- info.update(self.extra_vars_func())
- return template.render(**info)
|
