3155c44d9ab2b1f4d7780807e60486049f93781d
[vuplus_dvbapp] / lib / python / Screens / ChannelSelection.py
1 from Screen import Screen
2 from Components.Button import Button
3 from Components.ServiceList import ServiceList
4 from Components.ActionMap import ActionMap
5
6 from enigma import eServiceReference
7
8 from Screens.Menu import FixedMenu
9
10 import xml.dom.minidom
11
12 class ChannelContextMenu(FixedMenu):
13         def __init__(self, session, csel):
14                 self.csel = csel
15                 
16                 menu = [("back", self.close)]
17                 
18                 if csel.movemode:
19                         menu.append(("disable move mode", self.moveMode))
20                 else:
21                         menu.append(("enable move mode", self.moveMode))
22
23                 if csel.bouquet_mark_edit:
24                         menu.append(("end bouquet edit", self.bouquetMarkEnd))
25                         menu.append(("abort bouquet edit", self.bouquetMarkAbort))
26                 else:
27                         menu.append(("edit bouquet...", self.bouquetMarkStart))
28                 
29                 FixedMenu.__init__(self, session, "Channel Selection", menu)
30                 self.skinName = "Menu"
31
32         def moveMode(self):
33                 self.csel.setMoveMode(self.csel.movemode)
34                 self.close()
35         
36         def bouquetMarkStart(self):
37                 self.csel.startMarkedEdit()
38                 self.close()
39         
40         def bouquetMarkEnd(self):
41                 self.csel.endMarkedEdit(abort=False)
42                 self.close()
43
44         def bouquetMarkAbort(self):
45                 self.csel.endMarkedEdit(abort=True)
46                 self.close()
47  
48 class ChannelSelection(Screen):
49         def __init__(self, session):
50                 Screen.__init__(self, session)
51                 
52                 self.movemode = False
53                 self.bouquet_mark_edit = False
54                 
55                 ## FIXME
56                 self.__marked = [ ]
57                 
58                 self["key_red"] = Button("red")
59                 self["key_green"] = Button("green")
60                 self["key_yellow"] = Button("yellow")
61                 self["key_blue"] = Button("blue")
62                 
63                 self["list"] = ServiceList()
64                 self["list"].setRoot(eServiceReference("""1:0:1:0:0:0:0:0:0:0:(type == 1)"""))
65                 
66                 #self["okbutton"] = Button("ok", [self.channelSelected])
67                 
68                 class ChannelActionMap(ActionMap):
69                         def action(self, contexts, action):
70                                 if action[:7] == "bouquet":
71                                         print "setting root to " + action[8:]
72                                         l = self.csel["list"]
73                                         l.setMode(l.MODE_NORMAL)
74                                         l.setRoot(eServiceReference("1:0:1:0:0:0:0:0:0:0:" + action[8:]))
75                                 else:
76                                         ActionMap.action(self, contexts, action)
77
78                 self["actions"] = ChannelActionMap(["ChannelSelectActions", "OkCancelActions"], 
79                         {
80                                 "cancel": self.close,
81                                 "ok": self.channelSelected,
82                                 "mark": self.doMark,
83                                 "contextMenu": self.doContext,
84                                 "showFavourites": self.showFavourites
85                         })
86                 self["actions"].csel = self
87
88         #  marked edit mode
89         
90         def startMarkedEdit(self):
91                 self.bouquet_mark_edit = True
92                 self.clearMarks()
93                 
94                 # TODO
95                 marked = self.__marked
96                 
97                 l = self["list"]
98                 for x in marked:
99                         print "mark: " + str(x)
100                         l.addMarked(x)
101                 
102         def endMarkedEdit(self, abort):
103                 self.bouquet_mark_edit = True
104                 new_marked = self["list"].getMarked()
105                 self.__marked = new_marked
106                 self.clearMarks()
107                 self.bouquet_mark_edit = False
108
109         def clearMarks(self):
110                 self["list"].clearMarks()
111         
112         def doMark(self):
113                 if not self.bouquet_mark_edit:
114                         return
115                 
116                 ref = self["list"].getCurrent()
117                 if self["list"].isMarked(ref):
118                         self["list"].removeMarked(ref)
119                 else:
120                         self["list"].addMarked(ref)
121         
122         # ...
123         def channelSelected(self):
124                 self.session.nav.playService(self["list"].getCurrent())
125                 print "current: " + self["list"].getCurrent().toString()
126                 self.close()
127
128         #called from infoBar
129         def zap(self):
130                 self.session.nav.playService(self["list"].getCurrent())
131
132         def moveUp(self):
133                 self["list"].moveUp()
134
135         def moveDown(self):
136                 self["list"].moveDown()
137
138         def doContext(self):
139                 self.session.open(ChannelContextMenu, self)
140
141         def setMoveMode(self, mode):
142                 self.movemode = mode
143         
144         def showFavourites(self):
145                 l = self["list" ]
146                 l.setRoot(eServiceReference('1:0:1:0:0:0:0:0:0:0:(provider == "fav")'))
147                 l.setMode(l.MODE_FAVOURITES)