aboutsummaryrefslogtreecommitdiffstats
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rwxr-xr-xlib/isodumper.py465
-rwxr-xr-xlib/raw_format.py81
2 files changed, 281 insertions, 265 deletions
diff --git a/lib/isodumper.py b/lib/isodumper.py
index ffc56cc..d8b4a8f 100755
--- a/lib/isodumper.py
+++ b/lib/isodumper.py
@@ -1,6 +1,6 @@
-#coding:utf-8
+# coding:utf-8
-#!/usr/bin/python3
+# !/usr/bin/python3
#
# Copyright (c) 2007-2009 Canonical Ltd.
#
@@ -28,12 +28,13 @@
# ensure we're using the latest build, if called from our build environment
import sys
import importlib
+
try:
from isodumper import version
except:
import version
-sys.path.insert(0,'../../../build/swig/python3')
+sys.path.insert(0, '../../../build/swig/python3')
importlib.reload(sys)
###########
# imports #
@@ -51,13 +52,11 @@ import psutil
import manatools.args as args
-
class NoUDisks2(Exception):
pass
class UDisks2(object):
-
BLOCK = 'org.freedesktop.UDisks2.Block'
DRIVE = 'org.freedesktop.UDisks2.Drive'
SERVICE = 'org.freedesktop.UDisks2'
@@ -66,37 +65,38 @@ class UDisks2(object):
self.bus = SystemBus()
try:
self.iface = self.bus.get(self.SERVICE)
- except :
+ except:
raise NoUDisks2()
self._udisks2_obj_manager = self.iface["org.freedesktop.DBus.ObjectManager"]
def find_devices(self):
objects = self._udisks2_obj_manager.GetManagedObjects()
- re_drive = re.compile('(?P<path>.*?/drives/(?P<id>.*))')
- re_block = re.compile('(?P<path>.*?/block_devices/(?P<id>.*))')
- devs= [m.groupdict() for m in
- [re_drive.match(path) for path in (objects.keys())]
- if m]
+ re_drive = re.compile("(?P<path>.*?/drives/(?P<id>.*))")
+ re_block = re.compile("(?P<path>.*?/block_devices/(?P<id>.*))")
+ devs = [m.groupdict() for m in
+ [re_drive.match(path) for path in (objects.keys())]
+ if m]
blocks = [m.groupdict() for m in
- [re_block.match(path) for path in (objects.keys())]
- if m]
- list=[]
+ [re_block.match(path) for path in (objects.keys())]
+ if m]
+ list = []
for block in blocks:
if not (self.SERVICE + '.Partition' in objects[block['path']]): # we skip partition blocks
for dev in devs:
- if dev['path'] == objects[block['path']][self.BLOCK]['Drive']: # we look for device which is referenced in Drive attribute of the current block
- dev_obj =objects[dev['path']][self.DRIVE]
+ if dev['path'] == objects[block['path']][self.BLOCK]['Drive']:
+ # we look for device which is referenced in Drive attribute of the current block
+ dev_obj = objects[dev['path']][self.DRIVE]
if (dev_obj['ConnectionBus'] == 'usb' or dev_obj['ConnectionBus'] == 'sdio') and \
- (dev_obj['Removable'] == 1 or dev_obj['MediaRemovable'] == 1 ): # we keep only blocks where device ConnectionBus is sdio or usb
- item=[]
+ (dev_obj['Removable'] == 1 or dev_obj[
+ 'MediaRemovable'] == 1): # we keep only blocks where device ConnectionBus is sdio or usb
+ item = []
vend = dev_obj['Vendor']
if vend == "":
name = dev_obj['Model']
else:
- name = vend + " "+ dev_obj['Model']
- dev_name = block['path'].split('/')[-1]
- path = '/dev/'+block['path'].split('/')[-1]
+ name = vend + " " + dev_obj['Model']
+ path = '/dev/' + block['path'].split('/')[-1]
size = dev_obj['Size']
item.append(name)
item.append(path)
@@ -108,23 +108,21 @@ class UDisks2(object):
# keep only the end, like sdb
target = os.path.basename(target)
objects = self._udisks2_obj_manager.GetManagedObjects()
- re_drive = re.compile('(?P<path>.*?/drives/(?P<id>.*))')
re_block = re.compile('(?P<path>.*?/block_devices/(?P<id>.*))')
blocks = [m.groupdict() for m in
- [re_block.match(path) for path in (objects.keys())]
- if m]
+ [re_block.match(path) for path in (objects.keys())]
+ if m]
for block in blocks:
if (self.SERVICE + '.Partition' in objects[block['path']]): # we skip non partition blocks
# Table should be something like '/org/freedesktop/UDisks2/block_devices/sdb'
- if objects[block['path']][self.SERVICE + '.Partition']['Table'].split('/')[-1] == target :
- print("Unmounting: {}".format(block['path'].split('/')[-1]))
- if (self.SERVICE + '.Encrypted' in objects[block['path']]): # Is the partition encrypted ?
+ if objects[block['path']][self.SERVICE + '.Partition']['Table'].split('/')[-1] == target:
+ if (self.SERVICE + '.Encrypted' in objects[block['path']]): # Is the partition encrypted ?
path_to_encrypted = objects[block['path']][self.SERVICE + '.Encrypted']['CleartextDevice']
print(path_to_encrypted)
- if path_to_encrypted != '/' :
+ if path_to_encrypted != '/':
iface = self.bus.get(self.SERVICE, path_to_encrypted)
fs = iface[self.SERVICE + '.Filesystem']
- if fs.MountPoints : # partition is mounted
+ if fs.MountPoints: # partition is mounted
fs.Unmount({})
iface = self.bus.get(self.SERVICE, block['path'])
fs = iface[self.SERVICE + '.Encrypted']
@@ -132,63 +130,65 @@ class UDisks2(object):
else:
iface = self.bus.get(self.SERVICE, block['path'])
fs = iface[self.SERVICE + '.Filesystem']
- if fs.MountPoints : # partition is mounted
+ if fs.MountPoints: # partition is mounted
fs.Unmount({})
return True, ""
def eject(self, device):
''' device is expected like /dev/sda'''
block = os.path.basename(device)
- iface = self.bus.get(self.SERVICE,'/org/freedesktop/UDisks2/block_devices/' + block)
- drive = iface.Get(self.BLOCK,"Drive")
+ iface = self.bus.get(self.SERVICE, '/org/freedesktop/UDisks2/block_devices/' + block)
+ drive = iface.Get(self.BLOCK, "Drive")
i_drive = self.bus.get(self.SERVICE, drive)
i_drive.Eject({})
+
class Info(object):
- def __init__(self,title,richtext,text):
- self.title=title
- self.richtext=richtext
- self.text=text
+ def __init__(self, title, richtext, text):
+ self.title = title
+ self.richtext = richtext
+ self.text = text
+
class ParseCLI(args.AppArgs):
def __init(self, command):
super().__init__(command)
+
class IsoDumper(object):
def get_devices(self):
- self.list=self.u.find_devices()
- while len(self.list)==0:
+ self.list = self.u.find_devices()
+ while len(self.list) == 0:
if self.nodevDialog():
self.list = self.u.find_devices()
else:
return False
- break
- self.devicelist.addItem("",False)
- if len(self.list)>0:
+ self.devicelist.addItem("", False)
+ if len(self.list) > 0:
for name, path, size in self.list:
- if size != 0 :
- self.devicelist.addItem(str(name+' ('+path.lstrip()+') ' + self.sizeof_fmt(size)),False)
+ if size != 0:
+ self.devicelist.addItem(str(name + ' (' + path.lstrip() + ') ' + self.sizeof_fmt(size)), False)
return True
def udev_wait(self, operation):
- wait = Popen(["udevadm","settle","--timeout=15"], stderr=PIPE)
- outs, errs = wait.communicate()
- working=True
+ wait = Popen(["udevadm", "settle", "--timeout=15"], stderr=PIPE)
+ wait.communicate()
+ working = True
while working:
time.sleep(0.5)
wait.poll()
- rc=wait.returncode
+ rc = wait.returncode
if not (rc is None):
wait = None
- working= False
+ working = False
if rc != 0:
self.return_state = False
self.return_message = _("Timeout reached when {}".format(operation))
return False
return True
- def update_list(self ):
+ def update_list(self):
self.devicelist.deleteAllItems()
self.get_devices()
self.img_name = ""
@@ -199,42 +199,43 @@ class IsoDumper(object):
def device_selected(self):
selitem = self.devicelist.selectedItem()
- if selitem != None:
+ if selitem is not None:
self.dev = selitem.label()
for name, path, size in self.list:
- if self.dev.startswith(name+' ('+path.lstrip()):
- self.deviceSize=size
- self.device_name=name.rstrip().replace(' ', '')
- self.logger(_('Target Device: ')+ self.dev)
+ if self.dev.startswith(name + ' (' + path.lstrip()):
+ self.deviceSize = size
+ self.device_name = name.rstrip().replace(' ', '')
+ self.logger(_('Target Device: ') + self.dev)
self.formatbt.setEnabled()
self.ima.setEnabled()
self.backup_select.setEnabled()
break
- def sizeof_fmt(self, num):
+ @staticmethod
+ def sizeof_fmt(num):
# I18N these are units for files size
- for unit in [_('B'),_('KiB'),_('MiB'),_('GiB'),_('TiB')]:
+ for unit in [_('B'), _('KiB'), _('MiB'), _('GiB'), _('TiB')]:
if abs(num) < 1024.0:
return "%3.3f %s" % (num, unit)
num /= 1024.0
return "%.1f %s" % (num, _('TiB'))
def backup_choosed(self):
- # Add .img if not specified
- if not self.backup_img_name.lower().endswith('.img'):
- self.backup_img_name=self.backup_img_name+".img"
- head, tail = os.path.split(self.backup_img_name)
- self.backup_dest=self.backup_img_name
- self.backup_select.setLabel(tail)
- self.dialog.recalcLayout()
- self.backupbt.setEnabled()
-
- def do_format(self,format_type,name):
+ # Add .img if not specified
+ if not self.backup_img_name.lower().endswith('.img'):
+ self.backup_img_name = self.backup_img_name + ".img"
+ head, tail = os.path.split(self.backup_img_name)
+ self.backup_dest = self.backup_img_name
+ self.backup_select.setLabel(tail)
+ self.dialog.recalcLayout()
+ self.backupbt.setEnabled()
+
+ def do_format(self, format_type, name):
target = self.dev.split('(')[1].split(')')[0]
- info = Info(_("Formatting confirmation"),True,self.warning)
+ info = Info(_("Formatting confirmation"), True, self.warning)
if self.ask_YesOrNo(info):
- rc=self.raw_format(target, format_type, name)
- self.operation=False
+ rc = self.raw_format(target, format_type, name)
+ self.operation = False
if rc == 0:
message = _('The device was formatted successfully.')
self.logger(message)
@@ -274,55 +275,61 @@ class IsoDumper(object):
self.dialog.pollEvent()
def raw_format(self, usb_path, fstype, label):
- self.operation=True
+ self.operation = True
if os.geteuid() > 0:
- launcher='pkexec'
- self.process = Popen([launcher,'/usr/bin/python3', '-u', '/usr/lib/isodumper/raw_format.py','-d',usb_path,'-f',fstype, '-l', label, '-u', str(os.geteuid()), '-g', str(os.getgid())], shell=False, stdout=PIPE, preexec_fn=os.setsid)
+ launcher = 'pkexec'
+ self.process = Popen(
+ [launcher, '/usr/bin/python3', '-u', '/usr/lib/isodumper/raw_format.py', '-d', usb_path, '-f', fstype,
+ '-l', label, '-u', str(os.geteuid()), '-g', str(os.getgid())], shell=False, stdout=PIPE,
+ preexec_fn=os.setsid)
else:
- self.process = Popen(['/usr/bin/python3', '-u', '/usr/lib/isodumper/raw_format.py','-d',usb_path,'-f',fstype, '-l', label, '-u', str(os.geteuid()), '-g', str(os.getgid())], shell=False, stdout=PIPE, preexec_fn=os.setsid)
- working=True
+ self.process = Popen(
+ ['/usr/bin/python3', '-u', '/usr/lib/isodumper/raw_format.py', '-d', usb_path, '-f', fstype, '-l',
+ label, '-u', str(os.geteuid()), '-g', str(os.getgid())], shell=False, stdout=PIPE,
+ preexec_fn=os.setsid)
+ working = True
while working:
time.sleep(0.5)
self.process.poll()
- rc=self.process.returncode
+ rc = self.process.returncode
if rc is None:
- working=True
+ working = True
else:
self.process = None
- working= False
+ working = False
return rc
def backup_go(self):
dest = self.backup_img_name
if os.path.exists(dest):
- info = Info(_("Backup confirmation"),True,_("Do you want to overwrite the file?"))
- if not(self.ask_YesOrNo(info)):
+ info = Info(_("Backup confirmation"), True, _("Do you want to overwrite the file?"))
+ if not (self.ask_YesOrNo(info)):
self.returncode = 1
return True
st = os.statvfs(os.path.dirname(dest))
free = st.f_bavail * st.f_frsize
- if free<self.deviceSize :
- sizeM=str(self.deviceSize/(1024*1024))
- message=_("The destination directory is too small to receive the backup (%s Mb needed)")%(sizeM)
+ if free < self.deviceSize:
+ sizeM = str(self.deviceSize / (1024 * 1024))
+ message = _("The destination directory is too small to receive the backup (%s Mb needed)") % (sizeM)
self.logger(message)
self.emergency(message)
else:
- self.returncode=0
+ self.returncode = 0
source = self.dev.split('(')[1].split(')')[0]
- self.logger(_('Backup to:')+' '+dest)
+ self.logger(_('Backup to:') + ' ' + dest)
bus = SystemBus()
- iface = bus.get("org.mageia.Magiback","Isodumper")
+ iface = bus.get("org.mageia.Magiback", "Isodumper")
# Writing step
iface.do_write(source, dest, self.deviceSize)
- progress = iface.progress
- while not iface.done :
+ while not iface.done:
progress = iface.progress
self.progress.setValue(progress)
self.dialog.pollEvent()
time.sleep(.2)
success, message = iface.end()
if success:
- self.logger(_('{source} successfully written to {target}').format( source=source.split('/')[-1], target=dest))
+ self.logger(
+ _('{source} successfully written to {target}').format(source=source.split('/')[-1], target=dest))
self.progress.setEnabled()
self.progress.setValue(100)
self.dialog.pollEvent()
@@ -346,18 +353,20 @@ class IsoDumper(object):
self.logger(message)
self.emergency(message)
else:
- info = Info(_("Writing confirmation"),True,self.warning)
+ info = Info(_("Writing confirmation"), True, self.warning)
if self.ask_YesOrNo(info):
- if self.deviceSize> 1024*1024*1024*32 :
- info = Info(_("Warning"),True,_('The device is bigger than 32 Gbytes. Are you sure you want use it?'))
+ if self.deviceSize > 1024 * 1024 * 1024 * 32:
+ info = Info(_("Warning"), True,
+ _('The device is bigger than 32 Gbytes. Are you sure you want use it?'))
if self.ask_YesOrNo(info):
pass
else:
return
self.ima.setDisabled()
self.writebt.setDisabled()
- self.progress.setLabel(_('Writing {source} to {target}').format(source=source.split('/')[-1],target=target.split('/')[-1]))
- self.logger(_('Executing copy from ')+source+_(' to ')+target)
+ self.progress.setLabel(_('Writing {source} to {target}').format(source=source.split('/')[-1],
+ target=target.split('/')[-1]))
+ self.logger(_('Executing copy from ') + source + _(' to ') + target)
bus = SystemBus()
iface = bus.get("org.mageia.Magiback", "Isodumper")
success, message = self.u.do_unmount(target)
@@ -367,22 +376,24 @@ class IsoDumper(object):
iface.do_write(source, target, b)
iface.get_sum(source)
progress = iface.progress
- while not iface.done :
+ while not iface.done:
progress = iface.progress
self.progress.setValue(progress)
self.dialog.pollEvent()
time.sleep(.5)
success, message = iface.end()
if success:
- self.logger(_('Image {source} successfully written to {target}').format( source=source.split('/')[-1], target=target))
- self.logger(_('Bytes written: ')+str(b))
- self.progress.setLabel(_('Checking ')+target.split('/')[-1])
+ self.logger(
+ _('Image {source} successfully written to {target}').format(source=source.split('/')[-1],
+ target=target))
+ self.logger(_('Bytes written: ') + str(b))
+ self.progress.setLabel(_('Checking ') + target.split('/')[-1])
self.progress.setValue(0)
self.dialog.pollEvent()
# Checking
iface.check_write(target, source)
progress = iface.progress
- while progress < 100 :
+ while progress < 100:
progress = iface.progress
self.progress.setValue(progress)
self.dialog.pollEvent()
@@ -398,11 +409,11 @@ class IsoDumper(object):
self.progress.setValue(0)
if self.cryptcb.isChecked():
if self.cryptkey.value() == "":
- self.logger(_("No key for encrypted partition provided. Adding the partition aborted."))
+ self.logger(
+ _("No key for encrypted partition provided. Adding the partition aborted."))
else:
- iface.do_persistence(target,"mgalive-persist", self.cryptkey.value())
- progress = iface.progress
- while not iface.done :
+ iface.do_persistence(target, "mgalive-persist", self.cryptkey.value())
+ while not iface.done:
progress = iface.progress
self.progress.setValue(progress)
self.dialog.pollEvent()
@@ -413,8 +424,7 @@ class IsoDumper(object):
self.logger(iface.message)
else:
iface.do_persistence(target, "mgalive-persist", "")
- progress = iface.progress
- while not iface.done :
+ while not iface.done:
progress = iface.progress
self.progress.setValue(progress)
self.dialog.pollEvent()
@@ -438,33 +448,32 @@ class IsoDumper(object):
self.restore()
def success(self):
- self.operation=False
+ self.operation = False
self.final_unsensitive()
- info = Info(_("Success"),True,_("The operation completed successfully.\n\
+ info = Info(_("Success"), True, _("The operation completed successfully.\n\
You are free to unplug it now, a logfile \n\
(/home/-user- or /root)/.isodumper/isodumper.log will be saved when\n\
you close the application."))
- if self.ask_OK(info) :
- return
-
+ if self.ask_OK(info):
+ return
def confirm_close(self, *args):
- if self.operation==False: # no writing , backup nor format running
+ if not self.operation: # no writing , backup nor format running
self.close()
- else: # writing , backup or format running
- info = Info(_("Warning"),True,_("Writing is in progress. Exiting during writing \n\
+ else: # writing , backup or format running
+ info = Info(_("Warning"), True, _("Writing is in progress. Exiting during writing \n\
will make the device or the backup unusable.\n\
Are you sure you want to quit during writing?"))
- if self.ask_YesOrNo(info) :
+ if self.ask_YesOrNo(info):
return True
else:
return False
- def emergency(self,message):
- self.operation=False
- self.returncode=1
+ def emergency(self, message):
+ self.operation = False
+ self.returncode = 1
self.final_unsensitive()
- info = Info(_("Error"),True,message)
+ info = Info(_("Error"), True, message)
self.ask_OK(info)
def final_unsensitive(self):
@@ -490,56 +499,61 @@ class IsoDumper(object):
self.dialog = None
def write_logfile(self):
- logpath = os.path.join(os.path.expanduser('~'),'.isodumper')
- if not(os.path.isdir(logpath)):
- os.mkdir(logpath)
- logfile=open(logpath+'/isodumper.log',"w", encoding="utf-8")
+ logpath = os.path.join(os.path.expanduser('~'), '.isodumper')
+ if not (os.path.isdir(logpath)):
+ os.mkdir(logpath)
+ logfile = open(logpath + '/isodumper.log', "w", encoding="utf-8")
logfile.write(self.logview.logText())
logfile.close()
print((self.logview.logText()))
def logger(self, text):
- self.logview.appendLines(text+"\n")
+ self.logview.appendLines(text + "\n")
def activate_devicelist(self):
self.devicelist.setEnabled()
- self.logger(_('Image ')+": "+ self.img_name)
-# self.chooser.set_tooltip_text(self.img_name)
+ self.logger(_('Image ') + ": " + self.img_name)
+
+ # self.chooser.set_tooltip_text(self.img_name)
def help_dialog(self):
- info = Info(_("IsoDumper"),True,_("Mageia IsoDumper")+"<BR />\
-----------------<BR />"+
-_("This GUI program is primarily for safely writing a bootable ISO image to a USB flash drive, \
+ info = Info(_("IsoDumper"), True, _("Mageia IsoDumper") + "<BR />\
+----------------<BR />" +
+ _("This GUI program is primarily for safely writing a bootable ISO image to a USB flash drive, \
an operation devious & potentially hazardous when done by hand. As a bonus, it can also back up the \
entire previous contents of the flash drive onto the hard disc, and restore the flash drive \
-to its previous state subsequently.")+"<BR />"+
-_("It gives also a feature for formatting the USB device.")+"<BR /><BR />"+
-_("IsoDumper can be launched either from the menus, or a user console with the command 'isodumper'.")+"<BR />"+
-_("The root password is solicited when this is necessary for the program's operation.")+" <BR />"+
-_("The flash drive can be inserted beforehand or once the program is started. In the latter case, a \
+to its previous state subsequently.") + "<BR />" +
+ _("It gives also a feature for formatting the USB device.") + "<BR /><BR />" +
+ _(
+ "IsoDumper can be launched either from the menus, or a user console with the command 'isodumper'.") + "<BR />" +
+ _(
+ "The root password is solicited when this is necessary for the program's operation.") + " <BR />" +
+ _("The flash drive can be inserted beforehand or once the program is started. In the latter case, a \
dialogue will say that there is no flash drive inserted, and allow a 'retry' to find it once it is. <BR />\
-(You may have to close any automatically opened File Manager window).")+"<BR /><BR />"+
-_("The fields of the main window are as follows:<BR />\
+(You may have to close any automatically opened File Manager window).") + "<BR /><BR />" +
+ _("The fields of the main window are as follows:<BR />\
- Device to work on: the device of the USB flash drive, a drop-down list to choose from.<BR />\
- Write Image: to choose the source ISO image *.iso (or flash drive backup file *.img) to write out.<BR />\
-- Write to device: This button launches the operation - with a prior warning dialogue.") +"<BR />"+
-_("- Add a persistent partition: the remaining space will be used in a new partition where data from the Live system can be written and recovered between sessions.")+ "<BR />"+
-_("- Encrypt: the persistent partition will be encrypted with the key provided in <i>Key</i> field.")+ "<BR />"+
-_("The operation is shown in the progress bar beneath.")+"<BR />"+
-_("- Backup to: define the name and placement of the backup image file. The current flash drive \
+- Write to device: This button launches the operation - with a prior warning dialogue.") + "<BR />" +
+ _(
+ "- Add a persistent partition: the remaining space will be used in a new partition where data from the Live system can be written and recovered between sessions.") + "<BR />" +
+ _(
+ "- Encrypt: the persistent partition will be encrypted with the key provided in <i>Key</i> field.") + "<BR />" +
+ _("The operation is shown in the progress bar beneath.") + "<BR />" +
+ _("- Backup to: define the name and placement of the backup image file. The current flash drive \
will be backed up to a disc file. Note that the entire flash drive is preserved, regardless of its \
actual contents; ensure that you have the necessary free disc space (the same size as the USB device). \
-This backup file can be used later to restore the flash drive by selecting it as the source *.img file to write out.")+"<BR />"+
-_("- Backup the device: launch the backup operation.")+"<BR />"+
-_("- Format the device: create an unique partition on the entire volume in the specified format in FAT, \
-exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog box.")+"<BR />")
- if self.ask_OK(info) :
- return
+This backup file can be used later to restore the flash drive by selecting it as the source *.img file to write out.") + "<BR />" +
+ _("- Backup the device: launch the backup operation.") + "<BR />" +
+ _("- Format the device: create an unique partition on the entire volume in the specified format in FAT, \
+exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog box.") + "<BR />")
+ if self.ask_OK(info):
+ return
def __init__(self):
- APP="isodumper"
- DIR="/usr/share/locale"
+ APP = "isodumper"
+ DIR = "/usr/share/locale"
# Call translation catalog
gettext.install(APP, localedir=DIR, names=('ngettext',))
@@ -547,20 +561,19 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
current_pid = psutil.Process().pid
for pid in psutil.pids():
p = psutil.Process(pid)
- if p.name() == "isodumper" and p.pid != current_pid :
- info = Info(_("Error"),True,_("There is another instance of Isodumper already running."))
+ if p.name() == "isodumper" and p.pid != current_pid:
+ info = Info(_("Error"), True, _("There is another instance of Isodumper already running."))
self.ask_OK(info)
yui.YUILoader.deleteUI()
exit()
-
# TODO Read log level from command line
# define size of the selected device
- self.deviceSize=0
+ self.deviceSize = 0
# Operation running
- self.operation=False
+ self.operation = False
self.ChooseImage = _("Choose an image")
self.warning = _("Warning\nThis will destroy all data on the target device,\n\
@@ -595,69 +608,70 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
self.ancrage = self.atelier.createReplacePoint(self.dialog)
self.box = self.atelier.createVBox(self.ancrage)
self.bannerbox = self.atelier.createHBox(self.box)
- self.banner=self.atelier.createImage(self.bannerbox,"/usr/share/isodumper/header.svg")
+ self.banner = self.atelier.createImage(self.bannerbox, "/usr/share/isodumper/header.svg")
self.devicebox = self.atelier.createHBox(self.box)
- self.devicelist=self.atelier.createComboBox(self.devicebox,_("Device to work on:"),False)
+ self.devicelist = self.atelier.createComboBox(self.devicebox, _("Device to work on:"), False)
self.devicelist.setNotify()
- self.devicelist.setStretchable(0,True)
+ self.devicelist.setStretchable(0, True)
self.writebox = self.atelier.createHBox(self.box)
- self.atelier.createLabel(self.writebox,_("Write Image:"))
+ self.atelier.createLabel(self.writebox, _("Write Image:"))
# add file picker for image file
- self.ima=self.atelier.createPushButton(self.writebox,self.ChooseImage)
- self.ima.setStretchable(0,True)
+ self.ima = self.atelier.createPushButton(self.writebox, self.ChooseImage)
+ self.ima.setStretchable(0, True)
self.ima.setDisabled()
self.atelier.createHStretch(self.writebox)
- self.writebt = self.atelier.createPushButton(self.writebox, _("&Write to device" ))
+ self.writebt = self.atelier.createPushButton(self.writebox, _("&Write to device"))
self.writebt.setDisabled()
self.persistencebox = self.atelier.createHBox(self.box)
self.persistencecol1 = self.atelier.createVBox(self.persistencebox)
self.persistencecol2 = self.atelier.createVBox(self.persistencebox)
- self.persistencecb = self.atelier.createCheckBox(self.persistencecol1, _("Add a persistent partition in the remaining space"))
+ self.persistencecb = self.atelier.createCheckBox(self.persistencecol1,
+ _("Add a persistent partition in the remaining space"))
self.persistencebox2 = self.atelier.createHBox(self.box)
self.cryptcb = self.atelier.createCheckBox(self.persistencecol1, _("Encrypt partition"))
- self.cryptkey = self.atelier.createInputField(self.persistencecol2, _("Key:"),passwordMode = True)
+ self.cryptkey = self.atelier.createInputField(self.persistencecol2, _("Key:"), passwordMode=True)
self.persistencecb.setDisabled()
self.persistencecb.setNotify()
self.cryptcb.setDisabled()
self.cryptcb.setNotify()
self.cryptkey.setDisabled()
self.backupbox = self.atelier.createHBox(self.box)
- self.atelier.createLabel(self.backupbox,_("Backup to:"))
+ self.atelier.createLabel(self.backupbox, _("Backup to:"))
# add file picker for backup file name
- self.backup_select=self.atelier.createPushButton(self.backupbox,self.ChooseImage)
- self.backup_select.setStretchable(0,True)
+ self.backup_select = self.atelier.createPushButton(self.backupbox, self.ChooseImage)
+ self.backup_select.setStretchable(0, True)
self.backup_select.setDisabled()
self.atelier.createHStretch(self.backupbox)
- self.backupbt = self.atelier.createPushButton(self.backupbox, _("Backup the device" ))
+ self.backupbt = self.atelier.createPushButton(self.backupbox, _("Backup the device"))
self.backupbt.setDisabled()
self.formatbox = self.atelier.createHBox(self.box)
- self.atelier.createLabel(self.formatbox,_("Format the device in FAT, exFAT, NTFS or ext:"))
+ self.atelier.createLabel(self.formatbox, _("Format the device in FAT, exFAT, NTFS or ext:"))
self.atelier.createHStretch(self.formatbox)
- self.formatbt = self.atelier.createPushButton(self.formatbox, _("Format the device" ))
+ self.formatbt = self.atelier.createPushButton(self.formatbox, _("Format the device"))
self.formatbt.setDisabled()
self.progressbox = self.atelier.createHBox(self.box)
- self.progress = self.atelier.createProgressBar(self.progressbox,_("Progress"),100)
- self.progress.setStretchable(0,True)
+ self.progress = self.atelier.createProgressBar(self.progressbox, _("Progress"), 100)
+ self.progress.setStretchable(0, True)
self.reportbox = self.atelier.createHBox(self.box)
- self.reportbox.setWeight(1,30)
- self.logview = self.atelier.createLogView(self.reportbox,_("Report"), 10)
- self.logview.setStretchable(0,True)
+ self.reportbox.setWeight(1, 30)
+ self.logview = self.atelier.createLogView(self.reportbox, _("Report"), 10)
+ self.logview.setStretchable(0, True)
self.buttonsbox = self.atelier.createHBox(self.box)
- self.refreshbt = self.atelier.createPushButton(self.devicebox, _("Refresh" ))
- self.refreshbt.setStretchable(0,True)
- self.aboutbt = self.atelier.createPushButton(self.buttonsbox, _("About" ))
- self.aboutbt.setStretchable(0,True)
- self.helpbt = self.atelier.createPushButton(self.buttonsbox, _("Help" ))
- self.helpbt.setStretchable(0,True)
- self.quitbt = self.atelier.createPushButton(self.buttonsbox, _("Quit" ))
- self.quitbt.setStretchable(0,True)
+ self.refreshbt = self.atelier.createPushButton(self.devicebox, _("Refresh"))
+ self.refreshbt.setStretchable(0, True)
+ self.aboutbt = self.atelier.createPushButton(self.buttonsbox, _("About"))
+ self.aboutbt.setStretchable(0, True)
+ self.helpbt = self.atelier.createPushButton(self.buttonsbox, _("Help"))
+ self.helpbt.setStretchable(0, True)
+ self.quitbt = self.atelier.createPushButton(self.buttonsbox, _("Quit"))
+ self.quitbt.setStretchable(0, True)
self.u = None
try:
self.u = UDisks2()
- except :
- message = _('UDisks2 is not available on your system')
- self.logger(message)
- self.emergency(message)
+ except:
+ message = _('UDisks2 is not available on your system')
+ self.logger(message)
+ self.emergency(message)
if not self.get_devices():
self.dialog.destroy()
yui.YUILoader.deleteUI()
@@ -667,23 +681,23 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
self.ancrage.showChild()
def ask_format(self):
- atelier = yui.YUI.widgetFactory()
+ atelier = yui.YUI.widgetFactory()
dialog = atelier.createPopupDialog()
- vb=atelier.createVBox(dialog)
- label = atelier.createInputField(vb,_("Label for the device:"))
+ vb = atelier.createVBox(dialog)
+ label = atelier.createInputField(vb, _("Label for the device:"))
cr = atelier.createRadioButtonGroup(vb)
vb_c = atelier.createVBox(cr)
vb_c1 = atelier.createHBox(vb_c)
- format_fat = atelier.createRadioButton(atelier.createLeft(vb_c1),_("FAT 32 (Windows)"))
+ format_fat = atelier.createRadioButton(atelier.createLeft(vb_c1), _("FAT 32 (Windows)"))
vb_c4 = atelier.createHBox(vb_c)
- format_exfat = atelier.createRadioButton(atelier.createLeft(vb_c4),_("exFAT (Windows)"))
+ format_exfat = atelier.createRadioButton(atelier.createLeft(vb_c4), _("exFAT (Windows)"))
vb_c2 = atelier.createHBox(vb_c)
- format_ntfs = atelier.createRadioButton(atelier.createLeft(vb_c2),_("NTFS (Windows)"))
+ format_ntfs = atelier.createRadioButton(atelier.createLeft(vb_c2), _("NTFS (Windows)"))
vb_c3 = atelier.createHBox(vb_c)
- format_ext = atelier.createRadioButton(atelier.createLeft(vb_c3),_("ext4 (Linux)"))
+ format_ext = atelier.createRadioButton(atelier.createLeft(vb_c3), _("ext4 (Linux)"))
bb = atelier.createHBox(vb)
- executebt = atelier.createPushButton(bb,_("Execute"))
- cancelbt = atelier.createPushButton(bb,_("Cancel"))
+ executebt = atelier.createPushButton(bb, _("Execute"))
+ cancelbt = atelier.createPushButton(bb, _("Cancel"))
dialog.open()
returncode = True
while True:
@@ -713,10 +727,9 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
break
dialog.destroy()
if returncode:
- return True,format_type,format_label
+ return True, format_type, format_label
else:
- return False," "," "
-
+ return False, " ", " "
def ask_OK(self, info):
yui.YUI.widgetFactory
@@ -725,7 +738,7 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
dlg.setTitle(info.title)
dlg.setText(info.text, info.richtext)
dlg.setButtonLabel(_("OK"), yui.YMGAMessageBox.B_ONE)
- dlg.setMinSize(60, 10);
+ dlg.setMinSize(60, 10)
return dlg.show() == yui.YMGAMessageBox.B_ONE
def ask_YesOrNo(self, info):
@@ -735,30 +748,32 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
dlg.setTitle(info.title)
dlg.setText(info.text, info.richtext)
dlg.setButtonLabel(_("Yes"), yui.YMGAMessageBox.B_ONE)
- dlg.setButtonLabel(_("No"), yui.YMGAMessageBox.B_TWO)
- dlg.setMinSize(60, 8);
+ dlg.setButtonLabel(_("No"), yui.YMGAMessageBox.B_TWO)
+ dlg.setMinSize(60, 8)
return dlg.show() == yui.YMGAMessageBox.B_ONE
def aboutDialog(self):
- yui.YUI.widgetFactory;
+ yui.YUI.widgetFactory
mgafactory = yui.YMGAWidgetFactory.getYMGAWidgetFactory(yui.YExternalWidgets.externalWidgetFactory("mga"))
dlg = mgafactory.createAboutDialog("Isodumper", version.RELEASE, "GPLv2",
- _("Oliver Grawert<BR />Papoteur<BR />Pictures : Timothée Giet"), _("A tool for writing ISO images to a device")+"<BR />http://gitweb.mageia.org/software/isodumper", "")
- dlg.show();
+ _("Oliver Grawert<BR />Papoteur<BR />Pictures : Timothée Giet"), _(
+ "A tool for writing ISO images to a device") + "<BR />http://gitweb.mageia.org/software/isodumper", "")
+ dlg.show()
def nodevDialog(self):
yui.YUI.widgetFactory
mgafactory = yui.YMGAWidgetFactory.getYMGAWidgetFactory(yui.YExternalWidgets.externalWidgetFactory("mga"))
dlg = mgafactory.createDialogBox(yui.YMGAMessageBox.B_TWO)
dlg.setTitle("IsoDumper")
- dlg.setText(_("Warning\nNo target devices were found.\nYou need to plug in a USB Key to which the image can be written."))
+ dlg.setText(_(
+ "Warning\nNo target devices were found.\nYou need to plug in a USB Key to which the image can be written."))
dlg.setButtonLabel(_("Refresh"), yui.YMGAMessageBox.B_ONE)
- dlg.setButtonLabel(_("Cancel"), yui.YMGAMessageBox.B_TWO)
- dlg.setMinSize(60, 8);
+ dlg.setButtonLabel(_("Cancel"), yui.YMGAMessageBox.B_TWO)
+ dlg.setMinSize(60, 8)
return dlg.show() == yui.YMGAMessageBox.B_ONE
def handleevent(self):
- self.traitement=None
+ self.traitement = None
while True:
event = self.dialog.waitForEvent()
if event.eventType() == yui.YEvent.CancelEvent:
@@ -768,43 +783,43 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
self.close()
break
if event.widget() == self.ima:
- self.img_name=yui.YUI.app().askForExistingFile("","*.iso *.img",self.ChooseImage)
- if self.img_name != "" :
- self.ima.setLabel(os.path.basename(self.img_name))
- self.dialog.recalcLayout()
- self.writebt.setEnabled()
- self.activate_devicelist()
- self.persistencecb.setEnabled()
+ self.img_name = yui.YUI.app().askForExistingFile("", "*.iso *.img", self.ChooseImage)
+ if self.img_name != "":
+ self.ima.setLabel(os.path.basename(self.img_name))
+ self.dialog.recalcLayout()
+ self.writebt.setEnabled()
+ self.activate_devicelist()
+ self.persistencecb.setEnabled()
if event.widget() == self.persistencecb:
print(self.persistencecb.value)
- if self.persistencecb.isChecked() :
+ if self.persistencecb.isChecked():
self.cryptcb.setEnabled()
else:
self.cryptcb.setDisabled()
self.cryptkey.setDisabled()
self.cryptcb.setChecked(False)
if event.widget() == self.cryptcb:
- if self.cryptcb.isChecked() :
+ if self.cryptcb.isChecked():
self.cryptkey.setEnabled()
else:
self.cryptkey.setDisabled()
if event.widget() == self.writebt:
- self.wip_unsensitive()
- self.do_write()
- self.restore()
+ self.wip_unsensitive()
+ self.do_write()
+ self.restore()
if event.widget() == self.backupbt:
self.wip_unsensitive()
self.backup_go()
self.restore()
if event.widget() == self.backup_select:
- self.backup_img_name=yui.YUI.app().askForSaveFileName("","*.img",_("Backup to:"))
+ self.backup_img_name = yui.YUI.app().askForSaveFileName("", "*.img", _("Backup to:"))
if self.backup_img_name != '':
- self.backup_choosed()
+ self.backup_choosed()
if event.widget() == self.formatbt:
- code,format_type,name = self.ask_format()
- if code:
- self.do_format(format_type,name)
- self.restore()
+ code, format_type, name = self.ask_format()
+ if code:
+ self.do_format(format_type, name)
+ self.restore()
if event.widget() == self.devicelist:
self.device_selected()
try:
@@ -828,7 +843,7 @@ exFAT, NTFS or ext. You can specify a volume name and the format in a new dialog
def run(self):
try:
self.handleevent()
- except Exception as e:
+ except Exception:
import traceback
traceback.print_exc()
yui.YDialog.deleteAllDialogs()
diff --git a/lib/raw_format.py b/lib/raw_format.py
index 54e7bcb..b3bdb42 100755
--- a/lib/raw_format.py
+++ b/lib/raw_format.py
@@ -22,42 +22,44 @@
#
# Author isodumper software= papoteur <papoteur@mageialinux-online.org>
-import os, sys
+import sys
import getopt
from subprocess import call
sys.path.append('/usr/lib/isodumper')
+
def do_umount(target):
- mounts = get_mounted(target)
- if mounts:
- print('Unmounting all partitions of '+target+':')
- for mount in mounts:
- print('Trying to unmount '+mount[0]+'...')
- try:
- retcode = call('umount '+mount[0], shell=True)
- if retcode < 0:
- print('Error, umount '+mount[0]+' was terminated by signal '+str(retcode))
- sys.exit(6)
- else:
- if retcode == 0:
- print(mount[0]+' successfully unmounted')
- else:
- print('Error, umount '+mount[0]+' returned '+str(retcode))
- sys.exit(6)
- except OSError as e:
- print('Execution failed: '+str(e))
+ mounts = get_mounted(target)
+ if mounts:
+ print('Unmounting all partitions of ' + target + ':')
+ for mount in mounts:
+ print('Trying to unmount ' + mount[0] + '...')
+ try:
+ retcode = call('umount ' + mount[0], shell=True)
+ if retcode < 0:
+ print('Error, umount ' + mount[0] + ' was terminated by signal ' + str(retcode))
sys.exit(6)
+ else:
+ if retcode == 0:
+ print(mount[0] + ' successfully unmounted')
+ else:
+ print('Error, umount ' + mount[0] + ' returned ' + str(retcode))
+ sys.exit(6)
+ except OSError as e:
+ print('Execution failed: ' + str(e))
+ sys.exit(6)
def get_mounted(target):
- try:
- lines = [line.strip("\n").split(" ") for line in open ("/etc/mtab", "r").readlines()]
- return [mount for mount in lines if mount[0].startswith(target)]
- except:
- print('Could not read mtab !')
- sys.exit(6)
+ try:
+ lines = [line.strip("\n").split(" ") for line in open("/etc/mtab", "r").readlines()]
+ return [mount for mount in lines if mount[0].startswith(target)]
+ except:
+ print('Could not read mtab !')
+ sys.exit(6)
+
def raw_format(device_path, fstype, volume_label, uid, gid):
import parted
@@ -79,17 +81,17 @@ def raw_format(device_path, fstype, volume_label, uid, gid):
regions = disk.getFreeSpaceRegions()
if len(regions) > 0:
- #print "Build partition"
+ # print "Build partition"
# Define size
start = parted.sizeToSectors(1, "MiB", device.sectorSize)
- #print "start %s" % start
+ # print "start %s" % start
end = device.getLength() - start - 1024
- #print end
+ # print end
# Alignment
- #cylinder = device.endSectorToCylinder(end)
- #end = device.endCylinderToSector(cylinder)
- #print end
+ # cylinder = device.endSectorToCylinder(end)
+ # end = device.endCylinderToSector(cylinder)
+ # print end
try:
geometry = parted.Geometry(device=device, start=start, end=end)
except:
@@ -120,13 +122,11 @@ def raw_format(device_path, fstype, volume_label, uid, gid):
sys.exit(0)
-
-
-
def main():
# parse command line options
try:
- opts, args = getopt.getopt(sys.argv[1:], "hd:f:l:u:g:", ["help", "device=","filesystem=","label=","uid=","gid="])
+ opts, args = getopt.getopt(sys.argv[1:], "hd:f:l:u:g:",
+ ["help", "device=", "filesystem=", "label=", "uid=", "gid="])
except getopt.error as msg:
print(msg)
print("for help use --help")
@@ -134,7 +134,7 @@ def main():
for o, a in opts:
if o in ("-h", "--help"):
- print("Usage: %s -d device -f filesystem -l volume_label\n" % sys.argv[0])
+ print("Usage: %s -d device -f filesystem -l volume_label\n"% sys.argv[0])
print("-d|--device : device path")
print("-f|--filesystem : filesystem\n")
print("-l|--label : volume label\n")
@@ -145,7 +145,7 @@ def main():
elif o in ("-d"):
device = a
elif o in ("-f"):
- if a not in [ "fat32","exfat", "ntfs", "ext4" ]:
+ if a not in ["fat32", "exfat", "ntfs", "ext4"]:
print("Specify fat32, exfat, ntfs or ext4")
sys.exit(3)
fstype = a
@@ -158,11 +158,12 @@ def main():
argc = len(sys.argv)
if argc < 11:
- print("Too few arguments")
- print("for help use --help")
- exit(2)
+ print("Too few arguments")
+ print("for help use --help")
+ exit(2)
raw_format(device, fstype, label, uid, gid)
+
if __name__ == "__main__":
main()