cmlenz@820: import os
cmlenz@820: 
cmlenz@820: import cherrypy
cmlenz@820: from genshi.core import Stream
cmlenz@820: from genshi.output import encode, get_serializer
cmlenz@820: from genshi.template import Context, TemplateLoader
cmlenz@820: 
cmlenz@820: from geddit.lib import ajax
cmlenz@820: 
cmlenz@820: loader = TemplateLoader(
cmlenz@820:     os.path.join(os.path.dirname(__file__), '..', 'templates'),
cmlenz@820:     auto_reload=True
cmlenz@820: )
cmlenz@820: 
cmlenz@820: def output(filename, method='html', encoding='utf-8', **options):
cmlenz@820:     """Decorator for exposed methods to specify what template the should use
cmlenz@820:     for rendering, and which serialization method and options should be
cmlenz@820:     applied.
cmlenz@820:     """
cmlenz@820:     def decorate(func):
cmlenz@820:         def wrapper(*args, **kwargs):
cmlenz@820:             cherrypy.thread_data.template = loader.load(filename)
cmlenz@820:             opt = options.copy()
cmlenz@820:             if not ajax.is_xhr() and method == 'html':
cmlenz@820:                 opt.setdefault('doctype', 'html')
cmlenz@820:             serializer = get_serializer(method, **opt)
cmlenz@820:             stream = func(*args, **kwargs)
cmlenz@820:             if not isinstance(stream, Stream):
cmlenz@820:                 return stream
cmlenz@820:             return encode(serializer(stream), method=serializer,
cmlenz@820:                           encoding=encoding)
cmlenz@820:         return wrapper
cmlenz@820:     return decorate
cmlenz@820: 
cmlenz@820: def render(*args, **kwargs):
cmlenz@820:     """Function to render the given data to the template specified via the
cmlenz@820:     ``@output`` decorator.
cmlenz@820:     """
cmlenz@820:     if args:
cmlenz@820:         assert len(args) == 1, \
cmlenz@820:             'Expected exactly one argument, but got %r' % (args,)
cmlenz@820:         template = loader.load(args[0])
cmlenz@820:     else:
cmlenz@820:         template = cherrypy.thread_data.template
cmlenz@820:     ctxt = Context(url=cherrypy.url)
cmlenz@820:     ctxt.push(kwargs)
cmlenz@820:     return template.generate(ctxt)