Merge branch 'bug_249_current_skin_fallback' into experimental
[vuplus_dvbapp] / lib / python / Components / FileList.py
old mode 100644 (file)
new mode 100755 (executable)
index 5824747..222512e
@@ -1,39 +1,45 @@
-import re
-
+from re import compile as re_compile
 from os import path as os_path, listdir
 from MenuList import MenuList
 from Components.Harddisk import harddiskmanager
 
-from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
+from Tools.Directories import SCOPE_CURRENT_SKIN, resolveFilename, fileExists
 
-from enigma import RT_HALIGN_LEFT, eListbox, eListboxPythonMultiContent, \
+from enigma import RT_HALIGN_LEFT, eListboxPythonMultiContent, \
        eServiceReference, eServiceCenter, gFont
 from Tools.LoadPixmap import LoadPixmap
 
 EXTENSIONS = {
+               "m4a": "music",
+               "mp2": "music",
                "mp3": "music",
                "wav": "music",
                "ogg": "music",
+               "flac": "music",
                "jpg": "picture",
                "jpeg": "picture",
                "png": "picture",
                "bmp": "picture",
                "ts": "movie",
                "avi": "movie",
+               "divx": "movie",
                "mpg": "movie",
                "mpeg": "movie",
+               "mkv": "movie",
+               "mp4": "movie",
+               "mov": "movie"
        }
 
 def FileEntryComponent(name, absolute = None, isDir = False):
        res = [ (absolute, isDir) ]
        res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
        if isDir:
-               png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
+               png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
        else:
                extension = name.split('.')
                extension = extension[-1].lower()
                if EXTENSIONS.has_key(extension):
-                       png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
+                       png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
                else:
                        png = None
        if png is not None:
@@ -42,7 +48,7 @@ def FileEntryComponent(name, absolute = None, isDir = False):
        return res
 
 class FileList(MenuList):
-       def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, enableWrapAround = False, additionalExtensions = None):
+       def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
                MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
                self.additional_extensions = additionalExtensions
                self.mountpoints = []
@@ -52,6 +58,7 @@ class FileList(MenuList):
                self.showDirectories = showDirectories
                self.showMountpoints = showMountpoints
                self.showFiles = showFiles
+               self.isTop = isTop
                # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
                self.matchingPattern = matchingPattern
                self.inhibitDirs = inhibitDirs or []
@@ -113,8 +120,6 @@ class FileList(MenuList):
        def changeDir(self, directory, select = None):
                self.list = []
 
-               if directory and not os_path.isdir(directory):
-                       directory = None
                # if we are just entering from the list of mount points:
                if self.current_directory is None:
                        if directory and self.showMountpoints:
@@ -127,11 +132,14 @@ class FileList(MenuList):
 
                if directory is None and self.showMountpoints: # present available mountpoints
                        for p in harddiskmanager.getMountedPartitions():
-                               path = os_path.join(p.mountpoint,"")
-                               if not self.inhibitMounts or ((not path in self.inhibitMounts) and (not self.inParentDirs(path, self.inhibitDirs))):
+                               path = os_path.join(p.mountpoint, "")
+                               if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
                                        self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
                        files = [ ]
                        directories = [ ]
+               elif directory is None:
+                       files = [ ]
+                       directories = [ ]
                elif self.useServiceRef:
                        root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
                        if self.additional_extensions:
@@ -151,8 +159,11 @@ class FileList(MenuList):
                        directories.sort()
                        files.sort()
                else:
-                       if os_path.exists(directory):
-                               files = listdir(directory)
+                       if fileExists(directory):
+                               try:
+                                       files = listdir(directory)
+                               except:
+                                       files = []
                                files.sort()
                                tmpfiles = files[:]
                                for x in tmpfiles:
@@ -160,7 +171,7 @@ class FileList(MenuList):
                                                directories.append(directory + x + "/")
                                                files.remove(x)
 
-               if directory is not None and self.showDirectories:
+               if directory is not None and self.showDirectories and not self.isTop:
                        if directory == self.current_mountpoint and self.showMountpoints:
                                self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
                        elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
@@ -181,7 +192,7 @@ class FileList(MenuList):
                                        path = directory + x
                                        name = x
 
-                               if (self.matchingPattern is None) or re.compile(self.matchingPattern).search(path):
+                               if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
                                        self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
 
                self.l.setList(self.list)
@@ -241,3 +252,192 @@ class FileList(MenuList):
                self.refreshMountpoints()
                if self.current_directory is None:
                        self.refresh()
