add some extension-icons

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

diff --git a/lib/python/Components/FileList.py b/lib/python/Components/FileList.py
index a6b5bf2f520d4f4acbedf0e61f3c2157ee409976..31ffc98a191b383b8aa22672a3badd5a95a8be61 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,6 +17,18 @@ RT_VALIGN_TOP = 0
 RT_VALIGN_CENTER = 8
 RT_VALIGN_BOTTOM = 16
 
+EXTENSIONS = {
+               "mp3": "music",
+               "wav": "music",
+               "jpg": "picture",
+               "jpeg": "picture",
+               "png": "picture",
+               "ts": "movie",
+               "avi": "movie",
+               "mpg": "movie",
+               "mpeg": "movie",
+       }
+
 def FileEntryComponent(name, absolute, isDir = False):
        res = [ (absolute, isDir) ]
        res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 200, 20, 0, RT_HALIGN_LEFT ,name))
@@ -23,19 +36,22 @@ def FileEntryComponent(name, absolute, isDir = False):
                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"))
+               if EXTENSIONS.has_key(name[-3:]):
+                       png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/" + EXTENSIONS[name[-3:]] + ".png"))
        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):
+       def __init__(self, directory, showDirectories = True, showFiles = True, matchingPattern = None):
                GUIComponent.__init__(self)
                self.l = eListboxPythonMultiContent()
 
                self.showDirectories = showDirectories
                self.showFiles = showFiles
+               # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
+               self.matchingPattern = matchingPattern
                self.changeDir(directory)
 
                self.l.setFont(0, gFont("Regular", 18))
@@ -55,7 +71,12 @@ class FileList(HTMLComponent, GUIComponent, MenuList):
                                if self.showDirectories:
                                        self.list.append(FileEntryComponent(name = x, absolute = directory + x + "/" , isDir = True))
                        elif self.showFiles:
+                               if self.matchingPattern is not None:
+                                       if re.compile(self.matchingPattern).search(x):
+                                               self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False))
+                               else:
                                        self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False))
+                               
                self.l.setList(self.list)
                                
        def GUIcreate(self, parent):