add support for opening m3u, pls, extended m3u and extended pls playlists

[enigma2.git] / lib / python / Components / FileList.py

diff --git a/lib/python/Components/FileList.py b/lib/python/Components/FileList.py
index a6b5bf2f520d4f4acbedf0e61f3c2157ee409976..0d8245c29320b5708bbc826ab14274345b021f17 100644 (file)
--- a/lib/python/Components/FileList.py
+++ b/lib/python/Components/FileList.py
@@ -1,5 +1,6 @@
 from HTMLComponent import *
 from GUIComponent import *
+import re
 
 from MenuList import MenuList
 
@@ -16,26 +17,47 @@ RT_VALIGN_TOP = 0
 RT_VALIGN_CENTER = 8
 RT_VALIGN_BOTTOM = 16
 
-def FileEntryComponent(name, absolute, isDir = False):
+EXTENSIONS = {
+               "mp3": "music",
+               "wav": "music",
+               "ogg": "music",
+               "jpg": "picture",
+               "jpeg": "picture",
+               "png": "picture",
+               "ts": "movie",
+               "avi": "movie",
+               "mpg": "movie",
+               "mpeg": "movie",
+       }
+
+def FileEntryComponent(name, absolute = None, isDir = False):
        res = [ (absolute, isDir) ]
-       res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 200, 20, 0, RT_HALIGN_LEFT ,name))
+       res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 200, 20, 0, RT_HALIGN_LEFT, name))
        if isDir:
-               png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/directory.png"))
+               png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
        else:
-               # FIXME: detect file extensions correctly
-               png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/" + name[-3:] + ".png"))
+               extension = name.split('.')
+               extension = extension[-1]
+               if EXTENSIONS.has_key(extension):
+                       png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
+               else:
+                       png = None
        if png is not None:
                res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
        
        return res
 
-class FileList(HTMLComponent, GUIComponent, MenuList):
-       def __init__(self, directory, showDirectories = True, showFiles = True):
+class FileList(MenuList, HTMLComponent, GUIComponent):
+       def __init__(self, directory, showDirectories = True, showFiles = True, matchingPattern = None, useServiceRef = False, isTop = False):
                GUIComponent.__init__(self)
                self.l = eListboxPythonMultiContent()
-
+               
+               self.useServiceRef = useServiceRef
                self.showDirectories = showDirectories
                self.showFiles = showFiles
+               self.isTop = isTop
+               # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
+               self.matchingPattern = matchingPattern
                self.changeDir(directory)
 
                self.l.setFont(0, gFont("Regular", 18))
@@ -43,22 +65,78 @@ class FileList(HTMLComponent, GUIComponent, MenuList):
        def getSelection(self):
                return self.l.getCurrentSelection()[0]
        
+       def getFileList(self):
+               return self.list
+       
        def changeDir(self, directory):
                self.list = []
                
-               directories = os.listdir(directory)
+               directories = []
+               files = []
                
-               if directory != "/" and self.showDirectories:
+               if self.useServiceRef:
+                       root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
+                       serviceHandler = eServiceCenter.getInstance()
+                       list = serviceHandler.list(root)
+                       
+                       while 1:
+                               s = list.getNext()
+                               if not s.valid():
+                                       del list
+                                       break
+                               if s.flags & s.mustDescent:
+                                       directories.append(s.getPath())
+                               else:
+                                       files.append(s)
+                               print s.getName(), s.flags
+               else:
+                       files = os.listdir(directory)
+                       files.sort()
+                       tmpfiles = files[:]
+                       for x in tmpfiles:
+                               if os.path.isdir(directory + x):
+                                       directories.append(directory + x + "/")
+                                       files.remove(x)
+               
+               if directory != "/" and self.showDirectories and not self.isTop:
                        self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
-               for x in directories:
-                       if os.path.isdir(directory + x):
-                               if self.showDirectories:
-                                       self.list.append(FileEntryComponent(name = x, absolute = directory + x + "/" , isDir = True))
-                       elif self.showFiles:
-                                       self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False))
-               self.l.setList(self.list)
+
+               if self.showDirectories:
+                       for x in directories:
+                               name = x.split('/')[-2]
+                               self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
+
+               if self.showFiles:
+                       for x in files:
+                               if self.useServiceRef:
+                                       path = x.getPath()
+                                       name = path.split('/')[-1]
+                               else:
+                                       path = directory + x
+                                       name = x
                                
-       def GUIcreate(self, parent):
-               self.instance = eListbox(parent)
-               self.instance.setContent(self.l)
-               self.instance.setItemHeight(23)
\ No newline at end of file
+                               if self.matchingPattern is not None:
+                                       if re.compile(self.matchingPattern).search(path):
+                                               self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
+                               else:
+                                       self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
+                               
+               self.l.setList(self.list)
+               
+       def canDescent(self):
+               return self.getSelection()[1]
+       
+       def descent(self):
+               self.changeDir(self.getSelection()[0])
+               
+       def getFilename(self):
+               return self.getSelection()[0].getPath()
+
+       def getServiceRef(self):
+               return self.getSelection()[0]
+
+       GUI_WIDGET = eListbox
+
+       def postWidgetCreate(self, instance):
+               instance.setContent(self.l)
+               instance.setItemHeight(23)