#!/usr/bin/python
# Brett Kelly 2018

import os
import traceback

from vdsm.hook import hooking

def main():
    romfile = os.environ.get('hba_romfile') 
    domxml = hooking.read_domxml()
    for hostdev in domxml.getElementsByTagName('hostdev'):
        rom = domxml.createElement('rom')
        rom.setAttribute('file', romfile)
        hostdev.appendChild(rom)
        hooking.write_domxml(domxml)


if __name__ == '__main__':
    try:
        if os.environ.has_key('hba_romfile'):
            main()
    except:
        hooking.exit_hook(
            'rom: %s' % (
                traceback.format_exc()
            ),
            return_code=1
        )