Git Repositories

PEP8
authorCyrille Pontvieux <jrd@enialis.net>
Wed, 4 Jun 2014 17:25:34 +0000 (19:25 +0200)
committerCyrille Pontvieux <jrd@enialis.net>
Wed, 4 Jun 2014 17:25:34 +0000 (19:25 +0200)
src/lib/bootsetup_curses.py
src/lib/bootsetup_gtk.py
src/lib/config.py
src/lib/grub2.py
src/lib/lilo.py

index 7e00817..51c36a9 100644 (file)
@@ -16,8 +16,8 @@ import urwidm
 from gathercurses import *
 from bootsetup import *
 
+
 class BootSetupCurses(BootSetup):
-  
   gc = None
   _palette = [
       ('important', 'yellow', 'black', 'bold'),
@@ -33,7 +33,7 @@ class BootSetupCurses(BootSetup):
     self.gc = GatherCurses(self, self._version, self._bootloader, self._targetPartition, self._isTest, self._useTestData)
     self.gc.run()
 
-  def _show_ui_dialog(self, dialog, parent = None):
+  def _show_ui_dialog(self, dialog, parent=None):
     if not parent:
       parent = urwidm.Filler(urwidm.Divider(), 'top')
     uiToStop = False
@@ -49,13 +49,13 @@ class BootSetupCurses(BootSetup):
     if uiToStop:
       ui.stop()
 
-  def info_dialog(self, message, title = None, parent = None):
+  def info_dialog(self, message, title=None, parent=None):
     if not title:
       title = _("INFO")
     dialog = urwidm.TextDialog(('info', unicode(message)), 10, 60, ('important', unicode(title)))
     self._show_ui_dialog(dialog, parent)
 
-  def error_dialog(self, message, title = None, parent = None):
+  def error_dialog(self, message, title=None, parent=None):
     if not title:
       title = "/!\ " + _("ERROR")
     dialog = urwidm.TextDialog(('error', unicode(message)), 10, 60, ('important', unicode(title)))
index b01b51b..7a8d25e 100644 (file)
@@ -17,6 +17,7 @@ import gtk.glade
 from bootsetup import *
 from gathergui import *
 
+
 class BootSetupGtk(BootSetup):
   def run_setup(self):
     gtk.glade.bindtextdomain(self._appName, self._localeDir)
@@ -26,9 +27,9 @@ class BootSetupGtk(BootSetup):
       sys.exit(1)
     gg = GatherGui(self, self._version, self._bootloader, self._targetPartition, self._isTest, self._useTestData)
     gg.run()
-  
-  def info_dialog(self, message, title = None, parent = None):
-    dialog = gtk.MessageDialog(parent = parent, type = gtk.MESSAGE_INFO, buttons = gtk.BUTTONS_OK, flags = gtk.DIALOG_MODAL)
+
+  def info_dialog(self, message, title=None, parent=None):
+    dialog = gtk.MessageDialog(parent=parent, type=gtk.MESSAGE_INFO, buttons=gtk.BUTTONS_OK, flags=gtk.DIALOG_MODAL)
     if title:
       msg = "<b>{0}</b>\n\n{1}".format(unicode(title), unicode(message))
     else:
@@ -38,8 +39,8 @@ class BootSetupGtk(BootSetup):
     dialog.destroy()
     return result_info
 
-  def error_dialog(self, message, title = None, parent = None):
-    dialog = gtk.MessageDialog(parent = parent, type = gtk.MESSAGE_ERROR, buttons = gtk.BUTTONS_CLOSE, flags = gtk.DIALOG_MODAL)
+  def error_dialog(self, message, title=None, parent=None):
+    dialog = gtk.MessageDialog(parent=parent, type=gtk.MESSAGE_ERROR, buttons=gtk.BUTTONS_CLOSE, flags=gtk.DIALOG_MODAL)
     if title:
       msg = "<b>{0}</b>\n\n{1}".format(unicode(title), unicode(message))
     else:
index c15f0d3..5f2cc09 100644 (file)
@@ -15,6 +15,7 @@ import codecs
 import os
 import libsalt as slt
 
+
 class Config:
   """
   Configuration for BootSetup
@@ -28,7 +29,7 @@ class Config:
   is_test = False
   use_test_data = False
   is_live = False
-  
+
   def __init__(self, bootloader, target_partition, is_test, use_test_data):
     self.cur_bootloader = bootloader
     self.cur_boot_partition = target_partition and re.sub(r'/dev/', '', target_partition) or ''
@@ -54,20 +55,20 @@ class Config:
       self.is_live = slt.isSaLTLiveEnv()
     if self.use_test_data:
       self.disks = [
-            ['sda', 'msdos', 'WDC100 (100GB)'],
-            ['sdb', 'gpt', 'SGT350 (350GB)']
-          ]
+        ['sda', 'msdos', 'WDC100 (100GB)'],
+        ['sdb', 'gpt', 'SGT350 (350GB)']
+      ]
       self.partitions = [
-            ['sda1', 'ntfs', 'WinVista (20GB)'],
-            ['sda5', 'ext2', 'Salix (80GB)'],
-            ['sdb1', 'fat32', 'Data (300GB)'],
-            ['sdb2', 'ext4', 'Debian (50GB)']
-          ]
+        ['sda1', 'ntfs', 'WinVista (20GB)'],
+        ['sda5', 'ext2', 'Salix (80GB)'],
+        ['sdb1', 'fat32', 'Data (300GB)'],
+        ['sdb2', 'ext4', 'Debian (50GB)']
+      ]
       self.boot_partitions = [
-            ['sda5', 'ext2', 'linux', 'Salix', 'Salix 14.0'],
-            ['sda1', 'ntfs', 'chain', 'Windows', 'Vista'],
-            ['sdb2', 'ext4', 'linux', 'Debian', 'Debian 7']
-          ]
+        ['sda5', 'ext2', 'linux', 'Salix', 'Salix 14.0'],
+        ['sda1', 'ntfs', 'chain', 'Windows', 'Vista'],
+        ['sdb2', 'ext4', 'linux', 'Debian', 'Debian 7']
+      ]
       if not self.cur_boot_partition:
         self.cut_boot_partition = 'sda5'
     else:
@@ -92,7 +93,7 @@ class Config:
             break
         if osProbesPath:
           try:
-            os.remove("/var/lib/os-prober/labels") # ensure there is no previous labels
+            os.remove("/var/lib/os-prober/labels")  # ensure there is no previous labels
           except:
             pass
           self.__debug("Root device {0} ({1})".format(slashDevice, slashFS))
@@ -107,10 +108,10 @@ class Config:
           osProberPath = p
           break
       if osProberPath:
-        probes.extend(slt.execGetOutput(osProberPath, shell = False))
+        probes.extend(slt.execGetOutput(osProberPath, shell=False))
       self.__debug("Probes: " + unicode(probes))
       for probe in probes:
-        probe = unicode(probe).strip() # ensure clean line
+        probe = unicode(probe).strip()  # ensure clean line
         if probe[0] != '/':
           continue
         probe_info = probe.split(':')
@@ -118,7 +119,7 @@ class Config:
         probe_os = probe_info[1]
         probe_label = probe_info[2]
         probe_boottype = probe_info[3]
-        if probe_boottype == 'efi': # skip efi entry
+        if probe_boottype == 'efi':  # skip efi entry
           continue
         try:
           probe_fstype = [p[1] for p in self.partitions if p[0] == probe_dev][0]
index c7f1f76..e2b3497 100644 (file)
@@ -15,18 +15,18 @@ import sys
 import codecs
 import libsalt as slt
 
+
 class Grub2:
-  
   isTest = False
   _prefix = None
   _tmp = None
   _bootInBootMounted = False
   _procInBootMounted = False
-  
+
   def __init__(self, isTest):
     self.isTest = isTest
     self._prefix = "bootsetup.grub2-"
-    self._tmp = tempfile.mkdtemp(prefix = self._prefix)
+    self._tmp = tempfile.mkdtemp(prefix=self._prefix)
     slt.mounting._tempMountDir = os.path.join(self._tmp, 'mounts')
     self.__debug("tmp dir = " + self._tmp)
 
@@ -65,9 +65,9 @@ class Grub2:
     if mountPoint != '/' and os.path.exists(os.path.join(mountPoint, 'etc/fstab')):
       self.__debug("mp != / and etc/fstab exists, will try to mount /boot by chrooting")
       try:
-        self.__debug("grep -q /boot {mp}/etc/fstab && chroot {mp} /sbin/mount /boot".format(mp = mountPoint))
-        if slt.execCall("grep -q /boot {mp}/etc/fstab && chroot {mp} /sbin/mount /boot".format(mp = mount_point)):
-          self.__debug("/boot mounted in " + mp)
+        self.__debug("grep -q /boot {mp}/etc/fstab && chroot {mp} /sbin/mount /boot".format(mp=mountPoint))
+        if slt.execCall("grep -q /boot {mp}/etc/fstab && chroot {mp} /sbin/mount /boot".format(mp=mountPoint)):
+          self.__debug("/boot mounted in " + mountPoint)
           self._bootInBootMounted = True
       except:
         pass
@@ -79,9 +79,9 @@ class Grub2:
     if mountPoint != "/":
       self.__debug("mount point ≠ / so mount /dev, /proc and /sys in " + mountPoint)
       self._procInBootMounted = True
-      slt.execCall('mount -o bind /dev {mp}/dev'.format(mp = mountPoint))
-      slt.execCall('mount -o bind /proc {mp}/proc'.format(mp = mountPoint))
-      slt.execCall('mount -o bind /sys {mp}/sys'.format(mp = mountPoint))
+      slt.execCall('mount -o bind /dev {mp}/dev'.format(mp=mountPoint))
+      slt.execCall('mount -o bind /proc {mp}/proc'.format(mp=mountPoint))
+      slt.execCall('mount -o bind /sys {mp}/sys'.format(mp=mountPoint))
 
   def _unbindProcSysDev(self, mountPoint):
     """
@@ -89,32 +89,32 @@ class Grub2:
     """
     if self._procInBootMounted:
       self.__debug("mount point ≠ / so umount /dev, /proc and /sys in " + mountPoint)
-      slt.execCall('umount {mp}/dev'.format(mp = mountPoint))
-      slt.execCall('umount {mp}/proc'.format(mp = mountPoint))
-      slt.execCall('umount {mp}/sys'.format(mp = mountPoint))
+      slt.execCall('umount {mp}/dev'.format(mp=mountPoint))
+      slt.execCall('umount {mp}/proc'.format(mp=mountPoint))
+      slt.execCall('umount {mp}/sys'.format(mp=mountPoint))
 
   def _copyAndInstallGrub2(self, mountPoint, device):
     if self.isTest:
-      self.__debug("/usr/sbin/grub-install --boot-directory {bootdir} --no-floppy {dev}".format(bootdir = os.path.join(mountPoint, "boot"), dev = device))
+      self.__debug("/usr/sbin/grub-install --boot-directory {bootdir} --no-floppy {dev}".format(bootdir=os.path.join(mountPoint, "boot"), dev=device))
       return True
     else:
-      return slt.execCall("/usr/sbin/grub-install --boot-directory {bootdir} --no-floppy {dev}".format(bootdir = os.path.join(mountPoint, "boot"), dev = device))
+      return slt.execCall("/usr/sbin/grub-install --boot-directory {bootdir} --no-floppy {dev}".format(bootdir=os.path.join(mountPoint, "boot"), dev=device))
 
   def _installGrub2Config(self, mountPoint):
     if os.path.exists(os.path.join(mountPoint, 'etc/default/grub')) and os.path.exists(os.path.join(mountPoint, 'usr/sbin/update-grub')):
       self.__debug("grub2 package is installed on the target partition, so it will be used to generate the grub.cfg file")
       # assume everything is installed on the target partition, grub2 package included.
       if self.isTest:
-        self.__debug("chroot {mp} /usr/sbin/update-grub".format(mp = mountPoint))
+        self.__debug("chroot {mp} /usr/sbin/update-grub".format(mp=mountPoint))
       else:
-        slt.execCall("chroot {mp} /usr/sbin/update-grub".format(mp = mountPoint))
+        slt.execCall("chroot {mp} /usr/sbin/update-grub".format(mp=mountPoint))
     else:
       self.__debug("grub2 not installed on the target partition, so grub_mkconfig will directly be used to generate the grub.cfg file")
       # tiny OS installed on that mount point, so we cannot chroot on it to install grub2 config.
       if self.isTest:
-        self.__debug("/usr/sbin/grub-mkconfig -o {cfg}".format(cfg = os.path.join(mountPoint, "boot/grub/grub.cfg")))
+        self.__debug("/usr/sbin/grub-mkconfig -o {cfg}".format(cfg=os.path.join(mountPoint, "boot/grub/grub.cfg")))
       else:
-        slt.execCall("/usr/sbin/grub-mkconfig -o {cfg}".format(cfg = os.path.join(mountpoint, "boot/grub/grub.cfg")))
+        slt.execCall("/usr/sbin/grub-mkconfig -o {cfg}".format(cfg=os.path.join(mountPoint, "boot/grub/grub.cfg")))
 
   def _umountAll(self, mountPoint):
     self.__debug("umountAll")
@@ -123,7 +123,7 @@ class Grub2:
       self._unbindProcSysDev(mountPoint)
       if self._bootInBootMounted:
         self.__debut("/boot mounted in " + mountPoint + ", so umount it")
-        slt.execCall("chroot {mp} /sbin/umount /boot".format(mp = mountPoint))
+        slt.execCall("chroot {mp} /sbin/umount /boot".format(mp=mountPoint))
       if mountPoint != '/':
         self.__debug("umain mount point ≠ '/' → umount " + mountPoint)
         slt.umountDevice(mountPoint)
index 7bc6e9f..4b90245 100644 (file)
@@ -19,8 +19,8 @@ import libsalt as slt
 import subprocess
 from operator import itemgetter
 
+
 class Lilo:
-  
   isTest = False
   _prefix = None
   _tmp = None
@@ -54,7 +54,7 @@ bmp-table = 60,6,1,16
 bmp-timer = 65,29,0,255
 
 # Standard menu.
-# Or, you can comment out the bitmap menu above and 
+# Or, you can comment out the bitmap menu above and
 # use a boot message with the standard menu:
 # message = /boot/boot_message.txt
 
@@ -93,7 +93,7 @@ vga = {vga}
   def __init__(self, isTest):
     self.isTest = isTest
     self._prefix = "bootsetup.lilo-"
-    self._tmp = tempfile.mkdtemp(prefix = self._prefix)
+    self._tmp = tempfile.mkdtemp(prefix=self._prefix)
     slt.mounting._tempMountDir = os.path.join(self._tmp, 'mounts')
     self.__debug("tmp dir = " + self._tmp)
 
@@ -144,10 +144,10 @@ vga = {vga}
     if mountPoint != '/' and os.path.exists(fstab) and os.path.exists(bootdir):
       self.__debug("mp != / and etc/fstab + boot exists, will try to mount /boot by reading fstab")
       try:
-        self.__debug('set -- $(grep /boot {fstab}) && echo "$1,$3"'.format(fstab = fstab))
-        (bootDev, bootType) = slt.execGetOutput('set -- $(grep /boot {fstab}) && echo "$1,$3"'.format(fstab = fstab), shell = True)[0].split(',')
+        self.__debug('set -- $(grep /boot {fstab}) && echo "$1,$3"'.format(fstab=fstab))
+        (bootDev, bootType) = slt.execGetOutput('set -- $(grep /boot {fstab}) && echo "$1,$3"'.format(fstab=fstab), shell=True)[0].split(',')
         if bootDev and not os.path.ismount(bootdir):
-          mp = slt.mountDevice(bootDev, fsType = bootType, mountPoint = bootdir)
+          mp = slt.mountDevice(bootDev, fsType=bootType, mountPoint=bootdir)
           if mp:
             self._bootsMounted.append(mp)
             self.__debug("/boot mounted in " + mp)
@@ -173,20 +173,20 @@ vga = {vga}
           mountPointList[p[0]] = mp
           self._mountBootInPartition(mp)
         else:
-          raise Exception("Cannot mount {d}".format(d = dev))
+          raise Exception("Cannot mount {d}".format(d=dev))
 
   def _umountAll(self, mountPoint, mountPointList):
     self.__debug("umountAll")
     if mountPoint:
       for mp in self._bootsMounted:
         self.__debug("umounting " + unicode(mp))
-        slt.umountDevice(mp, deleteMountPoint = False)
+        slt.umountDevice(mp, deleteMountPoint=False)
       self._bootsMounted = []
       if mountPointList:
         self.__debug("umount other mount points: " + unicode(mountPointList))
         for mp in mountPointList.values():
           if mp == mountPoint:
-            continue # skip it, will be unmounted just next
+            continue  # skip it, will be unmounted just next
           self.__debug("umount " + unicode(mp))
           slt.umountDevice(mp)
       if mountPoint != '/':
@@ -211,7 +211,7 @@ vga = {vga}
           mp = mountPointList[p[0]]
           sections.extend(self._getLinuxLiloSections(device, fs, mp, label))
         else:
-          sys.err.write("The boot type {type} is not supported.\n".format(type = bootType))
+          sys.err.write("The boot type {type} is not supported.\n".format(type=bootType))
     return sections
 
   def _getChainLiloSection(self, device, label):
@@ -222,7 +222,7 @@ vga = {vga}
     return """# {label} chain section
   other = {device}
   label = {label}
-""".format(device = device, label = label)
+""".format(device=device, label=label)
 
   def _getLinuxLiloSections(self, device, fs, mp, label):
     """
@@ -230,17 +230,17 @@ vga = {vga}
     """
     sections = []
     self.__debug("Section 'linux' for " + device + "/" + fs + ", mounted on " + mp + " with label: " + label)
-    kernelList = sorted(glob.glob("{mp}/boot/vmlinuz*".format(mp = mp)))
-    initrdList = sorted(glob.glob("{mp}/boot/initr*".format(mp = mp)))
+    kernelList = sorted(glob.glob("{mp}/boot/vmlinuz*".format(mp=mp)))
+    initrdList = sorted(glob.glob("{mp}/boot/initr*".format(mp=mp)))
     for l in (kernelList, initrdList):
       for el in l:
         if os.path.isdir(el) or os.path.islink(el):
           l.remove(el)
     self.__debug("kernelList: " + unicode(kernelList))
     self.__debug("initrdList: " + unicode(initrdList))
-    uuid = slt.execGetOutput(['/sbin/blkid', '-s', 'UUID', '-o', 'value', device], shell = False)
+    uuid = slt.execGetOutput(['/sbin/blkid', '-s', 'UUID', '-o', 'value', device], shell=False)
     if uuid:
-      rootDevice = "/dev/disk/by-uuid/{uuid}".format(uuid = uuid[0])
+      rootDevice = "/dev/disk/by-uuid/{uuid}".format(uuid=uuid[0])
     else:
       rootDevice = device
     self.__debug("rootDevice = " + rootDevice)
@@ -252,15 +252,15 @@ vga = {vga}
   image = {image}
   initrd = {initrd}
   root = {root}
-""".format(image = k, initrd = i, root = rootDevice, label = l)
+""".format(image=k, initrd=i, root=rootDevice, label=l)
       else:
         section = """# {label} Linux section
   image = {image}
   root = {root}
-""".format(image = k, root = rootDevice, label = l)
+""".format(image=k, root=rootDevice, label=l)
       if fs == 'ext4':
-        section += '  append = "{append} "\n'.format(append = 'rootfstype=ext4')
-      section += "  read-only\n  label = {label}\n".format(label = l)
+        section += '  append = "{append} "\n'.format(append='rootfstype=ext4')
+      section += "  read-only\n  label = {label}\n".format(label=l)
       sections.append(section)
     return sections
 
@@ -270,17 +270,17 @@ vga = {vga}
       if len(kernelList) == 1:
         initrd = None
         if initrdList:
-          initrd = initrdList[0] # assume the only initrd match the only kernel
+          initrd = initrdList[0]  # assume the only initrd match the only kernel
         ret.append([kernelList[0], initrd, labelRef])
       else:
-        labelBase = labelRef[0:15-2] + "-"
+        labelBase = labelRef[0:15 - 2] + "-"
         n = 0
         for kernel in kernelList:
           n += 1
           kernelSuffix = os.path.basename(kernel).replace("vmlinuz", "")
           initrd = None
           for i in initrdList:
-            if kernelSuffix in i: # find the matching initrd
+            if kernelSuffix in i:  # find the matching initrd
               initrd = i
               break
           ret.append((kernel, initrd, labelBase + unicode(n)))
@@ -394,7 +394,7 @@ vga = {vga}
       yMax = None
       dMax = None
       # order the vesa modes by vertical size desc, horizontal size desc, color depth desc.
-      for vesaMode in sorted(vesaModes, key = itemgetter(1, 0, 2), reverse = True):
+      for vesaMode in sorted(vesaModes, key=itemgetter(1, 0, 2), reverse=True):
         (x, y, d, m) = vesaMode
         if m:
           self.__debug("trying {0} for y, {1} for x and {2} for d".format(y, x, d))
@@ -405,8 +405,8 @@ vga = {vga}
             mode = m
             break
       if mode:
-        self.__debug("Max mode found: {x}×{y}×{d}".format(x = xMax, y = yMax, d = dMax))
-        label = "{x}x{y}x{d}".format(x = xMax, y = yMax, d = dMax)
+        self.__debug("Max mode found: {x}×{y}×{d}".format(x=xMax, y=yMax, d=dMax))
+        label = "{x}x{y}x{d}".format(x=xMax, y=yMax, d=dMax)
     if not mode:
       mode = 'normal'
       label = 'text'
@@ -436,7 +436,7 @@ vga = {vga}
       (fb, fbLabel) = self._getFrameBufferConf()
       self.__debug("frame buffer mode = " + unicode(fb) + " " + unicode(fbLabel))
       f = open(self.getConfigurationPath(), "w")
-      f.write(self._cfgTemplate.format(boot = self._mbrDevice, mp = mp, vga = "{0} # {1}".format(fb, fbLabel)))
+      f.write(self._cfgTemplate.format(boot=self._mbrDevice, mp=mp, vga="{0} # {1}".format(fb, fbLabel)))
       for s in liloSections:
         f.write(s)
         f.write("\n")
@@ -470,9 +470,9 @@ vga = {vga}
         shutil.copyfile(self.getConfigurationPath(), os.path.join(mp, '/etc/bootsetup/lilo.conf'))
         # run lilo
         if self.isTest:
-          self.__debug('/sbin/lilo -t -v -C {mp}/etc/bootsetup/lilo.conf'.format(mp = mp))
-          slt.execCall('/sbin/lilo -t -v -C {mp}/etc/bootsetup/lilo.conf'.format(mp = mp))
+          self.__debug('/sbin/lilo -t -v -C {mp}/etc/bootsetup/lilo.conf'.format(mp=mp))
+          slt.execCall('/sbin/lilo -t -v -C {mp}/etc/bootsetup/lilo.conf'.format(mp=mp))
         else:
-          slt.execCall('/sbin/lilo -C {mp}/etc/bootsetup/lilo.conf'.format(mp = mp))
+          slt.execCall('/sbin/lilo -C {mp}/etc/bootsetup/lilo.conf'.format(mp=mp))
       finally:
         self._umountAll(mp, mpList)