bonobo_xml_example.py

import bonobo
from bonobo.config import ContextProcessor, use_context
from bonobo.constants import NOT_MODIFIED


def extract():
    yield 'hello'
    yield 'world'


@use_context
class MyXmlWriter(bonobo.FileWriter):
    @ContextProcessor
    def xml(self, context, file, *args, **kwargs):
        file.write('<document>\n')
        yield
        file.write('</document>\n')

    def __call__(self, file, context, line, *, fs):
        file.write('  <line>'+line+'</line>\n')
        return NOT_MODIFIED


def get_graph(**options):
    graph = bonobo.Graph()
    graph.add_chain(extract, MyXmlWriter('out.xml'))
    return graph


if __name__ == '__main__':
    parser = bonobo.get_argument_parser()
    with bonobo.parse_args(parser) as options:
        bonobo.run(
            get_graph(**options),
        )