Ticket #584: error.py

import os
import genshi.template.loader
import genshi.template.markup

template_src = """<module
        xmlns:py="http://genshi.edgewall.org/"
        xmlns:xi="http://www.w3.org/2001/XInclude" py:strip="True">
        <div py:content="module['name']" />
        <py:for each="module in module.get('children', [])">
                <!--! Here's where the recursion occurs -->
                <xi:include href="module.xml" />
        </py:for>
</module>
"""

"""
A module with a nested module
"""
module = dict(
        type='module',
        name='module 1',
        children=[
                dict(type='module', name='module 2'),
        ],
)

def test_file_loader():
        try:
                with open('module.xml', 'w') as template_file:
                        template_file.write(template_src)
                tmpl = genshi.template.loader.TemplateLoader(['.']).load('module.xml')
                print(tmpl.generate(module=module))
        finally:
                os.remove('module.xml')

class SimpleLoader:
        def load(self, name, **kwargs):
                return genshi.template.markup.MarkupTemplate(template_src, loader=self)

def test_simple_loader():
        tmpl = SimpleLoader().load('module.xml')
        print(tmpl.generate(module=module))

if __name__ == '__main__':
        # works with simple_loader
        test_simple_loader()
        # infinite recursion with file_loader
        #test_file_loader()