+
+
+def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
+       res = [ (absolute, isDir, selected, name) ]
+       res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
+       if isDir:
+               png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
+       else:
+               extension = name.split('.')
+               extension = extension[-1].lower()
+               if EXTENSIONS.has_key(extension):
+                       png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
+               else:
+                       png = None
+       if png is not None:
+               res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
+
+       if not name.startswith('<'):
+               if selected is False:
+                       icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_off.png"))
+                       res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
+               else:
+                       icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_on.png"))
+                       res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
+       
+       return res
+
+
+class MultiFileSelectList(FileList):
+       def __init__(self, preselectedFiles, directory, showMountpoints = False, matchingPattern = None, showDirectories = True, showFiles = True,  useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
+               self.selectedFiles = preselectedFiles
+               if self.selectedFiles is None:
+                       self.selectedFiles = []
+               FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles,  useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
+               self.changeDir(directory)                       
+               self.l.setItemHeight(25)
+               self.l.setFont(0, gFont("Regular", 20))
+               self.onSelectionChanged = [ ]
+
+       def selectionChanged(self):
+               for f in self.onSelectionChanged:
+                       f()
+
+       def changeSelectionState(self):
+               idx = self.l.getCurrentSelectionIndex()
+               count = 0
+               newList = []
+               for x in self.list:
+                       if idx == count:
+                               if x[0][3].startswith('<'):
+                                       newList.append(x)
+                               else:
+                                       if x[0][1] is True:
+                                               realPathname = x[0][0]
+                                       else:
+                                               realPathname = self.current_directory + x[0][0]
+                                       if x[0][2] == True:
+                                               SelectState = False
+                                               for entry in self.selectedFiles:
+                                                       if entry == realPathname:
+                                                               self.selectedFiles.remove(entry)
+       
+                                       else:
+                                               SelectState = True
+                                               alreadyinList = False   
+                                               for entry in self.selectedFiles:
+                                                       if entry == realPathname:
+                                                               alreadyinList = True
+                                               if not alreadyinList:
+                                                       self.selectedFiles.append(realPathname)
+                                       newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
+                       else:
+                               newList.append(x)
+                       
+                       count += 1
+               
+               self.list = newList
+               self.l.setList(self.list)
+       
+       def getSelectedList(self):
+               return self.selectedFiles
+
+       def changeDir(self, directory, select = None):
+               self.list = []
+
+               # if we are just entering from the list of mount points:
+               if self.current_directory is None:
+                       if directory and self.showMountpoints:
+                               self.current_mountpoint = self.getMountpointLink(directory)
+                       else:
+                               self.current_mountpoint = None
+               self.current_directory = directory
+               directories = []
+               files = []
+
+               if directory is None and self.showMountpoints: # present available mountpoints
+                       for p in harddiskmanager.getMountedPartitions():
+                               path = os_path.join(p.mountpoint, "")
+                               if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
+                                       self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
+                       files = [ ]
+                       directories = [ ]
+               elif directory is None:
+                       files = [ ]
+                       directories = [ ]
+               elif self.useServiceRef:
+                       root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
+                       if self.additional_extensions:
+                               root.setName(self.additional_extensions)
+                       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)
+                       directories.sort()
+                       files.sort()
+               else:
+                       if fileExists(directory):
+                               try:
+                                       files = listdir(directory)
+                               except:
+                                       files = []
+                               files.sort()
+                               tmpfiles = files[:]
+                               for x in tmpfiles:
+                                       if os_path.isdir(directory + x):
+                                               directories.append(directory + x + "/")
+                                               files.remove(x)
+
+               if directory is not None and self.showDirectories and not self.isTop:
+                       if directory == self.current_mountpoint and self.showMountpoints:
+                               self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
+                       elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
+                               self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
+
+               if self.showDirectories:
+                       for x in directories:
+                               if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
+                                       name = x.split('/')[-2]
+                                       alreadySelected = False
+                                       for entry in self.selectedFiles:
+                                               if entry  == x:
+                                                       alreadySelected = True                                  
+                                       if alreadySelected:             
+                                               self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
+                                       else:
+                                               self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
+                                               
+               if self.showFiles:
+                       for x in files:
+                               if self.useServiceRef:
+                                       path = x.getPath()
+                                       name = path.split('/')[-1]
+                               else:
+                                       path = directory + x
+                                       name = x
+
+                               if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
+                                       alreadySelected = False
+                                       for entry in self.selectedFiles:
+                                               if os_path.basename(entry)  == x:
+                                                       alreadySelected = True  
+                                       if alreadySelected:
+                                               self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
+                                       else:
+                                               self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
+
+               self.l.setList(self.list)
+
+               if select is not None:
+                       i = 0
+                       self.moveToIndex(0)
+                       for x in self.list:
+                               p = x[0][0]
+                               
+                               if isinstance(p, eServiceReference):
+                                       p = p.getPath()
+                               
+                               if p == select:
+                                       self.moveToIndex(i)
+                               i += 1
+