← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to turbosimpletal/zptsupport.py

refinements, from jam

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