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                 try:
  62                         file = open(filename, "r")
  63                 except IOError:
  64                         return None
  65                 while True:
  66                         entry = file.readline().strip()
  67                         if entry == "":
  68                                 break
  69                         if entry[0] != "#":
  70                                 # TODO: use e2 facilities to create a service ref from file
  71                                 if entry[0] == "/":
  72                                         self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry))
  73                                 elif entry.startswith("http"):
  74                                         self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + entry.replace(':',"%3a")))
  75                                 else:
  76                                         self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + entry))
  77                 file.close()
  78                 return self.list
  79
  80         def save(self, filename = None):
  81                 return self.ERROR
  82
  83 class PlaylistIOPLS(PlaylistIO):
  84         def __init__(self):
  85                 PlaylistIO.__init__(self)
  86
  87         def open(self, filename):
  88                 self.clear()
  89                 try:
  90                         file = open(filename, "r")
  91                 except IOError:
  92                         return None
  93                 entry = file.readline().strip()
  94                 if entry == "[playlist]": # extended pls
  95                         while True:
  96                                 entry = file.readline().strip()
  97                                 if entry == "":
  98                                         break
  99                                 if entry[0:4] == "File":
 100                                         pos = entry.find('=') + 1
 101                                         newentry = entry[pos:]
 102                                         # TODO: use e2 facilities to create a service ref from file
 103                                         if newentry[0] == "/":
 104                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + newentry))
 105                                         else:
 106                                                 self.addService(ServiceReference("4097:0:0:0:0:0:0:0:0:0:" + os.path.dirname(filename) + "/" + newentry))
 107                 else:
 108                         playlist = PlaylistIOM3U()
 109                         return playlist.open(filename)
 110                 file.close()
 111                 return self.list
 112
 113         def save(self, filename = None):
 114                 return self.ERROR