class HddFastRemove(Screen): def __init__(self, session): Screen.__init__(self, session) self.mdisks = Disks() self.mountpoints = MountPoints() self.mountpoints.read() self.disks = list () self.mounts = list () for disk in self.mdisks.disks: if disk[2] == True: diskname = disk[3] for partition in disk[5]: mp = "" rmp = "" try: mp = self.mountpoints.get(partition[0][:3], int(partition[0][3:])) rmp = self.mountpoints.getRealMount(partition[0][:3], int(partition[0][3:])) except Exception, e: pass if len(mp) > 0: self.disks.append(MountEntry(disk[3], "P.%s (Fixed: %s)" % (partition[0][3:], mp))) self.mounts.append(mp) elif len(rmp) > 0: self.disks.append(MountEntry(disk[3], "P.%s (Fast: %s)" % (partition[0][3:], rmp))) self.mounts.append(rmp) self["menu"] = List(self.disks) self["key_red"] = Button(_("Umount")) self["key_blue"] = Button(_("Exit")) self["actions"] = ActionMap(["OkCancelActions", "ColorActions"], { "blue": self.quit, "red": self.red, "cancel": self.quit, }, -2)
class HddPartitions(Screen): def __init__(self, session, disk): self.session = session Screen.__init__(self, session) self.disk = disk self.refreshMP(False) self["menu"] = List(self.partitions) self["menu"].onSelectionChanged.append(self.selectionChanged) self["key_red"] = Button("") self["key_green"] = Button("") self["key_yellow"] = Button("") self["key_blue"] = Button(_("Exit")) self["label_disk"] = Label("%s - %s" % (self.disk[0], self.disk[3])) self["actions"] = ActionMap(["OkCancelActions", "ColorActions"], { "blue": self.quit, "yellow": self.yellow, "green": self.green, "red": self.red, "cancel": self.quit, }, -2) if len(self.disk[5]) > 0: if self.disk[5][0][2] == "Linux": self["key_green"].setText(_("Check")) self["key_yellow"].setText(_("Format")) if self.disk[5][0][2] == "Linux swap": self["key_red"].setText("") else: mp = self.mountpoints.get(self.disk[0], 1) rmp = self.mountpoints.getRealMount(self.disk[0], 1) if len(mp) > 0 or len(rmp) > 0: self.mounted = True self["key_red"].setText(_("Umount")) else: self.mounted = False self["key_red"].setText(_("Mount")) def selectionChanged(self): self["key_green"].setText("") self["key_yellow"].setText("") if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][2] == "Linux": self["key_green"].setText(_("Check")) self["key_yellow"].setText(_("Format")) if self.disk[5][index][2] == "Linux swap": self["key_red"].setText("") else: mp = self.mountpoints.get(self.disk[0], index+1) rmp = self.mountpoints.getRealMount(self.disk[0], index+1) if len(mp) > 0 or len(rmp) > 0: self.mounted = True self["key_red"].setText(_("Umount")) else: self.mounted = False self["key_red"].setText(_("Mount")) def chkfs(self): disks = Disks() ret = disks.chkfs(self.disk[5][self.index][0][:3], self.index+1) if ret == 0: self.session.open(MessageBox, _("Check disk terminated with success"), MessageBox.TYPE_INFO) elif ret == -1: self.session.open(MessageBox, _("Cannot umount current drive.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again"), MessageBox.TYPE_ERROR) else: self.session.open(MessageBox, _("Error checking disk. The disk may be damaged"), MessageBox.TYPE_ERROR) def mkfs(self): disks = Disks() ret = disks.mkfs(self.disk[5][self.index][0][:3], self.index+1) if ret == 0: self.session.open(MessageBox, _("Format terminated with success"), MessageBox.TYPE_INFO) elif ret == -2: self.session.open(MessageBox, _("Cannot format current drive.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again"), MessageBox.TYPE_ERROR) else: self.session.open(MessageBox, _("Error formatting disk. The disk may be damaged"), MessageBox.TYPE_ERROR) def green(self): if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][2] == "Linux": self.index = index self.session.open(ExtraActionBox, "Checking disk %s" % self.disk[5][index][0], "Checking disk", self.chkfs) def yellow(self): if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][2] == "Linux": self.index = index self.session.open(ExtraActionBox, "Formatting disk %s" % self.disk[5][index][0], "Formatting disk", self.mkfs) def refreshMP(self, uirefresh = True): self.partitions = [] self.mountpoints = MountPoints() self.mountpoints.read() count = 1 for part in self.disk[5]: capacity = "%d MB" % (part[1] / (1024 * 1024)) mp = self.mountpoints.get(self.disk[0], count) rmp = self.mountpoints.getRealMount(self.disk[0], count) if len(mp) > 0: self.partitions.append(PartitionEntry("P. %d - %s (Fixed: %s)" % (count, part[2], mp), capacity)) elif len(rmp) > 0: self.partitions.append(PartitionEntry("P. %d - %s (Fast: %s)" % (count, part[2], rmp), capacity)) else: self.partitions.append(PartitionEntry("P. %d - %s" % (count, part[2]), capacity)) count += 1 if uirefresh: self["menu"].setList(self.partitions) def red(self): if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][2] == "Linux swap": return if len(self.partitions) > 0: self.sindex = self['menu'].getIndex() if self.mounted: mp = self.mountpoints.get(self.disk[0], self.sindex+1) rmp = self.mountpoints.getRealMount(self.disk[0], self.sindex+1) if len(mp) > 0: if self.mountpoints.isMounted(mp): if self.mountpoints.umount(mp): self.mountpoints.delete(mp) self.mountpoints.write() else: self.session.open(MessageBox, _("Cannot umount device.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again"), MessageBox.TYPE_ERROR) else: self.mountpoints.delete(mp) self.mountpoints.write() elif len(rmp) > 0: self.mountpoints.umount(rmp) self.refreshMP() else: self.session.openWithCallback(self.refreshMP, HddMount, self.disk[0], self.sindex+1) def quit(self): self.close()
class HddPartitions(Screen): def __init__(self, session, disk): self.session = session Screen.__init__(self, session) self.disk = disk self.refreshMP(False) self["menu"] = List(self.partitions) self["menu"].onSelectionChanged.append(self.selectionChanged) self["key_red"] = Button("") self["key_green"] = Button("") self["key_yellow"] = Button("") self["key_blue"] = Button(_("Exit")) self["label_disk"] = Label("%s - %s" % (self.disk[0], self.disk[3])) self["actions"] = ActionMap( ["OkCancelActions", "ColorActions"], { "blue": self.quit, "yellow": self.yellow, "green": self.green, "red": self.red, "cancel": self.quit, }, -2) if len(self.disk[5]) > 0: if self.disk[5][0][3] == "83" or self.disk[5][0][ 3] == "7" or self.disk[5][0][3] == "b": self["key_green"].setText(_("Check")) self["key_yellow"].setText(_("Format")) mp = self.mountpoints.get(self.disk[0], 1) rmp = self.mountpoints.getRealMount(self.disk[0], 1) if len(mp) > 0 or len(rmp) > 0: self.mounted = True self["key_red"].setText(_("Umount")) else: self.mounted = False self["key_red"].setText(_("Mount")) def selectionChanged(self): self["key_green"].setText("") self["key_yellow"].setText("") self["key_red"].setText("") if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][3] == "83" or self.disk[5][index][ 3] == "7" or self.disk[5][index][3] == "b": self["key_green"].setText(_("Check")) self["key_yellow"].setText(_("Format")) mp = self.mountpoints.get(self.disk[0], index + 1) rmp = self.mountpoints.getRealMount(self.disk[0], index + 1) if len(mp) > 0 or len(rmp) > 0: self.mounted = True self["key_red"].setText(_("Umount")) else: self.mounted = False self["key_red"].setText(_("Mount")) def chkfs(self): disks = Disks() ret = disks.chkfs(self.disk[5][self.index][0][:3], self.index + 1, self.fstype) if ret == 0: self.session.open(MessageBox, _("Check disk terminated with success"), MessageBox.TYPE_INFO) elif ret == -1: self.session.open( MessageBox, _("Cannot umount current drive.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again" ), MessageBox.TYPE_ERROR) else: self.session.open( MessageBox, _("Error checking disk. The disk may be damaged"), MessageBox.TYPE_ERROR) def mkfs(self): disks = Disks() ret = disks.mkfs(self.disk[5][self.index][0][:3], self.index + 1, self.fstype) if ret == 0: self.session.open(MessageBox, _("Format terminated with success"), MessageBox.TYPE_INFO) elif ret == -2: self.session.open( MessageBox, _("Cannot format current drive.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again" ), MessageBox.TYPE_ERROR) else: self.session.open( MessageBox, _("Error formatting disk. The disk may be damaged"), MessageBox.TYPE_ERROR) def isExt4Supported(self): return "ext4" in open("/proc/filesystems").read() def domkfs(self, result): if self.disk[5][self.index][3] == "83": if self.isExt4Supported(): if result < 2: self.fstype = result self.session.open( ExtraActionBox, "Formatting disk %s" % self.disk[5][self.index][0], "Formatting disk", self.mkfs) else: if result < 1: self.fstype = 1 self.session.open( ExtraActionBox, "Formatting disk %s" % self.disk[5][self.index][0], "Formatting disk", self.mkfs) elif self.disk[5][self.index][3] == "7": if result < 1: self.fstype = 2 self.session.open( ExtraActionBox, "Formatting disk %s" % self.disk[5][self.index][0], "Formatting disk", self.mkfs) elif self.disk[5][self.index][3] == "b": if result < 1: self.fstype = 3 self.session.open( ExtraActionBox, "Formatting disk %s" % self.disk[5][self.index][0], "Formatting disk", self.mkfs) def green(self): if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][3] == "83" or self.disk[5][index][ 3] == "7" or self.disk[5][index][3] == "b": self.index = index if self.disk[5][index][3] == "83": self.fstype = 0 elif self.disk[5][index][3] == "7": self.fstype = 2 elif self.disk[5][index][3] == "b": self.fstype = 3 self.session.open(ExtraActionBox, "Checking disk %s" % self.disk[5][index][0], "Checking disk", self.chkfs) def yellow(self): if len(self.disk[5]) > 0: self.index = self["menu"].getIndex() if self.disk[5][self.index][3] == "83": if self.isExt4Supported(): self.session.openWithCallback( self.domkfs, ExtraMessageBox, "Format as", "HDD Partitioner", [ ["Ext4", "partitionmanager.png"], ["Ext3", "partitionmanager.png"], ["Cancel", "cancel.png"], ], 1, 2) else: self.session.openWithCallback( self.domkfs, ExtraMessageBox, "Format as", "HDD Partitioner", [ ["Ext3", "partitionmanager.png"], ["Cancel", "cancel.png"], ], 1, 1) elif self.disk[5][self.index][3] == "7": self.session.openWithCallback( self.domkfs, ExtraMessageBox, "Format as", "HDD Partitioner", [ ["NTFS", "partitionmanager.png"], ["Cancel", "cancel.png"], ], 1, 1) elif self.disk[5][self.index][3] == "b": self.session.openWithCallback( self.domkfs, ExtraMessageBox, "Format as", "HDD Partitioner", [ ["Fat32", "partitionmanager.png"], ["Cancel", "cancel.png"], ], 1, 1) def refreshMP(self, uirefresh=True): self.partitions = [] self.mountpoints = MountPoints() self.mountpoints.read() count = 1 for part in self.disk[5]: capacity = "%d MB" % (part[1] / (1024 * 1024)) mp = self.mountpoints.get(self.disk[0], count) rmp = self.mountpoints.getRealMount(self.disk[0], count) if len(mp) > 0: self.partitions.append( PartitionEntry( "P. %d - %s (Fixed: %s)" % (count, part[2], mp), capacity)) elif len(rmp) > 0: self.partitions.append( PartitionEntry( "P. %d - %s (Fast: %s)" % (count, part[2], rmp), capacity)) else: self.partitions.append( PartitionEntry("P. %d - %s" % (count, part[2]), capacity)) count += 1 if uirefresh: self["menu"].setList(self.partitions) def red(self): if len(self.disk[5]) > 0: index = self["menu"].getIndex() if self.disk[5][index][3] != "83" and self.disk[5][index][ 3] != "7" and self.disk[5][index][3] != "b": return if len(self.partitions) > 0: self.sindex = self['menu'].getIndex() if self.mounted: mp = self.mountpoints.get(self.disk[0], self.sindex + 1) rmp = self.mountpoints.getRealMount(self.disk[0], self.sindex + 1) if len(mp) > 0: if self.mountpoints.isMounted(mp): if self.mountpoints.umount(mp): self.mountpoints.delete(mp) self.mountpoints.write() else: self.session.open( MessageBox, _("Cannot umount device.\nA record in progress, timeshit or some external tools (like samba and nfsd) may cause this problem.\nPlease stop this actions/applications and try again" ), MessageBox.TYPE_ERROR) else: self.mountpoints.delete(mp) self.mountpoints.write() elif len(rmp) > 0: self.mountpoints.umount(rmp) self.refreshMP() else: self.session.openWithCallback(self.refreshMP, HddMount, self.disk[0], self.sindex + 1) def quit(self): self.close()