← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to turbosimpletal/zptsupport.py

all tests pass, apart from the revision size limit one

Show diffs side-by-side

added added

removed removed

Lines of Context:
turbosimpletal/zptsupport.py
 
1
"TurboGears support for Zope Page Templates"
 
2
 
 
3
import StringIO
 
4
import logging
 
5
import os
 
6
import pkg_resources
 
7
 
 
8
#from zope.pagetemplate.pagetemplatefile import PageTemplateFile
 
9
from simpletal import simpleTAL, simpleTALES
 
10
 
 
11
log = logging.getLogger("turbogears.zptsupport")
 
12
 
 
13
_zpt_cache = {}
 
14
 
 
15
def zpt(tfile):
 
16
    tinstance = _zpt_cache.get(tfile)
 
17
    if tinstance is None:
 
18
        tinstance = _zpt_cache[tfile] = TemplateWrapper(simpleTAL.compileXMLTemplate(open(tfile)))
 
19
    return tinstance
 
20
 
 
21
class TemplateWrapper(object):
 
22
 
 
23
    def __init__(self, template):
 
24
        self.template = template
 
25
 
 
26
    def __call__(self, **kw):
 
27
        context = simpleTALES.Context(allowPythonPath=1)
 
28
        for k, v in kw.iteritems():
 
29
            context.addGlobal(k, v)
 
30
        s = StringIO.StringIO()
 
31
        self.template.expand(context, s)
 
32
        return s.getvalue()
 
33
 
 
34
    @property
 
35
    def macros(self):
 
36
        return self.template.macros
 
37
 
 
38
## class TGPageTemplateFile(PageTemplateFile):
 
39
 
 
40
##     def pt_getContext(self, args=(), options={}, **ignored):
 
41
##         namespace = super(TGPageTemplateFile, self).pt_getContext(
 
42
##             args, options, **ignored)
 
43
##         namespace.update(options)
 
44
##         return namespace
 
45
 
 
46
class TurboZpt(object):
 
47
    extension = "pt"
 
48
 
 
49
    def __init__(self, extra_vars_func=None):
 
50
        self.get_extra_vars = extra_vars_func
 
51
 
 
52
    def load_template(self, classname, loadingSite=False):
 
53
        """Searches for a template along the Python path.
 
54
 
 
55
        Template files must end in ".pt" and be in legitimate packages.
 
56
        Templates are automatically checked for changes and reloaded as
 
57
        neccessary.
 
58
        """
 
59
        divider = classname.rfind(".")
 
60
        if divider > -1:
 
61
            package = classname[0:divider]
 
62
            basename = classname[divider+1:]
 
63
        else:
 
64
            raise ValueError, "All templates must be in a package"
 
65
 
 
66
        tfile = pkg_resources.resource_filename(
 
67
            package, "%s.%s" % (basename, self.extension))
 
68
        return zpt(tfile)
 
69
 
 
70
    def render(self, info, format="html", fragment=False, template=None):
 
71
        """Renders data in the desired format.
 
72
 
 
73
        @param info: the data / context itself
 
74
        @type info: dict
 
75
        @para format: "html"
 
76
        @type format: "string"
 
77
        @para template: name of the template to use
 
78
        @type template: string
 
79
        """
 
80
        tinstance = self.load_template(template)
 
81
        #log.debug("Applying template %s" % (tinstance.filename))
 
82
        data = dict()
 
83
        if self.get_extra_vars:
 
84
            data.update(self.get_extra_vars())
 
85
        data.update(info)
 
86
        return tinstance(**data)