820
|
1 import os
|
|
2
|
|
3 import cherrypy
|
|
4 from genshi.core import Stream
|
|
5 from genshi.output import encode, get_serializer
|
|
6 from genshi.template import Context, TemplateLoader
|
|
7
|
|
8 from geddit.lib import ajax
|
|
9
|
|
10 loader = TemplateLoader(
|
|
11 os.path.join(os.path.dirname(__file__), '..', 'templates'),
|
|
12 auto_reload=True
|
|
13 )
|
|
14
|
|
15 def output(filename, method='html', encoding='utf-8', **options):
|
|
16 """Decorator for exposed methods to specify what template the should use
|
|
17 for rendering, and which serialization method and options should be
|
|
18 applied.
|
|
19 """
|
|
20 def decorate(func):
|
|
21 def wrapper(*args, **kwargs):
|
|
22 cherrypy.thread_data.template = loader.load(filename)
|
|
23 opt = options.copy()
|
|
24 if not ajax.is_xhr() and method == 'html':
|
|
25 opt.setdefault('doctype', 'html')
|
|
26 serializer = get_serializer(method, **opt)
|
|
27 stream = func(*args, **kwargs)
|
|
28 if not isinstance(stream, Stream):
|
|
29 return stream
|
|
30 return encode(serializer(stream), method=serializer,
|
|
31 encoding=encoding)
|
|
32 return wrapper
|
|
33 return decorate
|
|
34
|
|
35 def render(*args, **kwargs):
|
|
36 """Function to render the given data to the template specified via the
|
|
37 ``@output`` decorator.
|
|
38 """
|
|
39 if args:
|
|
40 assert len(args) == 1, \
|
|
41 'Expected exactly one argument, but got %r' % (args,)
|
|
42 template = loader.load(args[0])
|
|
43 else:
|
|
44 template = cherrypy.thread_data.template
|
|
45 ctxt = Context(url=cherrypy.url)
|
|
46 ctxt.push(kwargs)
|
|
47 return template.generate(ctxt)
|