Ticket #79: ElementTreeStream.py

from genshi.core import Attrs, Stream, QName

class ElementTreeStream(Stream):
        """
        Adapted from function ET in genshi.template from Genshi 0.3.4.
        
        This allows you to select(...) the resulting stream. It also unwraps lists
        passed in, specifically to handle the result of element.findall(...).
        """
        
        @classmethod
        def convert(cls, element):
                tag_name = element.tag
                if tag_name.startswith('{'):
                        tag_name = tag_name[1:]
                tag_name = QName(tag_name)
                attrib = Attrs(element.items())
                
                yield (Stream.START, (tag_name, attrib), (None, -1, -1))
                if element.text:
                        yield Stream.TEXT, element.text, (None, -1, -1)
                for child in element.getchildren():
                        for item in cls.convert(child):
                                yield item
                yield Stream.END, tag_name, (None, -1, -1)
                if element.tail:
                        yield Stream.TEXT, element.tail, (None, -1, -1)
        #
        
        def __iter__(self):
                if isinstance(self.events, list):
                        from itertools import chain
                        return chain(*(self.convert(element) for element in self.events))
                
                else:
                        return self.convert(self.events)
        #
#



# End