Add "directories=[]" param to LoraxTemplate

This commit is contained in:
Will Woods 2011-05-09 23:42:10 -04:00
parent 4343b12fc0
commit a15c5ded9c
1 changed files with 6 additions and 5 deletions

View File

@ -22,18 +22,19 @@
import sys import sys
import shlex import shlex
from mako.template import Template
from mako.lookup import TemplateLookup from mako.lookup import TemplateLookup
from mako.exceptions import RichTraceback from mako.exceptions import RichTraceback
class LoraxTemplate(object): class LoraxTemplate(object):
def __init__(self, directories=["/usr/share/lorax"]):
# we have to add ["/"] to the template lookup directories or the
# file includes won't work properly for absolute paths
self.directories = ["/"] + directories
def parse(self, template_file, variables): def parse(self, template_file, variables):
# we have to set the template lookup directories to ["/"], lookup = TemplateLookup(directories=self.directories)
# otherwise the file includes will not work properly template = lookup.get_template(template_file)
lookup = TemplateLookup(directories=["/"])
template = Template(filename=template_file, lookup=lookup)
try: try:
textbuf = template.render(**variables) textbuf = template.render(**variables)