lib/python/mitsql/interface/templates.py

   1 import cherrypy
   2 from mako.lookup import TemplateLookup
   3
   4 class MakoHandler(cherrypy.dispatch.LateParamPageHandler):
   5     """Callable which sets response.body."""
   6
   7     def __init__(self, template, next_handler):
   8         self.template = template
   9         self.next_handler = next_handler
  10
  11     def __call__(self):
  12         env = globals().copy()
  13         env.update(self.next_handler())
  14         return self.template.render(**env)
  15
  16 class MakoLoader(object):
  17
  18     def __init__(self):
  19         self.lookups = {}
  20
  21     def __call__(self, filename, directories, module_directory=None,
  22                  collection_size=-1):
  23         # Find the appropriate template lookup.
  24         key = (tuple(directories), module_directory)
  25         try:
  26             lookup = self.lookups[key]
  27         except KeyError:
  28             lookup = TemplateLookup(directories=directories,
  29                                     module_directory=module_directory,
  30                                     collection_size=collection_size,
  31                                     )
  32             self.lookups[key] = lookup
  33         cherrypy.request.lookup = lookup
  34
  35         # Replace the current handler.
  36         cherrypy.request.template = t = lookup.get_template(filename)
  37         cherrypy.request.handler = MakoHandler(t, cherrypy.request.handler)
  38
  39 cherrypy.tools.mako = cherrypy.Tool('on_start_resource', MakoLoader())