hdd format (initialize) works now

[enigma2.git] / lib / python / Screens / HarddiskSetup.py

diff --git a/lib/python/Screens/HarddiskSetup.py b/lib/python/Screens/HarddiskSetup.py
index 3fea87a8b568c7a7fb750c099f8beb972b15f9fc..a2b3ed4f8500160cee6d1efe0fcee8c68b8906a0 100644 (file)
--- a/lib/python/Screens/HarddiskSetup.py
+++ b/lib/python/Screens/HarddiskSetup.py
@@ -1,11 +1,56 @@
 from Screen import Screen
 from Components.ActionMap import ActionMap
+from Components.Harddisk import harddiskmanager                        #global harddiskmanager
+from Components.MenuList import MenuList
+from Components.Label import Label
+
+class HarddiskSetup(Screen):
+       def __init__(self, session, hdd):
+               Screen.__init__(self, session)
+               self.hdd = hdd
+               
+               cap = hdd.capacity() / 1000 * 512 / 1000
+               capstr = "Capacity: %d.%03d GB" % (cap / 1000, cap % 1000)
+
+               self["model"] = Label("Model: " + hdd.model())
+               self["capacity"] = Label(capstr)
+               
+               if hdd.index & 1:
+                       busstr = "Slave"
+               else:   
+                       busstr = "Master"
+               
+               self["bus"] = Label("Bus: " + busstr)
+               self["initialize"] = Label("Initialize")
+
+               self["actions"] = ActionMap(["OkCancelActions"],
+               {
+                       "ok": self.close,
+                       "cancel": self.close
+               })
+               
+               self["shortcuts"] = ActionMap(["ShortcutActions"],
+               {
+                       "red": self.hddInitialize
+               })
+
+       def hddInitialize(self):
+               self.hdd.initialize()
 
 class HarddiskSelection(Screen):
-  def __init__(self, session):
-    Screen.__init__(self, session)
+       def __init__(self, session):
+               Screen.__init__(self, session)
+
+               self["hddlist"] = MenuList(harddiskmanager.HDDList())
+               
+               self["actions"] = ActionMap(["OkCancelActions"],
+               {
+                       "ok": self.okbuttonClick ,
+                       "cancel": self.close
+               })
 
-    self["actions"] = ActionMap(["OkCancelActions"],
-      {
-        "cancel": self.close
-      })
+       def okbuttonClick(self):
+               selection = self["hddlist"].getCurrent()
+               self.session.open(HarddiskSetup, selection[1])
+               print "ok"
+               pass