3 from os import path as os_path, listdir
4 from MenuList import MenuList
5 from Components.Harddisk import harddiskmanager
7 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
9 from enigma import RT_HALIGN_LEFT, eListbox, eListboxPythonMultiContent, \
10 eServiceReference, eServiceCenter, gFont
11 from Tools.LoadPixmap import LoadPixmap
27 def FileEntryComponent(name, absolute = None, isDir = False):
28 res = [ (absolute, isDir) ]
29 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
31 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
33 extension = name.split('.')
34 extension = extension[-1].lower()
35 if EXTENSIONS.has_key(extension):
36 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
40 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
44 class FileList(MenuList):
45 def __init__(self, directory, showDirectories = True, showFiles = True, matchingPattern = None, useServiceRef = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
46 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
47 self.additional_extensions = additionalExtensions
48 self.mount_point = None
49 self.current_directory = None
50 self.useServiceRef = useServiceRef
51 self.showDirectories = showDirectories
52 self.showFiles = showFiles
54 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
55 self.matchingPattern = matchingPattern
56 self.changeDir(directory)
58 self.l.setFont(0, gFont("Regular", 18))
59 self.l.setItemHeight(23)
60 self.serviceHandler = eServiceCenter.getInstance()
62 def getSelection(self):
63 if self.l.getCurrentSelection() is None:
65 return self.l.getCurrentSelection()[0]
67 def getCurrentEvent(self):
68 l = self.l.getCurrentSelection()
69 if not l or l[0][1] == True:
72 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
74 def getFileList(self):
77 def changeDir(self, directory, select = None):
80 # if we are just entering from the list of mount points:
81 if self.current_directory is None:
83 self.mount_point = None
85 # Sort Mountpoints by length (longest first)
86 sortedp = harddiskmanager.getMountedPartitions()
87 sortedp.sort(key=lambda p: 0 - len(p.mountpoint))
89 # Search for the longest matching mp (should at least match /)
91 if directory.startswith(p.mountpoint):
92 self.mount_point = p.mountpoint
93 if p.mountpoint != "/":
94 self.mount_point += "/"
96 self.current_directory = directory
100 if directory is None: # present available mountpoints
101 print "listing partitions:"
102 for p in harddiskmanager.getMountedPartitions():
103 if p.mountpoint == "/":
104 self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint, isDir = True))
106 self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint + "/", isDir = True))
109 elif self.useServiceRef:
110 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
111 if self.additional_extensions:
112 root.setName(self.additional_extensions)
113 serviceHandler = eServiceCenter.getInstance()
114 list = serviceHandler.list(root)
121 if s.flags & s.mustDescent:
122 directories.append(s.getPath())
125 print s.getName(), s.flags
129 if os_path.exists(directory):
130 files = listdir(directory)
134 if os_path.isdir(directory + x):
135 directories.append(directory + x + "/")
138 if directory is not None and self.showDirectories and not self.isTop:
139 if directory == self.mount_point:
140 self.list.append(FileEntryComponent(name = ".. (" +_("List of Storage Devices") + ")", absolute = None, isDir = True))
142 self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
144 if self.showDirectories:
145 for x in directories:
146 name = x.split('/')[-2]
147 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
151 if self.useServiceRef:
153 name = path.split('/')[-1]
158 if self.matchingPattern is not None:
159 if re.compile(self.matchingPattern).search(path):
160 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
162 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
164 self.l.setList(self.list)
166 if select is not None:
172 if isinstance(p, eServiceReference):
179 def getCurrentDirectory(self):
180 return self.current_directory
182 def canDescent(self):
183 if self.getSelection() is None:
185 return self.getSelection()[1]
188 if self.getSelection() is None:
190 self.changeDir(self.getSelection()[0], select = self.current_directory)
192 def getFilename(self):
193 if self.getSelection() is None:
195 x = self.getSelection()[0]
196 if isinstance(x, eServiceReference):
200 def getServiceRef(self):
201 if self.getSelection() is None:
203 x = self.getSelection()[0]
204 if isinstance(x, eServiceReference):
209 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
212 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
215 self.changeDir(self.current_directory, self.getFilename())
217 def partitionListChanged(self, action, device):
218 if self.current_directory is None: