backup-sr-metadata.py

#!/usr/bin/python
# Back up the SR metadata and VDI list into an XML file
# (c) Anil Madhavapeddy, Citrix Systems Inc, 2008

import atexit
import XenAPI
import sys
import getopt
from xml.dom.minidom import Document

def logout():
    try:
        session.xenapi.session.logout()
    except:
        pass
atexit.register(logout)

def usage():
    print >> sys.stderr, "%s [-f <output file>]" % sys.argv[0]
    sys.exit(1)

def set_if_exists(xml, record, key):
    if record.has_key(key):
        xml.setAttribute(key, record[key])
    else:
        xml.setAttribute(key, "")
   
def main(argv):
    session = XenAPI.xapi_local()
    session.xenapi.login_with_password("", "")

    try:
        opts, args = getopt.getopt(argv, "hf:", [])
    except getopt.GetoptError, err:
        print str(err)
        usage()

    outfile = None
    for o,a in opts:
        if o == "-f":
            outfile = a

    if outfile == None:
        usage()

    f = open(outfile, 'w')

    srs = session.xenapi.SR.get_all_records()
    vdis = session.xenapi.SR.get_all_records()
 
    doc = Document()

    metaxml = doc.createElement("meta")
    doc.appendChild(metaxml)

    for srref in srs.keys():
        srrec = srs[srref]
        srxml = doc.createElement("sr")
        set_if_exists(srxml, srrec, 'uuid')
        set_if_exists(srxml, srrec, 'name_label')
        set_if_exists(srxml, srrec, 'name_description')
 
        for vdiref in srrec['VDIs']:
            try: 
                vdirec = session.xenapi.VDI.get_record(vdiref)
                vdixml = doc.createElement("vdi")
                set_if_exists(vdixml, vdirec, 'uuid')
                set_if_exists(vdixml, vdirec, 'name_label')
                set_if_exists(vdixml, vdirec, 'name_description')
                srxml.appendChild(vdixml)
            except:
                print >> sys.stderr, "Failed to get VDI record for: %s" % vdiref
 
        metaxml.appendChild(srxml)

    doc.writexml(f)
    f.close()
    session.xenapi.logout()

if __name__ == "__main__":
    main(sys.argv[1:])