web_template.py

"""
Benchmark for twisted.web.template rendering.
"""
from io import BytesIO
from time import time

from twisted.internet.defer import succeed
from twisted.python.compat import xrange
from twisted.web.template import (
    Element,
    TagLoader,
    XMLString,
    flatten,
    renderer,
    tags,
)

from benchlib import driver


class Elem(Element):
    loader = XMLString(
        b"""
        <div xmlns:t="http://twistedmatrix.com/ns/twisted.web.template/0.1">
        """
        + b"A" * 1000
        + b"""
        <div t:render="r" />
        <div t:render="r2" />
        <div t:render="r3">
        <t:slot name="meep" />
        </div>
        </div>
        """
    )

    def __init__(self, children=[]):
        super(Elem, self).__init__()
        self.children = children

    @renderer
    def r(self, req, tag):
        return tag([self.children, u'hi mom!'], attr=u'value')

    @renderer
    def r2(self, req, tag):
        return tags.div(u'foo', attr=u'value')

    @renderer
    def r3(self, req, tag):
        return tag.fillSlots(
            meep=(
                u'slotvalue',
                u'42',
                b'bar',
                tags.div(u'meep', attr=u'value'),
            )
        )


def render():
    child = Elem()
    for _ in xrange(20):
        child = Elem([child])
    root = TagLoader([child] * 10).load()
    out = BytesIO()
    flatten(None, root, out.write)


def main(reactor, duration):
    start = time()
    count = 0
    while time() - start < duration:
        render()
        count += 1
    return succeed(count)


if __name__ == '__main__':
    import sys
    import web_template

    driver(web_template.main, sys.argv)