code.vuplus.com Git - vuplus_dvbapp/blob - 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                                 remote_protos = ["http", "https", "udp", "rtsp", "rtp", "mmp"]
  77                                 if entry[0] == "/":
  78                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry)
  79                                 else:
  80                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + entry)
  81                                         for proto in remote_protos:
  82                                                 if entry.startswith(proto):
  83                                                         sref = ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry.replace(':',"%3a"))
  84                                 if self.displayname:
  85                                         sref.ref.setName(self.displayname)
  86                                         self.displayname = None
  87                                 self.addService(sref)
  88                 file.close()
  89                 return self.list
  90
  91         def save(self, filename = None):
  92                 return self.ERROR
  93
  94 class PlaylistIOPLS(PlaylistIO):
  95         def __init__(self):
  96                 PlaylistIO.__init__(self)
  97
  98         def open(self, filename):
  99                 self.clear()
 100                 try:
 101                         file = open(filename, "r")
 102                 except IOError:
 103                         return None
 104                 entry = file.readline().strip()
 105                 if entry == "[playlist]": # extended pls
 106                         while True:
 107                                 entry = file.readline().strip()
 108                                 if entry == "":
 109                                         break
 110                                 if entry[0:4] == "File":
 111                                         pos = entry.find('=') + 1
 112                                         newentry = entry[pos:]
 113                                         # TODO: use e2 facilities to create a service ref from file
 114                                         if newentry[0] == "/":
 115                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + newentry))
 116                                         else:
 117                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + newentry))
 118                 else:
 119                         playlist = PlaylistIOM3U()
 120                         return playlist.open(filename)
 121                 file.close()
 122                 return self.list
 123
 124         def save(self, filename = None):
 125                 return self.ERROR