# HG changeset patch # User cmlenz # Date 1164304097 0 # Node ID e1d659c87ddfd270bb28f8025b07572dfbe77ec0 # Parent 873ca2a7ec05c7f5f0dc96eb4226128477ac2e48 The builder API now accepts streams as children of elements and fragments. diff --git a/ChangeLog b/ChangeLog --- a/ChangeLog +++ b/ChangeLog @@ -30,6 +30,7 @@ * `MarkupTemplate`s can now be instantiated from markup streams, in addition to strings and file-like objects (ticket #69). * Improve handling of incorrectly nested tags in the HTML parser. + * The builder API now accepts streams as children of elements and fragments. Version 0.3.5 diff --git a/genshi/builder.py b/genshi/builder.py --- a/genshi/builder.py +++ b/genshi/builder.py @@ -46,7 +46,7 @@ def append(self, node): """Append an element or string as child node.""" - if isinstance(node, (Element, basestring, int, float, long)): + if isinstance(node, (Stream, Element, basestring, int, float, long)): # For objects of a known/primitive type, we avoid the check for # whether it is iterable for better performance self.children.append(node) @@ -63,6 +63,9 @@ if isinstance(child, Fragment): for event in child._generate(): yield event + elif isinstance(child, Stream): + for event in child: + yield event else: if not isinstance(child, basestring): child = unicode(child) diff --git a/genshi/tests/builder.py b/genshi/tests/builder.py --- a/genshi/tests/builder.py +++ b/genshi/tests/builder.py @@ -17,6 +17,7 @@ from genshi.builder import Element, tag from genshi.core import Attrs, Stream +from genshi.input import XML class ElementFactoryTestCase(unittest.TestCase): @@ -41,6 +42,15 @@ (None, -1, -1)), event) + def test_stream_as_child(self): + xml = list(tag.span(XML('Foo')).generate()) + self.assertEqual(5, len(xml)) + self.assertEqual((Stream.START, ('span', ()), (None, -1, -1)), xml[0]) + self.assertEqual((Stream.START, ('b', ()), (None, 1, 0)), xml[1]) + self.assertEqual((Stream.TEXT, 'Foo', (None, 1, 3)), xml[2]) + self.assertEqual((Stream.END, 'b', (None, 1, 6)), xml[3]) + self.assertEqual((Stream.END, 'span', (None, -1, -1)), xml[4]) + def suite(): suite = unittest.TestSuite()