turn magic into python
[enigma2.git] / lib / python / Components / MovieList.py
1 from HTMLComponent import *
2 from GUIComponent import *
3 from Tools.FuzzyDate import FuzzyTime
4 from ServiceReference import ServiceReference
5 from Components.MultiContent import MultiContentEntryText, RT_HALIGN_LEFT, RT_HALIGN_RIGHT
6
7 from enigma import eListboxPythonMultiContent, eListbox, gFont, iServiceInformation
8
9 from enigma import eServiceReference, eServiceCenter, \
10         eServiceCenterPtr, iListableServicePtr, \
11         iStaticServiceInformationPtr
12
13 #
14 # | name of movie              |
15 #
16 def MovieListEntry(serviceref, serviceHandler):
17         if serviceref.flags & eServiceReference.mustDescent:
18                 return None
19
20         info = serviceHandler.info(serviceref)
21         
22         if info is None:
23                 # ignore service which refuse to info
24                 return None
25         
26         len = info.getLength(serviceref)
27         if len > 0:
28                 len = "%d:%02d" % (len / 60, len % 60)
29         else:
30                 len = "?:??"
31         
32         begin = info.getInfo(serviceref, iServiceInformation.sTimeCreate)
33         res = [ (serviceref, begin) ]
34
35         res.append(MultiContentEntryText(pos=(0, 0), size=(560, 30), font = 0, flags = RT_HALIGN_LEFT, text = info.getName(serviceref)))
36         
37         description = info.getInfoString(serviceref, iServiceInformation.sDescription)
38
39         begin_string = ""
40         if begin > 0:
41                 t = FuzzyTime(begin)
42                 begin_string = t[0] + ", " + t[1]
43         
44         res.append(MultiContentEntryText(pos=(0, 30), size=(560, 20), font=1, flags=RT_HALIGN_LEFT, text=description))
45         res.append(MultiContentEntryText(pos=(0, 50), size=(270, 20), font=1, flags=RT_HALIGN_LEFT, text=begin_string))
46         res.append(MultiContentEntryText(pos=(290, 50), size=(270, 20), font=1, flags=RT_HALIGN_RIGHT, text=len))
47         
48         return res
49
50 class MovieList(HTMLComponent, GUIComponent):
51         def __init__(self, root):
52                 GUIComponent.__init__(self)
53                 self.l = eListboxPythonMultiContent()
54                 if root is not None:
55                         self.reload(root)
56                 self.l.setFont(0, gFont("Regular", 30))
57                 self.l.setFont(1, gFont("Regular", 18))
58                 
59         def moveToIndex(self, index):
60                 self.instance.moveSelectionTo(index)
61
62         def getCurrentIndex(self):
63                 return self.instance.getCurrentIndex()
64
65         def getCurrent(self):
66                 return self.l.getCurrentSelection()[0]
67         
68         def GUIcreate(self, parent):
69                 self.instance = eListbox(parent)
70                 self.instance.setContent(self.l)
71                 self.instance.setItemHeight(75)
72         
73         def GUIdelete(self):
74                 self.instance.setContent(None)
75                 self.instance = None
76
77         def reload(self, root = None):
78                 if root is not None:
79                         self.load(root)
80                 else:
81                         self.load(self.root)
82                 self.l.setList(self.list)
83
84         def removeService(self, service):
85                 for l in self.list[:]:
86                         if l[0][0] == service:
87                                 self.list.remove(l)
88                 self.l.setList(self.list)
89
90         def load(self, root):
91                 # this lists our root service, then building a 
92                 # nice list
93                 
94                 self.list = [ ]
95                 self.root = root
96                 
97                 serviceHandler = eServiceCenter.getInstance()
98                 list = serviceHandler.list(root)
99                 
100                 if list is None:
101                         raise "listing of movies failed"
102
103                 movieList = [ ]
104                 while 1:
105                         s = list.getNext()
106                         if not s.valid():
107                                 del list
108                                 break
109                         movieList.append(s)
110                 
111                 # now process them...
112                 for ref in movieList:
113                         a = MovieListEntry(ref, serviceHandler)
114                         if a is not None:
115                                 self.list.append(a)
116                 
117                 self.list.sort(key=lambda x: -x[0][1])
118
119         def moveTo(self, serviceref):
120                 found = 0
121                 count = 0
122                 for x in self.list:
123                         if str(ServiceReference(x[0][0])) == str(ServiceReference(serviceref)):
124                                 found = count
125                         count += 1
126                 self.instance.moveSelectionTo(found)
127
128         def moveDown(self):
129                 self.instance.moveSelection(self.instance.moveDown)