Выберите препроцессор Mako в зависимости от расширения файла? - PullRequest
Новая
       5

Выберите препроцессор Mako в зависимости от расширения файла?

6 голосов
/ 28 ноября 2011

Я хотел бы как-то инструмент mako.lookup.TemplateLookup, чтобы он применял определенные препроцессоры только для определенных расширений файлов.

В частности, у меня есть haml.preprocessor, который я хотел бы применить ко всем шаблонам, имя файла которых заканчивается на .haml.

Спасибо!

1 Ответ

4 голосов
/ 05 декабря 2011

Вы должны иметь возможность настроить 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>
...