Source code for wheezy.html.ext.mako

""" ``mako`` extension module.
"""

import re

from wheezy.html.ext.lexer import (
    InlinePreprocessor,
    Preprocessor,
    WhitespacePreprocessor,
)


class MakoPreprocessor(Preprocessor):
    def __init__(self, skip_imports=False):
        super(MakoPreprocessor, self).__init__(
            r"\$\{((?P<expr>.+?)\."
            r"(?P<widget>%(widgets)s){1}\((?P<params>.*?)\)\s*?"
            r"(?P<expr_filter>(\|\s*[\w,\s]+?|)))\}"
        )

    PREPEND = """\
<%!
from wheezy.html.utils import format_value
%>"""

    EXPRESSION = "${%(expr)s%(expr_filter)s}"

    ERROR_CLASS0 = """\\
%% if '%(name)s' in errors:
 class="error"\\
%% endif
"""

    ERROR_CLASS1 = """\\
%% if '%(name)s' in errors:
 class="error %(class)s"\\
%% else:
 class="%(class)s"\\
%% endif
"""

    MULTIPLE_HIDDEN = """\\
%% for item in %(value)s:
<input type="hidden" name="%(name)s" value="${item%(expr_filter)s}" />\\
%% endfor
"""

    INPUT = """\
<input id="%(id)s" name="%(name)s" type="%(type)s"%(attrs)s%(class)s\
%% if %(value)s%(condition)s:
 value="${%(func)s%(expr_filter)s}" />\\
%% else:
 />\\
%% endif
"""

    CHECKBOX = """\
<input id="%(id)s" name="%(name)s" type="checkbox" \
value="1"%(attrs)s%(class)s\
%% if %(value)s:
 checked="checked"\\
%% endif
 />"""

    MULTIPLE_CHECKBOX = """\\
%% for key, text in %(choices)s:
<label%(attrs)s%(class)s><input id="%(id)s" name="%(name)s" type="checkbox" \
value="1"%(attrs)s%(class)s\
%% if key in %(value)s:
 checked="checked"\\
%% endif
 />${text%(expr_filter)s}</label>\\
%% endfor
"""

    RADIO = """\\
%% for key, text in %(choices)s:
<label%(attrs)s%(class)s>\
<input type="radio" name="%(name)s"%(attrs)s \
value="${key%(expr_filter)s}"%(class)s\
%% if key == %(value)s:
 checked="checked"\\
%% endif
 />${text%(expr_filter)s}</label>\\
%% endfor
"""

    SELECT = """\
<select id="%(id)s" name="%(name)s"%(attrs)s%(class)s>\\
%% for key, text in %(choices)s:
<option value="${key%(expr_filter)s}"\\
%% if key == %(value)s:
 selected="selected"\\
%% endif
>${text%(expr_filter)s}</option>\\
%% endfor
</select>"""

    MULTIPLE_SELECT = """\
<select id="%(id)s" name="%(name)s" multiple="multiple"%(attrs)s%(class)s>\\
%% for key, text in %(choices)s:
<option value="${key%(expr_filter)s}"\\
%% if key in %(value)s:
 selected="selected"\\
%% endif
>${text%(expr_filter)s}</option>\\
%% endfor
</select>"""

    ERROR = """\\
%% if '%(name)s' in errors:
<span%(attrs)s>${errors['%(name)s'][-1]%(expr_filter)s}</span>\\
%% endif
"""

    MESSAGE = """\\
%% if %(value)s:
<span%(attrs)s>%(info)s</span>\\
%% endif
"""


widget_preprocessor = MakoPreprocessor()
whitespace_preprocessor = WhitespacePreprocessor(
    rules=[
        (re.compile(r"^ \s+|\s+$", re.MULTILINE), r""),
        (re.compile(r">\s+<"), r"><"),
    ]
)


RE_INLINE = re.compile(
    r'<%inline\s+file=("|\')(?P<path>.+?)\1\s*/>', re.MULTILINE
)


def inline_preprocessor(directories, fallback=False):
    """Inline preprocessor. Rewrite <%inline file="..." /> tag with
    file content. If fallback is ``True`` rewrite to
    <%include file="..." /> tag.

    >>> t = '1 <%inline file="master.html"/> 2'
    >>> m = RE_INLINE.search(t)
    >>> m.group('path')
    'master.html'
    >>> t[:m.start()], t[m.end():]
    ('1 ', ' 2')
    >>> m = RE_INLINE.search(' <%inline file="shared/footer.html"/>')
    >>> m.group('path')
    'shared/footer.html'
    """
    strategy = (
        fallback and (lambda path: '<%include file="' + path + '"/>') or None
    )
    return InlinePreprocessor(RE_INLINE, directories, strategy)