X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/105064cb459d5027ab5cfacb20f31ab04569ace2..e25c93ddcc6a2c73b413fced441624703901e22d:/lib/python/Components/FileList.py diff --git a/lib/python/Components/FileList.py b/lib/python/Components/FileList.py index 31ffc98a..3091436f 100644 --- a/lib/python/Components/FileList.py +++ b/lib/python/Components/FileList.py @@ -2,84 +2,191 @@ from HTMLComponent import * from GUIComponent import * import re +from os import path as os_path, listdir from MenuList import MenuList +from Components.Harddisk import harddiskmanager -from Tools.Directories import * +from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename -from enigma import * - -RT_HALIGN_LEFT = 0 -RT_HALIGN_RIGHT = 1 -RT_HALIGN_CENTER = 2 -RT_HALIGN_BLOCK = 4 - -RT_VALIGN_TOP = 0 -RT_VALIGN_CENTER = 8 -RT_VALIGN_BOTTOM = 16 +from enigma import RT_HALIGN_LEFT, loadPNG, eListbox, eListboxPythonMultiContent, \ + eServiceReference, eServiceCenter, gFont EXTENSIONS = { "mp3": "music", "wav": "music", + "ogg": "music", "jpg": "picture", "jpeg": "picture", "png": "picture", + "bmp": "picture", "ts": "movie", "avi": "movie", "mpg": "movie", "mpeg": "movie", } -def FileEntryComponent(name, absolute, isDir = False): +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, 470, 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 - if EXTENSIONS.has_key(name[-3:]): - png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/" + EXTENSIONS[name[-3:]] + ".png")) + extension = name.split('.') + extension = extension[-1].lower() + 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, matchingPattern = None): +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.mount_point = None + self.current_directory = None + 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)) - + self.l.setItemHeight(23) + def getSelection(self): + if self.l.getCurrentSelection() is None: + return None return self.l.getCurrentSelection()[0] - def changeDir(self, directory): + def getSelectionIndex(self): + return self.l.getCurrentSelectionIndex() + + def getFileList(self): + return self.list + + def changeDir(self, directory, select = None): self.list = [] - directories = os.listdir(directory) + # if we are just entering from the list of mount points: + if self.current_directory is None: + self.mount_point = directory + self.current_directory = directory + directories = [] + files = [] - if directory != "/" and self.showDirectories: - 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: - if self.matchingPattern is not None: - if re.compile(self.matchingPattern).search(x): - self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False)) + if directory is None: # present available mountpoints + print "listing partitions:" + for p in harddiskmanager.getMountedPartitions(): + if p.mountpoint == "/": + self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint, isDir = True)) + else: + self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint + "/", isDir = True)) + files = [ ] + directories = [ ] + elif 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 + directories.sort() + files.sort() + else: + files = 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 is not None and self.showDirectories and not self.isTop: + if directory == self.mount_point: + self.list.append(FileEntryComponent(name = ".. (" +_("List of Storage Devices") + ")", absolute = None, isDir = True)) + else: + self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True)) + + 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: - self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False)) + path = directory + x + name = x + 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) + + 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() - def GUIcreate(self, parent): - self.instance = eListbox(parent) - self.instance.setContent(self.l) - self.instance.setItemHeight(23) \ No newline at end of file + if p == select: + self.moveToIndex(i) + i += 1 + + def getCurrentDirectory(self): + return self.current_directory + + def canDescent(self): + if self.getSelection() is None: + return False + return self.getSelection()[1] + + def descent(self): + if self.getSelection() is None: + return + self.changeDir(self.getSelection()[0], select = self.current_directory) + + def getFilename(self): + if self.getSelection() is None: + return None + x = self.getSelection()[0] + if isinstance(x, eServiceReference): + x = x.getPath() + return x + + def getServiceRef(self): + if self.getSelection() is None: + return None + x = self.getSelection()[0] + if isinstance(x, eServiceReference): + return x + return None + + GUI_WIDGET = eListbox + + def postWidgetCreate(self, instance): + instance.setContent(self.l)