make some stuff even simpler
[enigma2.git] / lib / python / Screens / HarddiskSetup.py
1 from Screen import Screen
2 from Components.ActionMap import ActionMap
3 from Components.Harddisk import harddiskmanager                 #global harddiskmanager
4 from Components.MenuList import MenuList
5 from Components.Label import Label
6 from Screens.MessageBox import MessageBox
7
8 class HarddiskSetup(Screen):
9         def __init__(self, session, hdd):
10                 Screen.__init__(self, session)
11                 self.hdd = hdd
12                 
13                 self["model"] = Label("Model: " + hdd.model())
14                 self["capacity"] = Label("Capacity: " + hdd.capacity())
15                 self["bus"] = Label("Bus: " + hdd.bus())
16                 self["initialize"] = Label("Initialize")
17
18                 self["actions"] = ActionMap(["OkCancelActions"],
19                 {
20                         "ok": self.close,
21                         "cancel": self.close
22                 })
23                 
24                 self["shortcuts"] = ActionMap(["ShortcutActions"],
25                 {
26                         "red": self.hddInitialize
27                 })
28
29         def hddInitialize(self):
30                 if self.hdd.getIndex() == 2:            #CF
31                         print "not a good idea!"
32                         self.session.open(MessageBox, "not a good idea - this will kill our rootfs!")
33                 else:   
34                         print "this will start the initialize now!"
35                         #self.hdd.initialize()
36
37 class HarddiskSelection(Screen):
38         def __init__(self, session):
39                 Screen.__init__(self, session)
40
41                 self["hddlist"] = MenuList(harddiskmanager.HDDList())
42                 
43                 self["actions"] = ActionMap(["OkCancelActions"],
44                 {
45                         "ok": self.okbuttonClick ,
46                         "cancel": self.close
47                 })
48
49         def okbuttonClick(self):
50                 selection = self["hddlist"].getCurrent()
51                 self.session.open(HarddiskSetup, selection[1])
52                 print "ok"
53                 pass