Armin Ronacher | ed03db5 | 2007-02-28 22:39:44 +0100 | [diff] [blame] | 1 | # -*- coding: utf-8 -*_ |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 2 | # Template language benchmarks |
| 3 | # |
| 4 | # Objective: Generate a 1000x10 HTML table as fast as possible. |
| 5 | # adapted for jinja 1 |
| 6 | # |
| 7 | # Author: Jonas Borgström <jonas@edgewall.com> |
| 8 | # Author: Armin Ronacher <armin.ronacher@active-4.com> |
| 9 | |
| 10 | import cgi |
| 11 | import sys |
| 12 | import timeit |
Armin Ronacher | 5adf94f | 2007-03-29 22:11:59 +0200 | [diff] [blame] | 13 | import jdebug |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 14 | from StringIO import StringIO |
| 15 | |
| 16 | from genshi.builder import tag |
| 17 | from genshi.template import MarkupTemplate |
| 18 | |
| 19 | from jinja import Environment |
| 20 | |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 21 | try: |
| 22 | from django.conf import settings |
| 23 | settings.configure() |
| 24 | from django.template import Context as DjangoContext |
| 25 | from django.template import Template as DjangoTemplate |
| 26 | have_django = True |
| 27 | except ImportError: |
| 28 | have_django = False |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 29 | |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 30 | try: |
| 31 | from kid import Template as KidTemplate |
| 32 | have_kid = True |
| 33 | except ImportError: |
| 34 | have_kid = False |
| 35 | |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 36 | from Cheetah.Template import Template as CheetahTemplate |
| 37 | |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 38 | try: |
| 39 | from mako.template import Template as MakoTemplate |
| 40 | have_mako = True |
| 41 | except ImportError: |
| 42 | have_mako = False |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 43 | |
Armin Ronacher | fb5bebc | 2007-04-27 18:24:19 +0200 | [diff] [blame] | 44 | table = [dict(zip('abcdefghij', map(unicode,range(1, 11)))) |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 45 | for x in range(1000)] |
| 46 | |
| 47 | genshi_tmpl = MarkupTemplate(""" |
| 48 | <table xmlns:py="http://genshi.edgewall.org/"> |
| 49 | <tr py:for="row in table"> |
| 50 | <td py:for="c in row.values()" py:content="c"/> |
| 51 | </tr> |
| 52 | </table> |
| 53 | """) |
| 54 | |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 55 | if have_kid: |
| 56 | kid_tmpl = KidTemplate(""" |
| 57 | <table xmlns:py="http://purl.org/kid/ns#"> |
| 58 | <tr py:for="row in table"> |
| 59 | <td py:for="c in row.values()" py:content="c"/> |
| 60 | </tr> |
| 61 | </table> |
| 62 | """) |
| 63 | |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 64 | if have_django: |
| 65 | django_tmpl = DjangoTemplate(""" |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 66 | <table> |
| 67 | {% for row in table %} |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 68 | <tr>{% for col in row.values %}<td>{{ col }}</td>{% endfor %}</tr> |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 69 | {% endfor %} |
| 70 | </table> |
| 71 | """) |
| 72 | |
| 73 | jinja_tmpl = Environment().from_string(''' |
| 74 | <table> |
Armin Ronacher | 5adf94f | 2007-03-29 22:11:59 +0200 | [diff] [blame] | 75 | {% for row in table -%} |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 76 | <tr>{% for col in row.values() %}<td>{{ col }}</td>{% endfor %}</tr> |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 77 | {% endfor %} |
| 78 | </table> |
| 79 | ''') |
| 80 | |
| 81 | cheetah_tmpl = CheetahTemplate(''' |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 82 | <table> |
| 83 | #for $row in $table |
| 84 | <tr> |
| 85 | #for $col in $row.values() |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 86 | <td>$col</td> |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 87 | #end for |
| 88 | </tr> |
| 89 | #end for |
| 90 | </table> |
| 91 | ''', searchList=[{'table': table, 'escape': cgi.escape}]) |
| 92 | |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 93 | if have_mako: |
| 94 | mako_tmpl = MakoTemplate(''' |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 95 | <table> |
| 96 | % for row in table: |
| 97 | <tr> |
| 98 | % for col in row.values(): |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 99 | <td>${col}</td> |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 100 | % endfor |
| 101 | </tr> |
| 102 | % endfor |
| 103 | </table> |
| 104 | ''') |
| 105 | |
| 106 | def test_django(): |
| 107 | """Django Templates""" |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 108 | if not have_django: |
| 109 | return |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 110 | context = DjangoContext({'table': table}) |
| 111 | django_tmpl.render(context) |
| 112 | |
| 113 | def test_jinja(): |
| 114 | """Jinja Templates""" |
| 115 | jinja_tmpl.render(table=table) |
| 116 | |
| 117 | def test_genshi(): |
| 118 | """Genshi Templates""" |
| 119 | stream = genshi_tmpl.generate(table=table) |
| 120 | stream.render('html', strip_whitespace=False) |
| 121 | |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 122 | def test_kid(): |
| 123 | """Kid Templates""" |
| 124 | if not have_kid: |
| 125 | return |
| 126 | kid_tmpl.table = table |
| 127 | kid_tmpl.serialize(output="html") |
| 128 | |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 129 | def test_cheetah(): |
| 130 | """Cheetah Templates""" |
| 131 | cheetah_tmpl.respond() |
| 132 | |
| 133 | def test_mako(): |
| 134 | """Mako Templates""" |
Alexander Schremmer | 303a7e4 | 2007-02-28 22:44:14 +0100 | [diff] [blame] | 135 | if not have_mako: |
| 136 | return |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 137 | mako_tmpl.render(table=table) |
| 138 | |
| 139 | |
| 140 | def run(which=None, number=10): |
Armin Ronacher | e98c5f5 | 2007-04-21 09:39:06 +0200 | [diff] [blame] | 141 | tests = ['test_django', 'test_jinja', 'test_kid', 'test_genshi', |
| 142 | 'test_cheetah', 'test_mako'] |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 143 | |
| 144 | if which: |
| 145 | tests = filter(lambda n: n[5:] in which, tests) |
| 146 | |
| 147 | for test in [t for t in tests if hasattr(sys.modules[__name__], t)]: |
| 148 | t = timeit.Timer(setup='from __main__ import %s;' % test, |
| 149 | stmt='%s()' % test) |
| 150 | time = t.timeit(number=number) / number |
| 151 | |
| 152 | if time < 0.00001: |
| 153 | result = ' (not installed?)' |
| 154 | else: |
| 155 | result = '%16.2f ms' % (1000 * time) |
| 156 | print '%-35s %s' % (getattr(sys.modules[__name__], test).__doc__, result) |
| 157 | |
| 158 | |
| 159 | if __name__ == '__main__': |
| 160 | which = [arg for arg in sys.argv[1:] if arg[0] != '-'] |
| 161 | |
| 162 | if '-p' in sys.argv: |
Armin Ronacher | 5adf94f | 2007-03-29 22:11:59 +0200 | [diff] [blame] | 163 | from cProfile import Profile |
| 164 | from pstats import Stats |
| 165 | p = Profile() |
| 166 | p.runcall(test_jinja) |
| 167 | stats = Stats(p) |
Armin Ronacher | de478f6 | 2007-02-28 22:35:04 +0100 | [diff] [blame] | 168 | stats.strip_dirs() |
| 169 | stats.sort_stats('time', 'calls') |
| 170 | stats.print_stats() |
| 171 | else: |
| 172 | run(which) |