xml_process_help.py

import sys
import pprint
from optparse import OptionParser
from xml.etree.ElementTree import ElementTree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import SubElement as SE
import xml.dom.minidom as minidom

def migrate_aosp_to_snapshot(xml_input, aosp_remote, xml_output):
    tree = ElementTree(file=xml_input)
    root = tree.getroot()

    for child in root:
        if (child.tag == "remote"):
            print child.attrib["name"]
            aosp_remote = child.attrib["name"]
            fetch_str = child.attrib["fetch"]
            fetch_str = fetch_str.replace("a/aosp/", "")
            child.set("fetch", fetch_str)
            continue

        if (child.tag == "project"):
            print child.attrib["name"]
            real_path = child.attrib["name"]
            if not child.attrib.has_key("path"):
                child.set("path", real_path)

            if not child.attrib.has_key("remote"):
                child.set("remote", aosp_remote)

            real_path = "a/aosp/" + real_path
            child.set("name", real_path)
            continue

        if (child.tag == "default"):
            print child.tag
            default_ele = child
            continue

    root.remove(default_ele)
    tree.write(xml_output, 'utf-8')

def migrate_snapshot_to_otc_upstream(xml_input, aosp_remote, xml_output):
    tree = ElementTree(file=xml_input)
    root = tree.getroot()

    for child in root:
        if (child.tag == "remote"):
            remote_ele = child
            continue

        if (child.tag == "project"):
            print child.attrib["name"]
            real_path = child.attrib["name"]
            if not child.attrib.has_key("path"):
                child.set("path", real_path)

            if child.attrib.has_key("revision"):
                del child.attrib["revision"]

            child.set("name", real_path)
            continue

        if (child.tag == "default"):
            print child.tag
            default_ele = child
            continue

    tree.write(xml_output, 'utf-8')


def modify_groups(xml_input, xml_output):
    aosp_tree = ElementTree(file=xml_input)
    aosp_root = aosp_tree.getroot()
    for child in aosp_root:
        if (child.tag == "project"):
                    groups = child.attrib["groups"]
                    print "groups =     " + groups
                    real_groups = "aosp," + groups
                    print "real_group= " + real_groups
                    child.set("groups", real_groups)
#write to file
    aosp_tree.write(xml_output, 'utf-8')

def remove_revision(xml_input, xml_output):
    tree = ElementTree(file=xml_input)
    root = tree.getroot()

    for child in root:
        if (child.tag == "project"):
            print child.attrib["name"]
            if child.attrib.has_key("revision"):
                del child.attrib["revision"]
    tree.write(xml_output, 'utf-8')

def remove_remote(xml_input, xml_output):
    tree = ElementTree(file=xml_input)
    root = tree.getroot()

    for child in root:
        if (child.tag == "project"):
            print child.attrib["name"]
            if child.attrib.has_key("remote"):
                del child.attrib["remote"]
    tree.write(xml_output, 'utf-8')

def remove_groups(xml_input, xml_output):
    tree = ElementTree(file=xml_input)
    root = tree.getroot()

    for child in root:
        if (child.tag == "project"):
            print child.attrib["name"]
            if child.attrib.has_key("groups"):
                del child.attrib["groups"]
    tree.write(xml_output, 'utf-8')

def main():
    usage = "usage: %prog [options] arg"
    parser = OptionParser(usage)
    parser.add_option("-i", "--in_file", dest="xml_input",
                        help="xml input file")
    parser.add_option("-j", "--in_file2", dest="xml_input2",
                        help="xml input file")
    parser.add_option("-r", "--remot", dest="aosp_remote",
                        help="aosp xml remote")
    parser.add_option("-o", "--out_file", dest="xml_output",
                        help="xml output file")
    parser.add_option("-t", "--type", dest="mig_type",
                        help="migration type"
                        "'aosp_to_snapshot' means migrate MCG AOSP mirror to snapshot"
                        "'snapshot_to_otc' means migrate snapshot to otc upstream branch"
                        "'del_groups' mean del groups"
                        "'del_remote' mean del remote"
                        "'modify_groups' mean modify groups"
                        "'del_revision' means delete revision")

    (options, args) = parser.parse_args()

    if options.mig_type == "aosp_to_snapshot":
        migrate_aosp_to_snapshot(options.xml_input, options.aosp_remote, options.xml_output);
    elif options.mig_type == "snapshot_to_otc":
        migrate_snapshot_to_otc_upstream(options.xml_input, options.aosp_remote, options.xml_output);
    elif options.mig_type == "del_revision":
        remove_revision(options.xml_input, options.xml_output);
    elif options.mig_type == "del_groups":
        remove_groups(options.xml_input, options.xml_output);
    elif options.mig_type == "del_remote":
        remove_remote(options.xml_input, options.xml_output);
    elif options.mig_type == "modify_groups":
        modify_groups(options.xml_input, options.xml_output);
if __name__ == "__main__":
    main()