comparison mercurial/templater.py @ 21846:8f23f8096606

templater: introduce word function

This function allows returning only the nth "word" from a string. By default
a string is split as by Python's split() function default, but an optional
third parameter can also override what string the string is split by.

21845:04f5b5e3792e

    if text.startswith(patn):
        return text
    return ''


methods = {
    "string": lambda e, c: (runstring, e[1]),
    "rawstring": lambda e, c: (runrawstring, e[1]),
    "symbol": lambda e, c: (runsymbol, e[1]),
    "group": lambda e, c: compileexp(e[1], c),

    "rstdoc": rstdoc,
    "shortest": shortest,
    "startswith": startswith,
    "strip": strip,
    "sub": sub,
}

# template engine

path = ['templates', '../templates']

21846:8f23f8096606

    if text.startswith(patn):
        return text
    return ''


def word(context, mapping, args):
    """return nth word from a string"""
    if not (2 <= len(args) <= 3):
        raise error.ParseError(_("word expects two or three arguments, got %d")
                               % len(args))

    num = int(stringify(args[0][0](context, mapping, args[0][1])))
    text = stringify(args[1][0](context, mapping, args[1][1]))
    if len(args) == 3:
        splitter = stringify(args[2][0](context, mapping, args[2][1]))
    else:
        splitter = None

    tokens = text.split(splitter)
    if num >= len(tokens):
        return ''
    else:
        return tokens[num]

methods = {
    "string": lambda e, c: (runstring, e[1]),
    "rawstring": lambda e, c: (runrawstring, e[1]),
    "symbol": lambda e, c: (runsymbol, e[1]),
    "group": lambda e, c: compileexp(e[1], c),

    "rstdoc": rstdoc,
    "shortest": shortest,
    "startswith": startswith,
    "strip": strip,
    "sub": sub,
    "word": word,
}

# template engine

path = ['templates', '../templates']