last step of repo switches

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@1716 ce84e21b-d406-0410-9b95-82705330c041

1 files changed, 198 insertions, 0 deletions

diff --git a/tools/convert-metadata.py b/tools/convert-metadata.py
new file mode 100755
index 000000000..012262ea3
--- /dev/null
+++ b/tools/convert-metadata.py
@@ -0,0 +1,198 @@
+#!/usr/bin/env python
+
+import glob
+import os
+import stat
+import sys
+import lxml.etree
+
+def create_if_noent(name):
+    if name[0] != '/':
+        name = os.getcwd() + '/' + name
+    for count in range(len(name.split('/')))[1:]:
+        partial = '/'.join(name.split('/')[:count+1])
+        try:
+            os.stat(partial)
+        except:
+            os.mkdir(partial)
+
+def pretty_print(element, level=0):
+    '''Produce a pretty-printed text representation of element'''
+    if isinstance(element, lxml.etree._Comment):
+        return (level * " ") + lxml.etree.tostring(element)
+    if element.text:
+        fmt = "%s<%%s %%s>%%s</%%s>" % (level*" ")
+        data = (element.tag, (" ".join(["%s='%s'" % (key, element.attrib[key]) for key in element.attrib])),
+                element.text, element.tag)
+    numchild = len(element.getchildren())
+    if numchild:
+        fmt = "%s<%%s %%s>\n" % (level*" ",) + (numchild * "%s") + "%s</%%s>\n" % (level*" ")
+        data = (element.tag, ) + (" ".join(["%s='%s'" % (key, element.attrib[key]) for key in element.attrib]),)
+        data += tuple([pretty_print(entry, level+2) for entry in element.getchildren()]) + (element.tag, )
+    else:
+        fmt = "%s<%%s %%s/>\n" % (level * " ")
+        data = (element.tag, " ".join(["%s='%s'" % (key, element.attrib[key]) for key in element.attrib]))
+    return fmt % data
+
+if __name__ == '__main__':
+    try:
+        [source, dest] = sys.argv[1:3]
+    except ValueError:
+        print "Usage: convert-metadata.py <oldpath> <newpath>"
+        raise SystemExit, 1
+    contained = {}
+    create_if_noent(dest)
+    mdata = lxml.etree.parse(source + '/etc/metadata.xml')
+    newm = lxml.etree.Element("Groups", version='3.0')
+    newc = lxml.etree.Element("Clients", version='3.0')
+    newi = lxml.etree.Element("Images", version='3.0')
+    names = [block.get('name') for block in mdata.getroot().findall('.//Profile')]
+    for block in mdata.getroot().getchildren():
+        if block.tag == 'Profile':
+            newp = lxml.etree.SubElement(newm, 'Group', profile='true', name=block.get('name'),
+                                         public=block.get('public', 'false'))
+        elif block.tag == 'Client':
+            lxml.etree.SubElement(newc, 'Client', name=block.get('name'), profile=block.get('profile'),
+                                  pingable='N', pingtime='0')
+            continue
+        elif block.tag == 'Class':
+            if block.get('name') in names:
+                print "%s is a dup profile/class; collapsing" % (block.get('name'))
+                newp = newm.xpath('./Group[@name="%s"]' % block.get('name'))[0]
+                try:
+                    oldincl = newp.xpath('./Group[@name="%s"]' % block.get('name'))[0]
+                    newp.remove(oldincl)
+                except:
+                    print "failed to locate old class inclusion for %s" % (block.get('name'))
+            else:
+                newp = lxml.etree.SubElement(newm, 'Group', profile='false', name=block.get('name'))
+        elif block.tag == 'Image':
+            newi.append(block)
+            continue
+        elif block.tag == None:
+            continue
+        else:
+            print "Unknown block tag %s" % block.tag
+            continue
+        for child in block.getchildren():
+            if child.tag == 'Class':
+                lxml.etree.SubElement(newp, 'Group', name=child.get('name'))
+                if contained.has_key(child.get('name')):
+                    contained[child.get('name')].append(block.get('name'))
+                else:
+                    contained[child.get('name')] = [block.get('name')]
+            elif child.tag == 'Attribute':
+                lxml.etree.SubElement(newp, 'Group', name=child.get('name'), scope=child.get('scope'))
+            elif child.tag == 'Bundle':
+                lxml.etree.SubElement(newp, "Bundle", name=child.get('name'))
+            else:
+                print "Unknown child tag %s" % child.tag
+
+    iinfo = lxml.etree.parse(source + '/etc/imageinfo.xml')
+    for system in iinfo.findall('./System'):
+        if system.get('name') == 'debian':
+            lxml.etree.SubElement(newm, 'Group', name=system.get('name'), toolset='debian')
+        elif system.get('name') == 'redhat':
+            lxml.etree.SubElement(newm, 'Group', name=system.get('name'), toolset='rh')
+        elif system.get('name') == 'solaris':
+            lxml.etree.SubElement(newm, 'Group', name=system.get('name'), toolset='solaris')
+        else:
+            lxml.etree.SubElement(newm, 'Group', name=system.get('name'))
+        for image in system.findall('./Image'):
+            newi = lxml.etree.SubElement(newm, 'Group', name=image.get('name'))
+            lxml.etree.SubElement(newi, 'Group', name=system.get('name'))
+
+    create_if_noent(dest)
+    create_if_noent(dest + '/Metadata')
+    open(dest + '/Metadata/groups.xml', 'w').write(pretty_print(newm))
+    open(dest + '/Metadata/clients.xml', 'w').write(pretty_print(newc))
+
+    print "*** Image interpolation is not performed automatically"
+
+    bsrc = source + '/Bundler/'
+    bdst = dest + '/Bundler/'
+    create_if_noent(bdst)
+    for bundle in glob.glob(bsrc + '*.xml'):
+        bname = bundle.split('/')[-1]
+        bdata = lxml.etree.parse(bundle).getroot()
+        for sys in bdata.findall('./System'):
+            sys.tag = 'Group'
+        open(bdst + bname, 'w').write(pretty_print(bdata))
+
+    os.system("rsync -a %s/SSHbase %s > /dev/null" % (source, dest))
+    create_if_noent("%s/etc" % ( dest))
+    os.system("cp %s/etc/*report*xml %s/etc" % (source, dest))
+    os.system("cp %s/etc/statistics.xml %s/etc" % (source, dest))
+
+    # handle base
+    basedata = lxml.etree.parse("%s/etc/base.xml" % source)
+    left = basedata.getroot().getchildren()
+    while left:
+        next = left.pop()
+        if next.tag in ['Image', 'Class']:
+            next.tag = 'Group'
+        left += next.getchildren()
+
+    create_if_noent("%s/Base" % dest)
+    open(dest + '/Base/converted.xml', 'w').write(pretty_print(basedata.getroot()))
+
+    # handle packages
+    create_if_noent("%s/Pkgmgr" % dest)
+    for pkgsrc in glob.glob("%s/Pkgmgr/*.xml" % source):
+        pname = pkgsrc.split('/')[-1]
+        pdata = lxml.etree.parse(pkgsrc).getroot()
+        image = pdata.get('image')
+        del pdata.attrib['image']
+        for loc in pdata.findall('./Location'):
+            loc.tag = 'Group'
+            loc.set('name', image)
+            if loc.attrib.has_key('uri'):
+                pdata.set('uri', loc.get('uri'))
+                del loc.attrib['uri']
+            if loc.attrib.has_key('type'):
+                pdata.set('type', loc.get('type'))
+                del loc.attrib['type']
+        pdata.set('priority', '0')
+        open(dest + '/Pkgmgr/' + pname, 'w').write(pretty_print(pdata))
+
+    # handle services.xml
+    svcdata = lxml.etree.parse("%s/etc/services.xml" % source)
+    left = svcdata.getroot().getchildren()
+    while left:
+        next = left.pop()
+        if next.tag in ['Image', 'Class']:
+            next.tag = 'Group'
+        left += next.getchildren()
+
+    svcdata.getroot().set('priority', '0')
+    create_if_noent("%s/Svcmgr" % dest)
+    open(dest + '/Svcmgr/converted.xml', 'w').write(pretty_print(svcdata.getroot()))
+
+    # handle Cfg
+    os.chdir("%s/Cfg" % source)
+    p = os.popen("find . -depth -type d -print")
+    path = p.readline().strip()
+    paths = []
+    while path:
+        paths.append(path)
+        create_if_noent("%s/Cfg/%s" % (dest, path))
+        path = p.readline().strip()
+    # paths are now created
+    for path in paths:
+        for filename in os.listdir(path):
+            if stat.S_ISDIR(os.stat(path + '/' + filename)[stat.ST_MODE]):
+                continue
+            if filename in [':info', path.split('/')[-1]]:
+                os.system("cp -f %s/%s %s/Cfg/%s/%s" % (path, filename, dest, path, filename))
+            else:
+                meta = filename[len(path.split('/')[-1])+1:]
+                if meta[:2] == 'H_':
+                    os.system("cp -f %s/%s %s/Cfg/%s/%s" % (path, filename, dest, path, filename))
+                elif meta[:1] in ['C', 'B', 'I']:
+                    print filename, "moved"
+                    meta = 'G' + meta[1:]
+                    os.system("cp -f %s/%s %s/Cfg/%s/%s.%s" % (path, filename, dest,
+                                                                    path, path.split('/')[-1], meta))
+                else:
+                    print "=========> don't know what to do with %s/%s" % (path, filename)
+