pandemonium_engine/doc/tools/doc_merge.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import xml.etree.ElementTree as ET


tree = ET.parse(sys.argv[1])
old_doc = tree.getroot()

tree = ET.parse(sys.argv[2])
new_doc = tree.getroot()

f = file(sys.argv[3], "wb")
tab = 0

old_classes = {}


def write_string(_f, text, newline=True):
    for t in range(tab):
        _f.write("\t")
    _f.write(text)
    if newline:
        _f.write("\n")


def escape(ret):
    ret = ret.replace("&", "&")
    ret = ret.replace("<", "&gt;")
    ret = ret.replace(">", "&lt;")
    ret = ret.replace("'", "&apos;")
    ret = ret.replace('"', "&quot;")
    return ret


def inc_tab():
    global tab
    tab += 1


def dec_tab():
    global tab
    tab -= 1


write_string(f, '<?xml version="1.0" encoding="UTF-8" ?>')
write_string(f, '<doc version="' + new_doc.attrib["version"] + '">')


def get_tag(node, name):
    tag = ""
    if name in node.attrib:
        tag = " " + name + '="' + escape(node.attrib[name]) + '" '
    return tag


def find_method_descr(old_class, name):

    methods = old_class.find("methods")
    if methods != None and len(list(methods)) > 0:
        for m in list(methods):
            if m.attrib["name"] == name:
                description = m.find("description")
                if description != None and description.text.strip() != "":
                    return description.text

    return None


def find_signal_descr(old_class, name):

    signals = old_class.find("signals")
    if signals != None and len(list(signals)) > 0:
        for m in list(signals):
            if m.attrib["name"] == name:
                description = m.find("description")
                if description != None and description.text.strip() != "":
                    return description.text

    return None


def find_constant_descr(old_class, name):

    if old_class is None:
        return None
    constants = old_class.find("constants")
    if constants != None and len(list(constants)) > 0:
        for m in list(constants):
            if m.attrib["name"] == name:
                if m.text.strip() != "":
                    return m.text
    return None


def write_class(c):
    class_name = c.attrib["name"]
    print("Parsing Class: " + class_name)
    if class_name in old_classes:
        old_class = old_classes[class_name]
    else:
        old_class = None

    category = get_tag(c, "category")
    inherits = get_tag(c, "inherits")
    write_string(f, '<class name="' + class_name + '" ' + category + inherits + ">")
    inc_tab()

    write_string(f, "<brief_description>")

    if old_class != None:
        old_brief_descr = old_class.find("brief_description")
        if old_brief_descr != None:
            write_string(f, escape(old_brief_descr.text.strip()))

    write_string(f, "</brief_description>")

    write_string(f, "<description>")
    if old_class != None:
        old_descr = old_class.find("description")
        if old_descr != None:
            write_string(f, escape(old_descr.text.strip()))

    write_string(f, "</description>")

    methods = c.find("methods")
    if methods != None and len(list(methods)) > 0:

        write_string(f, "<methods>")
        inc_tab()

        for m in list(methods):
            qualifiers = get_tag(m, "qualifiers")

            write_string(f, '<method name="' + escape(m.attrib["name"]) + '" ' + qualifiers + ">")
            inc_tab()

            for a in list(m):
                if a.tag == "return":
                    typ = get_tag(a, "type")
                    write_string(f, "<return" + typ + ">")
                    write_string(f, "</return>")
                elif a.tag == "argument":

                    default = get_tag(a, "default")

                    write_string(
                        f,
                        '<argument index="'
                        + a.attrib["index"]
                        + '" name="'
                        + escape(a.attrib["name"])
                        + '" type="'
                        + a.attrib["type"]
                        + '"'
                        + default
                        + ">",
                    )
                    write_string(f, "</argument>")

            write_string(f, "<description>")
            if old_class != None:
                old_method_descr = find_method_descr(old_class, m.attrib["name"])
                if old_method_descr:
                    write_string(f, escape(escape(old_method_descr.strip())))

            write_string(f, "</description>")
            dec_tab()
            write_string(f, "</method>")
        dec_tab()
        write_string(f, "</methods>")

    signals = c.find("signals")
    if signals != None and len(list(signals)) > 0:

        write_string(f, "<signals>")
        inc_tab()

        for m in list(signals):

            write_string(f, '<signal name="' + escape(m.attrib["name"]) + '">')
            inc_tab()

            for a in list(m):
                if a.tag == "argument":

                    write_string(
                        f,
                        '<argument index="'
                        + a.attrib["index"]
                        + '" name="'
                        + escape(a.attrib["name"])
                        + '" type="'
                        + a.attrib["type"]
                        + '">',
                    )
                    write_string(f, "</argument>")

            write_string(f, "<description>")
            if old_class != None:
                old_signal_descr = find_signal_descr(old_class, m.attrib["name"])
                if old_signal_descr:
                    write_string(f, escape(old_signal_descr.strip()))
            write_string(f, "</description>")
            dec_tab()
            write_string(f, "</signal>")
        dec_tab()
        write_string(f, "</signals>")

    constants = c.find("constants")
    if constants != None and len(list(constants)) > 0:

        write_string(f, "<constants>")
        inc_tab()

        for m in list(constants):

            write_string(f, '<constant name="' + escape(m.attrib["name"]) + '" value="' + m.attrib["value"] + '">')
            old_constant_descr = find_constant_descr(old_class, m.attrib["name"])
            if old_constant_descr:
                write_string(f, escape(old_constant_descr.strip()))
            write_string(f, "</constant>")

        dec_tab()
        write_string(f, "</constants>")

    dec_tab()
    write_string(f, "</class>")


for c in list(old_doc):
    old_classes[c.attrib["name"]] = c

for c in list(new_doc):
    write_class(c)
write_string(f, "</doc>\n")