#!/usr/bin/python

import os
import sys
import hooking
import traceback


'''
vmname inject vdsm hook
====================
hook is getting VM name and writes it into bios 'SKU Number' field.

'''

if True:
    try:
        domxml = hooking.read_domxml()
        names = domxml.getElementsByTagName('name')[0]
        vmname = names.childNodes[0].nodeValue
        system = domxml.getElementsByTagName('system')[0]
        entry = domxml.createElement('entry')
        entry.setAttribute('name','sku')
        text = domxml.createTextNode(vmname)
        entry.appendChild(text)
        system.appendChild(entry)
        hooking.write_domxml(domxml)

    except:
        sys.stderr.write('name-inject: [unexpected error]: %s\n' %
                         traceback.format_exc())
        sys.exit(2)