def __init__(self):
QtGui.QDialog.__init__(self)
global version
global zip
self.ui = Ui_Dialog()
self.ui.setupUi(self)
self.ui.close.clicked.connect(self.on_close_Click)
self.ui.browse_iso.clicked.connect(self.browse_iso)
self.ui.detect_usb.clicked.connect(self.onRefereshClick)
self.ui.create.clicked.connect(self.onCreateClick)
self.ui.uninstall.clicked.connect(self.uninstall_distro)
self.ui.comboBox.activated[str].connect(self.onComboChange)
# Syslinux Tab
self.ui.install_syslinux.clicked.connect(self.onInstall_syslinuxClick)
self.ui.edit_syslinux.clicked.connect(self.onedit_syslinux)
# QEMU Tab
self.ui.browse_iso_qemu.clicked.connect(self.on_Qemu_Browse_iso_Click)
self.ui.boot_iso_qemu.clicked.connect(self.on_Qemu_Boot_iso_Click)
self.ui.boot_usb_qemu.clicked.connect(self.on_Qemu_Boot_usb_Click)
self.ui.tabWidget.removeTab(3)
version = open(resource_path(os.path.join("tools", "version.txt")), 'r').read().strip()
var.gbl_mbusb_version = version
if sys.platform.startswith("linux"):
zip = resource_path(os.path.join("tools", "7zip", "linux", "7z"))
var.zip = zip
else:
zip = resource_path(os.path.join("tools", "7zip", "windows", "7z.exe"))
var.zip = zip
if sys.platform.startswith("linux"):
if os.geteuid() != 0:
if os.system('which sudo') == 0:
for x in xrange(3):
input, ok = QtGui.QInputDialog.getText(self, 'Password',
'Enter user var.password::', QtGui.QLineEdit.Password)
if ok:
var.password = str(input)
if os.popen('echo ' + var.password + ' | sudo -S id -u').read().strip() == '0':
break
if x == 2:
print "You have entered wrong password 3 times. Exiting now. "
sys.exit(0)
else:
print "Password not entered. Exiting now. "
sys.exit(0)
elif os.system('which gksu') == 0:
os.system("gksu -d " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which gksudo') == 0:
os.system("gksudo -d " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which kdesu') == 0:
os.system("kdesu -t " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which kdesudo') == 0:
os.system("kdesudo -t " + sys.executable + " " + sys.argv[0])
sys.exit(0)
else:
QtGui.QMessageBox.information(self, 'No root...',
'multibootusb require Please install sudo, gksu, kdesu, gksudo or kdesudo then restart multibootusb.')
sys.exit(0)
detected_device = self.find_usb()
for device in detected_device:
self.ui.comboBox.addItem(str(device))
if self.ui.comboBox.currentText():
self.onComboChange()
class AppGui(qemu.AppGui, detect_iso.AppGui, update_cfg.AppGui, uninstall_distro.AppGui, install_syslinux.AppGui,
QtGui.QDialog, Ui_Dialog):
def __init__(self):
QtGui.QDialog.__init__(self)
global version
global zip
self.ui = Ui_Dialog()
self.ui.setupUi(self)
self.ui.close.clicked.connect(self.on_close_Click)
self.ui.browse_iso.clicked.connect(self.browse_iso)
self.ui.detect_usb.clicked.connect(self.onRefereshClick)
self.ui.create.clicked.connect(self.onCreateClick)
self.ui.uninstall.clicked.connect(self.uninstall_distro)
self.ui.comboBox.activated[str].connect(self.onComboChange)
# Syslinux Tab
self.ui.install_syslinux.clicked.connect(self.onInstall_syslinuxClick)
self.ui.edit_syslinux.clicked.connect(self.onedit_syslinux)
# QEMU Tab
self.ui.browse_iso_qemu.clicked.connect(self.on_Qemu_Browse_iso_Click)
self.ui.boot_iso_qemu.clicked.connect(self.on_Qemu_Boot_iso_Click)
self.ui.boot_usb_qemu.clicked.connect(self.on_Qemu_Boot_usb_Click)
self.ui.tabWidget.removeTab(3)
version = open(resource_path(os.path.join("tools", "version.txt")), 'r').read().strip()
var.gbl_mbusb_version = version
if sys.platform.startswith("linux"):
zip = resource_path(os.path.join("tools", "7zip", "linux", "7z"))
var.zip = zip
else:
zip = resource_path(os.path.join("tools", "7zip", "windows", "7z.exe"))
var.zip = zip
if sys.platform.startswith("linux"):
if os.geteuid() != 0:
if os.system('which sudo') == 0:
for x in xrange(3):
input, ok = QtGui.QInputDialog.getText(self, 'Password',
'Enter user var.password::', QtGui.QLineEdit.Password)
if ok:
var.password = str(input)
if os.popen('echo ' + var.password + ' | sudo -S id -u').read().strip() == '0':
break
if x == 2:
print "You have entered wrong password 3 times. Exiting now. "
sys.exit(0)
else:
print "Password not entered. Exiting now. "
sys.exit(0)
elif os.system('which gksu') == 0:
os.system("gksu -d " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which gksudo') == 0:
os.system("gksudo -d " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which kdesu') == 0:
os.system("kdesu -t " + sys.executable + " " + sys.argv[0])
sys.exit(0)
elif os.system('which kdesudo') == 0:
os.system("kdesudo -t " + sys.executable + " " + sys.argv[0])
sys.exit(0)
else:
QtGui.QMessageBox.information(self, 'No root...',
'multibootusb require Please install sudo, gksu, kdesu, gksudo or kdesudo then restart multibootusb.')
sys.exit(0)
detected_device = self.find_usb()
for device in detected_device:
self.ui.comboBox.addItem(str(device))
if self.ui.comboBox.currentText():
self.onComboChange()
def onRefereshClick(self):
detected_device = self.find_usb()
for device in detected_device:
self.ui.comboBox.addItem(str(device))
if self.ui.comboBox.currentText():
self.onComboChange()
self.onComboChange()
def onComboChange(self):
global usb_device
global usb_file_system
global usb_mount
global sys_cfg_file
global usb_size_avail
global usb_size_total
global usb_size_used
global usb_uuid
global usb_label
usb_details = self.get_usb_details()
usb_device = str(usb_details[0])
var.gbl_usb_device = usb_device
usb_uuid = str(usb_details[1])
var.gbl_usb_uuid = usb_uuid
usb_label = str(usb_details[2])
var.usb_label = usb_label
usb_mount = str(usb_details[3])
var.usb_mount = usb_mount
usb_file_system = str(usb_details[4])
var.usb_file_system = usb_file_system
usb_size_total = int(usb_details[5])
usb_size_avail = int(usb_details[6])
usb_size_used = int(usb_details[7])
sys_cfg_file = os.path.join(str(usb_details[3]), "multibootusb", "syslinux.cfg")
var.gbl_sys_cfg_file = sys_cfg_file
self.update_list_box(sys_cfg_file)
def browse_iso(self):
self.ui.lineEdit.clear()
iso_link = QtGui.QFileDialog.getOpenFileName(self, 'Select an iso...', "", "ISO Files (*.iso)")
if iso_link:
self.ui.lineEdit.insert(iso_link)
else:
print ("File not selected.")
def onCreateClick(self):
self.create_mbusb()
def copytree(self, src, dst, symlinks=False, ignore=None):
for item in os.listdir(src):
s = os.path.join(src, item)
d = os.path.join(dst, item)
if os.path.isdir(s):
shutil.copytree(s, d, symlinks, ignore)
else:
shutil.copy2(s, d)
def find_usb(self):
found = []
if sys.platform.startswith("linux"):
bus = dbus.SystemBus()
ud_manager_obj = bus.get_object("org.freedesktop.UDisks", "/org/freedesktop/UDisks")
ud_manager = dbus.Interface(ud_manager_obj, 'org.freedesktop.UDisks')
for dev in ud_manager.EnumerateDevices():
device_obj = bus.get_object("org.freedesktop.UDisks", dev)
device_props = dbus.Interface(device_obj, dbus.PROPERTIES_IFACE)
if device_props.Get('org.freedesktop.UDisks.Device',
"DriveConnectionInterface") == "usb" and device_props.Get(
'org.freedesktop.UDisks.Device', "DeviceIsPartition"):
if device_props.Get('org.freedesktop.UDisks.Device', "DeviceIsMounted"):
device_file = device_props.Get('org.freedesktop.UDisks.Device', "DeviceFile")
found.append(device_file)
#else:
# print "Device not mounted"
return found
else:
oFS = win32com.client.Dispatch("Scripting.FileSystemObject")
oDrives = oFS.Drives
for drive in oDrives:
if drive.DriveType == 1 and drive.IsReady:
found.append(drive)
return found
def get_usb_details(self):
if sys.platform.startswith("linux"):
selected_usb_part = str(self.ui.comboBox.currentText())[4:]
bus = dbus.SystemBus()
device_obj = bus.get_object("org.freedesktop.UDisks", "/org/freedesktop/UDisks/devices" + selected_usb_part)
device_props = dbus.Interface(device_obj, dbus.PROPERTIES_IFACE)
selected_usb_device = device_props.Get('org.freedesktop.UDisks.Device', "DeviceFile")
selected_usb_uuid = device_props.Get('org.freedesktop.UDisks.Device', "IdUuid")
selected_usb_label = device_props.Get('org.freedesktop.UDisks.Device', "IdLabel")
selected_usb_mount_path = device_props.Get('org.freedesktop.UDisks.Device', "DeviceMountPaths")[0]
#selected_usb_size = device_props.Get('org.freedesktop.UDisks.Device', "PartitionSize")
selected_usb_file_system = device_props.Get('org.freedesktop.UDisks.Device', "IdType")
selected_usb_total_size = psutil.disk_usage(selected_usb_mount_path)[0]
selected_usb_avail_size = psutil.disk_usage(selected_usb_mount_path)[2]
selected_usb_used_size = psutil.disk_usage(selected_usb_mount_path)[1]
else:
selected_usb_part = str(self.ui.comboBox.currentText())[:2]
oFS = win32com.client.Dispatch("Scripting.FileSystemObject")
d = oFS.GetDrive(oFS.GetDriveName(oFS.GetAbsolutePathName(selected_usb_part)))
selected_usb_device = d.DriveLetter
serno = "%X" % (long(d.SerialNumber) & 0xFFFFFFFF)
selected_usb_uuid = serno[:4] + '-' + serno[4:]
selected_usb_label = d.VolumeName
selected_usb_mount_path = selected_usb_device + ":\\"
selected_usb_file_system = d.FileSystem
selected_usb_total_size = psutil.disk_usage(selected_usb_mount_path)[0]
selected_usb_avail_size = psutil.disk_usage(selected_usb_mount_path)[2]
selected_usb_used_size = psutil.disk_usage(selected_usb_mount_path)[1]
self.ui.usb_size_ttl.setText("Filesystem :: " + selected_usb_file_system)
self.ui.usb_size_avl.setText("Size :: " + str(self.bytes2human(psutil.disk_usage(selected_usb_mount_path)[0])))
self.ui.usb_label.setText("Label :: " + selected_usb_label)
self.ui.usb_dev.setText("USB Device :: " + selected_usb_device)
self.ui.usb_mount.setText("Mount :: " + str(selected_usb_mount_path))
var.usb_mount_count = len(str(self.ui.usb_mount.text()[9:]))
print "USB mount count is " + str(var.usb_mount_count)
#[0] = selected_usb_device
#[1] = selected_usb_uuid
#[2] = selected_usb_label
#[3] = selected_usb_mount_path
#[4] = selected_usb_file_system
#[5 = selected_usb_total_size
#[6] = selected_usb_avail_size
#[7] = selected_usb_used_size
return (
selected_usb_device, selected_usb_uuid, selected_usb_label, selected_usb_mount_path, selected_usb_file_system,
selected_usb_total_size, selected_usb_avail_size, selected_usb_used_size)
def update_list_box(self, sys_cfg_file):
if sys_cfg_file:
self.ui.listWidget.clear()
if os.path.exists(sys_cfg_file):
for line in open(sys_cfg_file):
if "#start " in line:
installed_distro = (line)[7:]
self.ui.listWidget.addItem(installed_distro)
def create_mbusb(self):
global required_syslinux_install
if not self.ui.comboBox.currentText():
QtGui.QMessageBox.information(self, 'No USB...',
'No USB found. (Step 1)\n\nInsert USB disk and restart multibootusb')
self.ui.lineEdit.clear()
elif not self.ui.lineEdit.text():
QtGui.QMessageBox.information(self, 'No ISO...',
'No ISO selected. (Step 2)\n\nPlease choose an iso and click create')
else:
iso_path = str(self.ui.lineEdit.text())
var.iso9660fs = ISO9660(iso_path)
print self.ui.lineEdit.text()
if platform.system() == "Windows":
iso_path = iso_path.replace("/", "\\")
iso_name = os.path.basename(iso_path)
self.ui.lineEdit.clear()
mbusb_dir_content = resource_path(os.path.join("tools", "multibootusb"))
iso_size = int(os.path.getsize(iso_path))
mbusb_usb_dir = os.path.join(str(usb_mount), "multibootusb")
install_dir = os.path.join(mbusb_usb_dir, os.path.splitext(iso_name)[0])
var.install_dir = install_dir
var.install_dir_count = len(var.install_dir)
print "Testing integrity of " + iso_name
self.ui.status.setText("Testing integrity of " + iso_name)
QtGui.qApp.processEvents()
# Somebody can help me to animate this image...
"""
movie = QtGui.QMovie(resource_path(os.path.join("tools","checking.gif")))
movie.setCacheMode(QtGui.QMovie.CacheAll)
movie.setSpeed(100)
self.ui.status.setMovie(movie)
#self.ui.status.setPixmap(QtGui.QPixmap(resource_path(os.path.join("tools","checking.gif"))))
movie.start()
"""
#if subprocess.call(zip + " t " + iso_path, shell=True) != 0:
if not var.iso9660fs.checkIntegrity():
error_7zip = "yes"
distro = None
print "Integrity check fail..."
else:
# Extract necessary files to find distro...
print "Integrity passed..."
var.iso9660fs.writeDir("/", iso_cfg_ext_dir, ".cfg")
var.iso_file_list = var.iso9660fs.readDir("/")
#print var.iso_file_list
if not any("isolinux.bin" in s for s in var.iso_file_list):
#if not os.path.exists(os.path.join(iso_cfg_ext_dir, "isolinux.bin")):
var.distro_isolinux_exist = "no"
print "isolinux.bin not found on the distro..."
else:
for f in var.iso_file_list:
if 'isolinux.bin' in f:
var.distro_isolinux_bin_path = "/multibootusb" + os.path.dirname(f)
print os.path.dirname(f)
# Get the list of files from iso. So that later on it can be used for detecting other distros.
#var.iso_file_content = os.popen(zip + ' l ' + iso_path).read()
error_7zip = "no"
self.ui.status.clear()
distro = self.detect_iso(iso_cfg_ext_dir)
#print var.iso_file_content
if var.cfg_read_err == "yes":
QtGui.QMessageBox.information(self, 'Read Error...',
'Could not read config files to identify distro.')
else:
if not distro:
if re.search(r'sources', var.iso_file_list, re.I):
distro = "windows"
#self.detect_iso_zip_info()
if distro:
var.distro = distro
print var.distro
if error_7zip == "yes":
self.ui.status.setText("Integrity check failed on " + iso_name)
QtGui.QMessageBox.information(self, 'Integrity error...',
'Please check the integrity of downloaded ISO.')
self.ui.status.clear()
elif not var.distro:
QtGui.QMessageBox.information(self, 'No support...',
'Sorry. ' + iso_name + ' is not supported at the moment\n\nPlease email this issue to [email protected]')
elif iso_size > usb_size_avail:
print iso_size
print usb_size_avail
QtGui.QMessageBox.information(self, 'No Space...',
'Sorry.\n\nThere is no space available on ' + usb_device)
else:
if os.path.exists(install_dir):
QtGui.QMessageBox.information(self, 'Already exist...',
iso_name + ' is already installed on ' + usb_device)
else:
reply = QtGui.QMessageBox.question(self, 'Review selection...',
'Selected USB disk:: %s\n' % usb_device +
'USB mount point:: %s\n' % usb_mount +
'Selected distro:: %s\n\n' % iso_name +
'Would you like to install the selected distro?',
QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
if not os.path.exists(mbusb_usb_dir):
required_syslinux_install = 'yes'
shutil.copytree(mbusb_dir_content, os.path.join(str(usb_mount), "multibootusb"))
else:
required_syslinux_install = 'no'
os.makedirs(install_dir)
#out_dir = "-o" + install_dir
inintial_size = os.path.getsize(iso_path)
self.ui.status.setText("Installing " + iso_name)
def copy_process():
if var.distro == "opensuse":
var.iso9660fs.writeDir("/", install_dir, "boot")
if platform.system() == "Windows":
var.extract_file_name = "Copying " + iso_name
subprocess.call(["xcopy",iso_path,var.usb_mount], shell=True)
else:
var.extract_file_name = "Copying " + iso_name
shutil.copy(iso_path, var.usb_mount)
elif var.distro == "windows":
var.iso9660fs.writeDir("/", var.usb_mount)
elif var.distro == "ipfire":
var.iso9660fs.writeDir("/", install_dir, "boot")
var.iso9660fs.writeDir("/", var.usb_mount, ".tlz")
elif var.distro == "zenwalk" or var.distro == "zenwalk":
var.iso9660fs.writeDir("/", install_dir, "kernel")
var.iso9660fs.writeDir("/", install_dir, "isolinux")
if platform.system() == "Windows":
var.extract_file_name = "Copying " + iso_name
subprocess.call(["xcopy",iso_path,install_dir], shell=True)
else:
var.extract_file_name = "Copying " + iso_name
shutil.copy(iso_path, install_dir)
elif var.distro == "pclinuxos":
var.iso9660fs.writeDir("/", install_dir, "isolinux")
if platform.system() == "Windows":
var.extract_file_name = "Copying " + iso_name
subprocess.call(["xcopy",iso_path,install_dir], shell=True)
else:
var.extract_file_name = "Copying " + iso_name
shutil.copy(iso_path, install_dir)
elif var.distro == "salix-live":
var.iso9660fs.writeDir("/", install_dir, "boot")
if platform.system() == "Windows":
var.extract_file_name = "Copying " + iso_name
if subprocess.call("xcopy " + iso_path + " " + install_dir, shell=True):
print "ISO copied to " + install_dir
else:
var.extract_file_name = "Copying " + iso_name
shutil.copy(iso_path, install_dir)
else:
var.iso9660fs.writeDir("/", install_dir)
inintial_usb_size = int(psutil.disk_usage(usb_mount)[1])
thrd = threading.Thread(target=copy_process, name="copy_process")
thrd.start()
while thrd.is_alive():
current_size = int(psutil.disk_usage(usb_mount)[1])
diff_size = int(inintial_usb_size - current_size)
percentage = float(1.0 * diff_size) / inintial_size * 100
self.ui.status.setText(var.extract_file_name)
self.ui.progressBar.setValue(abs(percentage))
QtGui.qApp.processEvents()
print "All Completed..."
self.ui.progressBar.setValue(100)
self.ui.progressBar.setValue(0)
sys_cfg_file = os.path.join(str(usb_mount), "multibootusb", "syslinux.cfg")
self.install_syslinux(usb_device)
self.update_distro_cfg_files(distro, iso_name, install_dir)
self.update_list_box(sys_cfg_file)
self.ui.status.clear()
if sys.platform.startswith("linux"):
os.system('sync')
for files in os.listdir(iso_cfg_ext_dir):
if os.path.isdir(os.path.join(iso_cfg_ext_dir, files)):
#print (os.path.join(iso_cfg_ext_dir, files))
os.chmod(os.path.join(iso_cfg_ext_dir, files),0o777)
shutil.rmtree(os.path.join(iso_cfg_ext_dir, files))
else:
#print (os.path.join(iso_cfg_ext_dir, files))
os.chmod(os.path.join(iso_cfg_ext_dir, files), 0777)
os.unlink(os.path.join(iso_cfg_ext_dir, files))
if os.path.exists(os.path.join(iso_cfg_ext_dir, files)):
os.remove(os.path.join(iso_cfg_ext_dir, files))
QtGui.QMessageBox.information(self, 'Installation Completed...',
iso_name + ' is successfully installed.')
def get_size(self, path):
total_size = 0
for dirpath, dirnames, filenames in os.walk(path):
for f in filenames:
fp = os.path.join(dirpath, f)
total_size += os.path.getsize(fp)
return total_size
def bytes2human(self, n):
# http://code.activestate.com/recipes/578019
# >>> bytes2human(10000)
# '9.8K'
# >>> bytes2human(100001221)
# '95.4M'
symbols = ('K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y')
prefix = {}
for i, s in enumerate(symbols):
prefix[s] = 1 << (i + 1) * 10
for s in reversed(symbols):
if n >= prefix[s]:
value = float(n) / prefix[s]
return '%.1f%s' % (value, s)
return "%sB" % n
def install_syslinux(self, usb_device):
# Crossplatform function to install syslinux on selected device.
usb_mount_count = len(str(self.ui.usb_mount.text()[9:]))
if not var.distro_isolinux_exist == "no":
var.distro_isolinux_bin_path = self.detect_distro_isobin(var.install_dir)
print "isolinux.bin found on " + var.distro_isolinux_bin_path
var.distro_syslinux_dir_path = os.path.dirname(var.distro_isolinux_bin_path)[usb_mount_count:]
var.distro_syslinux_version = self.distro_syslinux_version(var.distro_isolinux_bin_path)
else:
var.distro_isolinux_bin_path = None
print "isolinux.bin not found..."
if not var.distro_syslinux_version == None:
if sys.platform.startswith("linux"):
extension = str(var.distro_syslinux_version)
if var.distro_syslinux_version == "3":
var.syslinux_options = " -d "
else:
var.syslinux_options = " -i -d "
var.defautl_syslinux_version = resource_path(os.path.join("tools", "syslinux", "bin", 'syslinux4'))
else:
extension = str(var.distro_syslinux_version) + ".exe"
if var.distro_syslinux_version == "3":
var.syslinux_options = " -d "
else:
var.syslinux_options = " -i -d "
var.defautl_syslinux_version = resource_path(os.path.join("tools", "syslinux", "bin", 'syslinux4') + ".exe")
var.defautl_syslinux_version = var.defautl_syslinux_version[:-4]
extension = var.distro_syslinux_version + ".exe"
#var.syslinux_version = resource_path(os.path.join("tools", "syslinux", "bin", "syslinux")) + extension
var.syslinux_version = var.defautl_syslinux_version[:-1] + extension # Have to use this instead of resource_path as it breaks after building executable.
if not var.distro_isolinux_exist == "no":
self.install_syslinux_distro_dir(var.distro_syslinux_dir_path, usb_device, var.mbr_bin, var.usb_mount, var.syslinux_version,
var.syslinux_options) # To install syslinux on distro isolinux directory.
self.install_syslinux_distro_dir( "multibootusb", usb_device, var.mbr_bin, var.usb_mount, var.defautl_syslinux_version , " -i -d ")
else:
self.install_syslinux_distro_dir( "multibootusb", usb_device, var.mbr_bin, var.usb_mount, var.defautl_syslinux_version , " -i -d ")
def onInstall_syslinuxClick(self):
var.sys_tab = "yes"
if sys.platform.startswith("linux"):
var.defautl_syslinux_version = resource_path(os.path.join("tools", "syslinux", "bin", 'syslinux4'))
else:
var.defautl_syslinux_version = resource_path(os.path.join("tools", "syslinux", "bin", 'syslinux4.exe'))
src = str(resource_path(os.path.join("tools", "multibootusb")))
dst = str(os.path.join(str(usb_mount), "multibootusb"))
if self.ui.install_sys_all.isChecked() or self.ui.install_sys_only.isChecked():
self.install_syslinux_distro_dir( "multibootusb", usb_device, var.mbr_bin, var.usb_mount, var.defautl_syslinux_version , " -i -d ")
if self.ui.install_sys_all.isChecked():
self.copytree(src, dst)
else:
QtGui.QMessageBox.information(self, 'No selection...',
'Please select from one of the option above.')
def onedit_syslinux(self):
# Function to edit syslinux.cfg file on editors like gedit, notepad etc.
# Suggest me more editor which can be included in to this function.
global editor_exit_status
if not os.path.exists(sys_cfg_file):
print "syslinux.cfg file not found..."
QtGui.QMessageBox.information(self, 'File not found...', 'Sorry. Unable to locate syslinux.cfg file.')
else:
if sys.platform.startswith("linux"):
for editor in editors_linux:
if os.system('which ' + editor) == 0:
print "Found " + editor
editor_exit_status = subprocess.Popen(editor + " " + sys_cfg_file, shell=True).pid
if not editor_exit_status:
print "syslinux.cfg file successfully opened for append."
break
else:
for editor in editors_win:
if not self.which(editor) == None:
print editor
#editor_exit_status = subprocess.Popen(editor + " " + sys_cfg_file, shell=True).pid
var.editor = subprocess.Popen(editor + " " + sys_cfg_file, shell=True).pid
break
def which(self, program):
import os
def is_exe(fpath):
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
fpath, fname = os.path.split(program)
if fpath:
if is_exe(program):
return program
else:
for path in os.environ["PATH"].split(os.pathsep):
path = path.strip('"')
exe_file = os.path.join(path, program)
if is_exe(exe_file):
return exe_file
return None
def on_close_Click(self):
# I am having trouble here. Unable to find if process like 7zip extraction, qemu, editor opened status etc.
# Help required to identify/ running opened processes.
global quit_ready
if not var.qemu_usb == "":
if psutil.pid_exists(var.qemu_usb):
print "QEMU process exist..."
QtGui.QMessageBox.information(self, 'Process exist...',
'QEMU is running.\nPlease close QEMU before terminating multibootusb.')
else:
var.qemu_usb = ""
elif not var.qemu_iso == "":
if psutil.pid_exists(var.qemu_iso):
print "QEMU process exist..."
QtGui.QMessageBox.information(self, 'Process exist...',
'QEMU is running.\nPlease close QEMU before terminating multibootusb.')
else:
var.qemu_iso = ""
elif not var.editor == "":
if psutil.pid_exists(var.editor):
print "Syslinux.cfg is opened for edit..."
QtGui.QMessageBox.information(self, 'Process exist...',
'syslinux.cfg is open for edit.\nPlease save and close file before terminating multibootusb.')
else:
var.editor = ""
if var.qemu_usb == "" or var.qemu_iso == "" or var.editor == "":
quit_ready = "yes"
print "Closing multibootusb..."
QtGui.qApp.closeAllWindows()
def closeEvent(self, event):
quit_msg = "Do you really want to exit multibootusb?"
reply = QtGui.QMessageBox.question(self, 'Exit...',
quit_msg, QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
event.accept()
else:
event.ignore()
