X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/f458abcfbe30c3e3062a41b88d3244147bdc0607..51d1d57abb876d78807d95e0aa9ce3dd56e5f259:/lib/python/Components/FileList.py

diff --git a/lib/python/Components/FileList.py b/lib/python/Components/FileList.py
index a6b5bf2f..a0b01b2e 100644
--- 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, 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
-		png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/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):
+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,80 @@ 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
+			directories.sort()
+			files.sort()
+		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)