Removed __future__ import from wmako for consistency.
[wrw.git] / wrw / wmako.py
CommitLineData
3d866605
FT
1import os, threading
2from mako import template, lookup, filters
3import util, form, session
4
5# It seems Mako isn't thread-safe.
6makolock = threading.Lock()
7
8class liblookup(lookup.TemplateLookup):
9 def __init__(self, *args, **kwargs):
10 lookup.TemplateLookup.__init__(self, *args, **kwargs)
11
12 def adjust_uri(self, uri, relativeto):
13 return uri
14
15libdirs = []
16homedir = os.getenv("HOME")
17if homedir is not None:
18 usrdir = os.path.join(homedir, "wmako")
19 if os.path.exists(usrdir):
20 libdirs.append(usrdir)
21libdirs.append(os.path.join(os.path.dirname(__file__), "makolib"))
22cachedir = os.path.join("/tmp/", "mako-" + str(os.getuid()))
23defargs = {"output_encoding": "utf-8",
24 "input_encoding": "utf-8",
25 "default_filters": ["decode.utf8"],
26 "module_directory": cachedir,
27 }
28lib = liblookup(directories = libdirs, **defargs)
29
30if not os.path.exists(cachedir):
31 os.mkdir(cachedir)
32def handle(req, filename, **kw):
33 with makolock:
34 tt = template.Template(filename = filename, lookup = lib, **defargs)
35 req.ohead["Content-Type"] = "text/html; charset=utf-8"
36 return [tt.render(request = req, **kw)]
37
38@util.wsgiwrap
39def application(req):
40 return handle(req, req.filename,
41 form = form.formdata(req),
42 session = session.get(req))