makeiso helper script

makeiso.py

# python ~/makeiso.py payload.exe
try:
    from cStringIO import StringIO as BytesIO
except ImportError:
    from io import BytesIO

import pycdlib
import sys
args = sys.argv[1:]
iso = pycdlib.PyCdlib()
iso.new(interchange_level=4)

targetfilenameFirst = args[0]

targetfilename = '{}'.format(targetfilenameFirst)
targetfilehandle = open(targetfilename, 'rb')
targetfilebody = targetfilehandle.read()

iso.add_fp(BytesIO(targetfilebody), len(targetfilebody), '/' + targetfilename + ';1')

iso.write('{}.iso'.format(targetfilenameFirst))
iso.close()

targetfilehandle.close()
print('{}.iso has been created!'.format(targetfilenameFirst))