mako.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import
from mako.lookup import TemplateLookup
_lookup = None
from flask import request, session, url_for
from myojin import custom_script

def init(app,globals=None):
    global _globals, _lookup
    app.logger.debug(globals)
    
    _lookup = init_lookup(app.root_path + "/templates", app.root_path + "/tmp", globals=globals)
    _globals = globals or dict()
    _globals['DEBUG'] = app.config.get('DEBUG', False)
    _globals['url_for'] = url_for
_globals = None

def get_template(template_name):
    tmpl = _lookup.get_template(template_name)
    return tmpl
    
def init_lookup(directory, module_directory, globals=None):
    return TemplateLookup(directories=[directory], module_directory=module_directory,
                          input_encoding="utf-8",
                          output_encoding="utf-8",
                          default_filters=['h'])
    
def render(template_name, ctx=dict(), to_unicode=False):#, *args, **kws):
    tmpl = get_template(template_name)
    
    # TODO: cache tmpl
    render = tmpl.render if to_unicode else tmpl.render_unicode
    return render(request=request,session=session, **dict(
        mako_utils.items() +
        _globals.items() + ctx.items()))

from mako.filters import html_escape
from markupsafe import escape, Markup
def newline_filter(s, escape=False):
    if isinstance(s, str):
        s = unicode(s)
    elif isinstance(s, unicode):
        pass
    else:   
        assert False
    if escape:
        s = cgi.escape(s, True)
    return Markup(unicode(s).replace(u'\n',u'<br/>\n'))

from json import dumps
def json_dumps(s):
    assert isinstance(s, (tuple,list,dict))
    return Markup(dumps(s))

def debug_space(s):
    assert isinstance(s, unicode)
    return unicode(s).replace(u' ',u'　')

try:
    from flaskext.babel import gettext # as _
except ImportError, e:
    def gettext(*args, **kws):
        from flaskext.babel import gettext

def comma_num_filter(n):
    if isinstance(n, basestring):
        n = int(n)
    return "{:,d}".format(n)

def get_locale():
    try:
        from flaskext import babel
        return babel.get_locale()
    except:
        return None

mako_utils = dict(
    _=gettext,
    newline=newline_filter,
    debug_space=debug_space,
    JSON=json_dumps,
    comma=comma_num_filter,
    get_locale=get_locale
)