+
+ def unmount(self):
+ cmd = "/bin/umount " + self.devidex + "part*"
+ res = system(cmd)
+ return (res >> 8)
+
+ def createPartition(self):
+ cmd = "/sbin/sfdisk -f " + self.devidex + "disc"
+ sfdisk = popen(cmd, "w")
+ sfdisk.write("0,\n;\n;\n;\ny\n")
+ sfdisk.close()
+ return 0
+
+ def mkfs(self):
+ cmd = "/sbin/mkfs.ext3 -T largefile -m0 " + self.devidex + "part1"
+ res = system(cmd)
+ return (res >> 8)
+
+ def mount(self):
+ cmd = "/bin/mount -t ext3 " + self.devidex + "part1 /hdd"
+ res = system(cmd)
+ return (res >> 8)
+
+ def createMovieFolder(self):
+ res = system("mkdir " + resolveFilename(SCOPE_HDD))
+ return (res >> 8)
+
+ errorList = [ _("Everything is fine"), _("Creating partition failed"), _("Mkfs failed"), _("Mount failed"), _("Create movie folder failed"), _("Unmount failed")]
+
+ def initialize(self):
+ self.unmount()
+
+ if self.createPartition() != 0:
+ return -1
+
+ if self.mkfs() != 0:
+ return -2
+
+ if self.mount() != 0:
+ return -3
+
+ if self.createMovieFolder() != 0:
+ return -4
+
+ return 0
+
+def existHDD(num):
+ mediafile = tryOpen(num2prochdx(num) + "media")
+
+ if mediafile == "":
+ return -1
+
+ line = mediafile.readline()
+ mediafile.close()
+
+ if line.startswith("disk"):
+ return 1
+
+ return -1
+
+class Partition:
+ def __init__(self, mountpoint, description = ""):
+ self.mountpoint = mountpoint
+ self.description = description
+
+ def stat(self):
+ return statvfs(self.mountpoint)
+
+ def free(self):
+ try:
+ s = self.stat()
+ return s.f_bavail * s.f_bsize
+ except OSError:
+ return None
+
+ def total(self):
+ try:
+ s = self.stat()
+ return s.f_blocks * s.f_bsize
+ except OSError:
+ return None
+
+ def mounted(self):
+ # THANK YOU PYTHON FOR STRIPPING AWAY f_fsid.
+ # TODO: can os.path.ismount be used?
+ procfile = tryOpen("/proc/mounts")
+ for n in procfile.readlines():
+ if n.split(' ')[1] == self.mountpoint:
+ return True
+ return False
+
+class HarddiskManager:
+ def __init__(self):
+ hddNum = 0
+ self.hdd = [ ]
+
+ self.partitions = [ ]
+
+ for hddNum in range(8):
+ if existHDD(hddNum):
+ hdd = Harddisk(hddNum)
+ self.hdd.append(hdd)
+
+ # currently, this is just an enumeration of what's possible,
+ # this probably has to be changed to support automount stuff.
+ # still, if stuff is mounted into the correct mountpoints by
+ # external tools, everything is fine (until somebody inserts
+ # a second usb stick.)
+ p = [
+ ("/media/hdd", _("Harddisk")),
+ ("/media/card", _("Card")),
+ ("/media/cf", _("Compact Flash")),
+ ("/media/mmc1", _("MMC Card")),
+ ("/media/net", _("Network Mount")),
+ ("/media/ram", _("Ram Disk")),
+ ("/media/usb", _("USB Stick")),
+ ("/", _("Internal Flash"))
+ ]
+
+ for x in p:
+ self.partitions.append(Partition(mountpoint = x[0], description = x[1]))
+
+ def HDDCount(self):
+ return len(self.hdd)
+
+ def HDDList(self):
+ list = [ ]
+ for hd in self.hdd:
+ hdd = hd.model() + " ("
+ hdd += hd.bus()
+ cap = hd.capacity()
+ if cap != "":
+ hdd += ", " + cap
+ hdd += ")"
+ list.append((hdd, hd))
+
+ return list
+
+ def getMountedPartitions(self):
+ return [x for x in self.partitions if x.mounted()]
+
+harddiskmanager = HarddiskManager()