选择Mako预处理器,根据文件扩展名进行?

时间:2011-11-28 18:51:14

标签: python preprocessor haml mako

我想以某种方式检测mako.lookup.TemplateLookup,使其仅对某些文件扩展名应用某些预处理程序。

具体来说,我有一个haml.preprocessor我想要应用于文件名以.haml结尾的所有模板。

谢谢!

1 个答案:

答案 0 :(得分:4)

您应该能够自定义TemplateLookup以获得所需的行为。

customlookup.py

from mako.lookup import TemplateLookup
import haml

class Lookup(TemplateLookup):
    def get_template(self, uri):
        if uri.rsplit('.')[1] == 'haml':
            # change preprocessor used for this template
            default = self.template_args['preprocessor']
            self.template_args['preprocessor'] = haml.preprocessor
            template = super(Lookup, self).get_template(uri)
            # change it back
            self.template_args['preprocessor'] = default
        else:
            template = super(Lookup, self).get_template(uri)
        return template

lookup = Lookup(['.'])
print lookup.get_template('index.haml').render()

index.haml

<%inherit file="base.html"/>

<%block name="content">
  %h1 Hello
</%block>

base.html文件

<html>
  <body>
    <%block name="content"/>
  </body>
</html>