utils/reparse_graph.py

#!/usr/bin/env python3

import lib.rr_graph.graph as graph
import lxml.etree as ET


def main():
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument("rr_graph_in")
    parser.add_argument("rr_graph_out")
    args = parser.parse_args()

    fn = args.rr_graph_in
    fn_out = args.rr_graph_out
    assert fn != fn_out

    print('Loading graph')
    g = graph.Graph(rr_graph_file=fn)
    print('Converting to XML')
    e = g.to_xml()
    print('Dumping')
    open(fn_out, 'w').write(ET.tostring(e, pretty_print=True).decode('ascii'))

    print('Exiting')


if __name__ == "__main__":
    main()