make some stuff even simpler

[enigma2.git] / lib / python / Components / Harddisk.py

diff --git a/lib/python/Components/Harddisk.py b/lib/python/Components/Harddisk.py
index c50f0b8c435c1e8dd1698906a2b07e19e7891c26..dcf7657058de9bfff572629cd8540e7e2729a94d 100644 (file)
--- a/lib/python/Components/Harddisk.py
+++ b/lib/python/Components/Harddisk.py
@@ -21,11 +21,21 @@ class Harddisk:
                self.prochdx = num2prochdx(index)
                self.devidex = "/dev/ide/host%d/bus%d/target%d/lun0/" % (host, bus, target)
 
+       def getIndex(self):
+               return self.index
+
+       def bus(self):
+               #TODO: add the host
+               if self.index & 1:
+                       return "Slave"
+               else:
+                       return "Master"
+
        def capacity(self):
                procfile = tryOpen(self.prochdx + "capacity")
                
                if procfile == "":
-                       return -1
+                       return ""
 
                line = procfile.readline()
                procfile.close()
@@ -33,10 +43,12 @@ class Harddisk:
                try:
                        cap = int(line)
                except:
-                       return -1
+                       return ""
                
-               return cap      
-                                               
+               cap = cap / 1000 * 512 / 1000
+               
+               return "%d.%03d GB" % (cap/1024, cap%1024)
+                                                               
        def model(self):
                procfile = tryOpen(self.prochdx + "model")
                
@@ -150,28 +162,20 @@ class HarddiskManager:
                        
                        if hddNum > 8:
                                break
-               
+
        def HDDList(self):
                list = [ ]
                for hd in self.hdd:
-                       cap = hd.capacity() / 1000 * 512 / 1000
-                       print cap
                        hdd = hd.model() + " (" 
-                       if hd.index & 1:
-                               hdd += "slave"
-                       else:   
-                               hdd += "master"
-                       if cap > 0:
-                               hdd += ", %d.%03d GB" % (cap/1024, cap%1024)
+                       hdd += hd.bus()
+                       cap = hd.capacity()     
+                       if cap != "":
+                               hdd += ", " + cap
                        hdd += ")"
-
-                       print hdd
-                       
-#                      if hd.index == 0:
-#                              if hd.initialize() == 0:
-#                                      print "hdd status ok"
-#                              else:
-#                                      print "hdd status ok"
-
                        list.append((hdd, hd))
+
                return list
+
+harddiskmanager = HarddiskManager()
+
+