1 from re import compile as re_compile
2 from os import path as os_path, listdir
3 from MenuList import MenuList
4 from Components.Harddisk import harddiskmanager
6 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
8 from enigma import RT_HALIGN_LEFT, eListboxPythonMultiContent, \
9 eServiceReference, eServiceCenter, gFont
10 from Tools.LoadPixmap import LoadPixmap
33 def FileEntryComponent(name, absolute = None, isDir = False):
34 res = [ (absolute, isDir) ]
35 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
37 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
39 extension = name.split('.')
40 extension = extension[-1].lower()
41 if EXTENSIONS.has_key(extension):
42 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
46 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
50 class FileList(MenuList):
51 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
52 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
53 self.additional_extensions = additionalExtensions
55 self.current_directory = None
56 self.current_mountpoint = None
57 self.useServiceRef = useServiceRef
58 self.showDirectories = showDirectories
59 self.showMountpoints = showMountpoints
60 self.showFiles = showFiles
62 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
63 self.matchingPattern = matchingPattern
64 self.inhibitDirs = inhibitDirs or []
65 self.inhibitMounts = inhibitMounts or []
67 self.refreshMountpoints()
68 self.changeDir(directory)
69 self.l.setFont(0, gFont("Regular", 18))
70 self.l.setItemHeight(23)
71 self.serviceHandler = eServiceCenter.getInstance()
73 def refreshMountpoints(self):
74 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
75 self.mountpoints.sort(reverse = True)
77 def getMountpoint(self, file):
78 file = os_path.join(os_path.realpath(file), "")
79 for m in self.mountpoints:
80 if file.startswith(m):
84 def getMountpointLink(self, file):
85 if os_path.realpath(file) == file:
86 return self.getMountpoint(file)
90 mp = self.getMountpoint(file)
92 file = os_path.dirname(file)
93 while last != "/" and mp == self.getMountpoint(file):
95 file = os_path.dirname(file)
96 return os_path.join(last, "")
98 def getSelection(self):
99 if self.l.getCurrentSelection() is None:
101 return self.l.getCurrentSelection()[0]
103 def getCurrentEvent(self):
104 l = self.l.getCurrentSelection()
105 if not l or l[0][1] == True:
108 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
110 def getFileList(self):
113 def inParentDirs(self, dir, parents):
114 dir = os_path.realpath(dir)
116 if dir.startswith(p):
120 def changeDir(self, directory, select = None):
123 # if we are just entering from the list of mount points:
124 if self.current_directory is None:
125 if directory and self.showMountpoints:
126 self.current_mountpoint = self.getMountpointLink(directory)
128 self.current_mountpoint = None
129 self.current_directory = directory
133 if directory is None and self.showMountpoints: # present available mountpoints
134 for p in harddiskmanager.getMountedPartitions():
135 path = os_path.join(p.mountpoint, "")
136 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
137 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
140 elif directory is None:
143 elif self.useServiceRef:
144 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
145 if self.additional_extensions:
146 root.setName(self.additional_extensions)
147 serviceHandler = eServiceCenter.getInstance()
148 list = serviceHandler.list(root)
155 if s.flags & s.mustDescent:
156 directories.append(s.getPath())
162 if os_path.exists(directory):
163 files = listdir(directory)
167 if os_path.isdir(directory + x):
168 directories.append(directory + x + "/")
171 if directory is not None and self.showDirectories and not self.isTop:
172 if directory == self.current_mountpoint and self.showMountpoints:
173 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
174 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
175 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
177 if self.showDirectories:
178 for x in directories:
179 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
180 name = x.split('/')[-2]
181 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
185 if self.useServiceRef:
187 name = path.split('/')[-1]
192 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
193 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
195 self.l.setList(self.list)
197 if select is not None:
203 if isinstance(p, eServiceReference):
210 def getCurrentDirectory(self):
211 return self.current_directory
213 def canDescent(self):
214 if self.getSelection() is None:
216 return self.getSelection()[1]
219 if self.getSelection() is None:
221 self.changeDir(self.getSelection()[0], select = self.current_directory)
223 def getFilename(self):
224 if self.getSelection() is None:
226 x = self.getSelection()[0]
227 if isinstance(x, eServiceReference):
231 def getServiceRef(self):
232 if self.getSelection() is None:
234 x = self.getSelection()[0]
235 if isinstance(x, eServiceReference):
240 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
243 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
246 self.changeDir(self.current_directory, self.getFilename())
248 def partitionListChanged(self, action, device):
249 self.refreshMountpoints()
250 if self.current_directory is None:
254 def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
255 res = [ (absolute, isDir, selected, name) ]
256 res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
258 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
260 extension = name.split('.')
261 extension = extension[-1].lower()
262 if EXTENSIONS.has_key(extension):
263 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
267 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
269 if not name.startswith('<'):
270 if selected is False:
271 icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_off.png"))
272 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
274 icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_on.png"))
275 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
280 class MultiFileSelectList(FileList):
281 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):
282 self.selectedFiles = preselectedFiles
283 if self.selectedFiles is None:
284 self.selectedFiles = []
285 FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles, useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
286 self.changeDir(directory)
287 self.l.setItemHeight(25)
288 self.l.setFont(0, gFont("Regular", 20))
289 self.onSelectionChanged = [ ]
291 def selectionChanged(self):
292 for f in self.onSelectionChanged:
295 def changeSelectionState(self):
296 idx = self.l.getCurrentSelectionIndex()
301 if x[0][3].startswith('<'):
305 realPathname = x[0][0]
307 realPathname = self.current_directory + x[0][0]
310 for entry in self.selectedFiles:
311 if entry == realPathname:
312 self.selectedFiles.remove(entry)
316 alreadyinList = False
317 for entry in self.selectedFiles:
318 if entry == realPathname:
320 if not alreadyinList:
321 self.selectedFiles.append(realPathname)
322 newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
329 self.l.setList(self.list)
331 def getSelectedList(self):
332 return self.selectedFiles
334 def changeDir(self, directory, select = None):
337 # if we are just entering from the list of mount points:
338 if self.current_directory is None:
339 if directory and self.showMountpoints:
340 self.current_mountpoint = self.getMountpointLink(directory)
342 self.current_mountpoint = None
343 self.current_directory = directory
347 if directory is None and self.showMountpoints: # present available mountpoints
348 for p in harddiskmanager.getMountedPartitions():
349 path = os_path.join(p.mountpoint, "")
350 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
351 self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
354 elif directory is None:
357 elif self.useServiceRef:
358 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
359 if self.additional_extensions:
360 root.setName(self.additional_extensions)
361 serviceHandler = eServiceCenter.getInstance()
362 list = serviceHandler.list(root)
369 if s.flags & s.mustDescent:
370 directories.append(s.getPath())
376 if os_path.exists(directory):
377 files = listdir(directory)
381 if os_path.isdir(directory + x):
382 directories.append(directory + x + "/")
385 if directory is not None and self.showDirectories and not self.isTop:
386 if directory == self.current_mountpoint and self.showMountpoints:
387 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
388 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
389 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
391 if self.showDirectories:
392 for x in directories:
393 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
394 name = x.split('/')[-2]
395 alreadySelected = False
396 for entry in self.selectedFiles:
398 alreadySelected = True
400 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
402 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
406 if self.useServiceRef:
408 name = path.split('/')[-1]
413 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
414 alreadySelected = False
415 for entry in self.selectedFiles:
416 if os_path.basename(entry) == x:
417 alreadySelected = True
419 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
421 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
423 self.l.setList(self.list)
425 if select is not None:
431 if isinstance(p, eServiceReference):