Merge branch 'master' of git.opendreambox.org:/git/enigma2
[enigma2.git] / lib / python / Components / Playlist.py
1 from ServiceReference import ServiceReference
2 import os
3
4 class PlaylistIO:       
5         def __init__(self):
6                 self.list = []
7         
8         # returns a list of services or None if filename is not a valid playlist
9         def open(self, filename):
10                 return None
11         
12         OK = 0
13         FILEEXISTS = 1
14         WRITEERROR = 2
15         ERROR = 3
16         UNSUPPORTED_FILES_IN_PLAYLIST = 4
17         
18         def save(self, filename = None):
19                 return self.ERROR
20                 
21         def clear(self):
22                 del self.list[:]
23                 
24         def addService(self, service):
25                 self.list.append(service)
26                 
27                 
28 class PlaylistIOInternal(PlaylistIO):
29         def __init__(self):
30                 PlaylistIO.__init__(self)
31         
32         def open(self, filename):
33                 self.clear()
34                 try:
35                         file = open(filename, "r")
36                 except IOError:
37                         return None
38                 while True:
39                         entry = file.readline().strip()
40                         if entry == "":
41                                 break
42                         self.addService(ServiceReference(entry))
43                 file.close()
44                 return self.list
45                 
46         def save(self, filename = None):
47                 print "Writing playlist into file", filename
48                 file = open(filename, "w")
49                 for x in self.list:
50                         file.write(str(x) + "\n")
51                 file.close()
52                 
53                 return self.OK
54         
55 class PlaylistIOM3U(PlaylistIO):
56         def __init__(self):
57                 PlaylistIO.__init__(self)
58         
59         def open(self, filename):
60                 self.clear()
61                 self.displayname = None
62                 try:
63                         file = open(filename, "r")
64                 except IOError:
65                         return None
66                 while True:
67                         entry = file.readline().strip()
68                         if entry == "":
69                                 break
70                         if entry.startswith("#EXTINF:"):
71                                 extinf = entry.split(',',1)
72                                 if len(extinf) > 1:
73                                         self.displayname = extinf[1]
74                                 # TODO: use e2 facilities to create a service ref from file
75                         elif entry[0] != "#":
76                                 if entry[0] == "/":
77                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry)
78                                 elif entry.startswith("http"):
79                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry.replace(':',"%3a"))
80                                 else:
81                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + entry)
82                                 if self.displayname:
83                                         sref.ref.setName(self.displayname)
84                                         self.displayname = None
85                                 self.addService(sref)
86                 file.close()
87                 return self.list
88                 
89         def save(self, filename = None):
90                 return self.ERROR
91         
92 class PlaylistIOPLS(PlaylistIO):
93         def __init__(self):
94                 PlaylistIO.__init__(self)
95         
96         def open(self, filename):
97                 self.clear()
98                 try:
99                         file = open(filename, "r")
100                 except IOError:
101                         return None
102                 entry = file.readline().strip()
103                 if entry == "[playlist]": # extended pls
104                         while True:
105                                 entry = file.readline().strip()
106                                 if entry == "":
107                                         break
108                                 if entry[0:4] == "File":
109                                         pos = entry.find('=') + 1
110                                         newentry = entry[pos:]
111                                         # TODO: use e2 facilities to create a service ref from file
112                                         if newentry[0] == "/":
113                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + newentry))
114                                         else:
115                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + newentry))
116                 else:
117                         playlist = PlaylistIOM3U()
118                         return playlist.open(filename)
119                 file.close()
120                 return self.list
121                 
122         def save(self, filename = None):
123                 return self.ERROR